127 lines
3.5 KiB
C++
127 lines
3.5 KiB
C++
// This file is part of the Luau programming language and is licensed under MIT License; see LICENSE.txt for details
|
|
|
|
#include "Fixture.h"
|
|
#include "Luau/ConstraintGraphBuilder.h"
|
|
|
|
#include "doctest.h"
|
|
|
|
using namespace Luau;
|
|
|
|
TEST_SUITE_BEGIN("ConstraintGraphBuilder");
|
|
|
|
TEST_CASE_FIXTURE(ConstraintGraphBuilderFixture, "hello_world")
|
|
{
|
|
AstStatBlock* block = parse(R"(
|
|
local a = "hello"
|
|
local b = a
|
|
)");
|
|
|
|
cgb.visit(block);
|
|
auto constraints = collectConstraints(cgb.rootScope);
|
|
|
|
REQUIRE(2 == constraints.size());
|
|
|
|
ToStringOptions opts;
|
|
CHECK("string <: a" == toString(*constraints[0], opts));
|
|
CHECK("a <: b" == toString(*constraints[1], opts));
|
|
}
|
|
|
|
TEST_CASE_FIXTURE(ConstraintGraphBuilderFixture, "primitives")
|
|
{
|
|
AstStatBlock* block = parse(R"(
|
|
local s = "hello"
|
|
local n = 555
|
|
local b = true
|
|
local n2 = nil
|
|
)");
|
|
|
|
cgb.visit(block);
|
|
auto constraints = collectConstraints(cgb.rootScope);
|
|
|
|
REQUIRE(3 == constraints.size());
|
|
|
|
ToStringOptions opts;
|
|
CHECK("string <: a" == toString(*constraints[0], opts));
|
|
CHECK("number <: b" == toString(*constraints[1], opts));
|
|
CHECK("boolean <: c" == toString(*constraints[2], opts));
|
|
}
|
|
|
|
TEST_CASE_FIXTURE(ConstraintGraphBuilderFixture, "nil_primitive")
|
|
{
|
|
AstStatBlock* block = parse(R"(
|
|
local function a() return nil end
|
|
local b = a()
|
|
)");
|
|
|
|
cgb.visit(block);
|
|
auto constraints = collectConstraints(cgb.rootScope);
|
|
|
|
ToStringOptions opts;
|
|
REQUIRE(5 <= constraints.size());
|
|
|
|
CHECK("*blocked-1* ~ gen () -> (a...)" == toString(*constraints[0], opts));
|
|
CHECK("b ~ inst *blocked-1*" == toString(*constraints[1], opts));
|
|
CHECK("() -> (c...) <: b" == toString(*constraints[2], opts));
|
|
CHECK("c... <: d" == toString(*constraints[3], opts));
|
|
CHECK("nil <: a..." == toString(*constraints[4], opts));
|
|
}
|
|
|
|
TEST_CASE_FIXTURE(ConstraintGraphBuilderFixture, "function_application")
|
|
{
|
|
AstStatBlock* block = parse(R"(
|
|
local a = "hello"
|
|
local b = a("world")
|
|
)");
|
|
|
|
cgb.visit(block);
|
|
auto constraints = collectConstraints(cgb.rootScope);
|
|
|
|
REQUIRE(4 == constraints.size());
|
|
|
|
ToStringOptions opts;
|
|
CHECK("string <: a" == toString(*constraints[0], opts));
|
|
CHECK("b ~ inst a" == toString(*constraints[1], opts));
|
|
CHECK("(string) -> (c...) <: b" == toString(*constraints[2], opts));
|
|
CHECK("c... <: d" == toString(*constraints[3], opts));
|
|
}
|
|
|
|
TEST_CASE_FIXTURE(ConstraintGraphBuilderFixture, "local_function_definition")
|
|
{
|
|
AstStatBlock* block = parse(R"(
|
|
local function f(a)
|
|
return a
|
|
end
|
|
)");
|
|
|
|
cgb.visit(block);
|
|
auto constraints = collectConstraints(cgb.rootScope);
|
|
|
|
REQUIRE(2 == constraints.size());
|
|
|
|
ToStringOptions opts;
|
|
CHECK("*blocked-1* ~ gen (a) -> (b...)" == toString(*constraints[0], opts));
|
|
CHECK("a <: b..." == toString(*constraints[1], opts));
|
|
}
|
|
|
|
TEST_CASE_FIXTURE(ConstraintGraphBuilderFixture, "recursive_function")
|
|
{
|
|
AstStatBlock* block = parse(R"(
|
|
local function f(a)
|
|
return f(a)
|
|
end
|
|
)");
|
|
|
|
cgb.visit(block);
|
|
auto constraints = collectConstraints(cgb.rootScope);
|
|
|
|
REQUIRE(4 == constraints.size());
|
|
|
|
ToStringOptions opts;
|
|
CHECK("*blocked-1* ~ gen (a) -> (b...)" == toString(*constraints[0], opts));
|
|
CHECK("c ~ inst (a) -> (b...)" == toString(*constraints[1], opts));
|
|
CHECK("(a) -> (d...) <: c" == toString(*constraints[2], opts));
|
|
CHECK("d... <: b..." == toString(*constraints[3], opts));
|
|
}
|
|
|
|
TEST_SUITE_END();
|