108 lines
3.0 KiB
C++
108 lines
3.0 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);
|
|
std::vector<const Constraint*> constraints = collectConstraints(cgb.rootScope);
|
|
|
|
REQUIRE(2 == constraints.size());
|
|
|
|
ToStringOptions opts;
|
|
CHECK("a <: string" == toString(*constraints[0], opts));
|
|
CHECK("b <: a" == 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);
|
|
std::vector<const Constraint*> constraints = collectConstraints(cgb.rootScope);
|
|
|
|
REQUIRE(4 == constraints.size());
|
|
|
|
ToStringOptions opts;
|
|
CHECK("a <: string" == toString(*constraints[0], opts));
|
|
CHECK("b <: number" == toString(*constraints[1], opts));
|
|
CHECK("c <: boolean" == toString(*constraints[2], opts));
|
|
CHECK("d <: nil" == toString(*constraints[3], opts));
|
|
}
|
|
|
|
TEST_CASE_FIXTURE(ConstraintGraphBuilderFixture, "function_application")
|
|
{
|
|
AstStatBlock* block = parse(R"(
|
|
local a = "hello"
|
|
local b = a("world")
|
|
)");
|
|
|
|
cgb.visit(block);
|
|
std::vector<const Constraint*> constraints = collectConstraints(cgb.rootScope);
|
|
|
|
REQUIRE(4 == constraints.size());
|
|
|
|
ToStringOptions opts;
|
|
CHECK("a <: string" == toString(*constraints[0], opts));
|
|
CHECK("b ~ inst a" == toString(*constraints[1], opts));
|
|
CHECK("(string) -> (c, d...) <: b" == toString(*constraints[2], opts));
|
|
CHECK("e <: c" == 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);
|
|
std::vector<const Constraint*> constraints = collectConstraints(cgb.rootScope);
|
|
|
|
REQUIRE(2 == constraints.size());
|
|
|
|
ToStringOptions opts;
|
|
CHECK("a ~ gen (b) -> (c...)" == toString(*constraints[0], opts));
|
|
CHECK("b <: c..." == 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);
|
|
std::vector<const Constraint*> constraints = collectConstraints(cgb.rootScope);
|
|
|
|
REQUIRE(4 == constraints.size());
|
|
|
|
ToStringOptions opts;
|
|
CHECK("a ~ gen (b) -> (c...)" == toString(*constraints[0], opts));
|
|
CHECK("d ~ inst a" == toString(*constraints[1], opts));
|
|
CHECK("(b) -> (e, f...) <: d" == toString(*constraints[2], opts));
|
|
CHECK("e <: c..." == toString(*constraints[3], opts));
|
|
}
|
|
|
|
TEST_SUITE_END();
|