2022-06-03 23:15:45 +01:00
|
|
|
// 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);
|
2022-07-01 00:52:43 +01:00
|
|
|
auto constraints = collectConstraints(NotNull(cgb.rootScope));
|
2022-06-03 23:15:45 +01:00
|
|
|
|
|
|
|
REQUIRE(2 == constraints.size());
|
|
|
|
|
|
|
|
ToStringOptions opts;
|
2022-06-17 02:05:14 +01:00
|
|
|
CHECK("string <: a" == toString(*constraints[0], opts));
|
|
|
|
CHECK("a <: b" == toString(*constraints[1], opts));
|
2022-06-03 23:15:45 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE_FIXTURE(ConstraintGraphBuilderFixture, "primitives")
|
|
|
|
{
|
|
|
|
AstStatBlock* block = parse(R"(
|
|
|
|
local s = "hello"
|
|
|
|
local n = 555
|
|
|
|
local b = true
|
|
|
|
local n2 = nil
|
|
|
|
)");
|
|
|
|
|
|
|
|
cgb.visit(block);
|
2022-07-01 00:52:43 +01:00
|
|
|
auto constraints = collectConstraints(NotNull(cgb.rootScope));
|
2022-06-03 23:15:45 +01:00
|
|
|
|
2022-06-17 02:05:14 +01:00
|
|
|
REQUIRE(3 == constraints.size());
|
2022-06-03 23:15:45 +01:00
|
|
|
|
|
|
|
ToStringOptions opts;
|
2022-06-17 02:05:14 +01:00
|
|
|
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);
|
2022-07-01 00:52:43 +01:00
|
|
|
auto constraints = collectConstraints(NotNull(cgb.rootScope));
|
2022-06-17 02:05:14 +01:00
|
|
|
|
|
|
|
ToStringOptions opts;
|
2022-09-02 00:14:03 +01:00
|
|
|
REQUIRE(4 <= constraints.size());
|
2022-06-17 02:05:14 +01:00
|
|
|
|
|
|
|
CHECK("*blocked-1* ~ gen () -> (a...)" == toString(*constraints[0], opts));
|
2022-09-02 00:14:03 +01:00
|
|
|
CHECK("call *blocked-1* with { result = *blocked-tp-1* }" == toString(*constraints[1], opts));
|
|
|
|
CHECK("*blocked-tp-1* <: b" == toString(*constraints[2], opts));
|
|
|
|
CHECK("nil <: a..." == toString(*constraints[3], opts));
|
2022-06-03 23:15:45 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE_FIXTURE(ConstraintGraphBuilderFixture, "function_application")
|
|
|
|
{
|
|
|
|
AstStatBlock* block = parse(R"(
|
|
|
|
local a = "hello"
|
|
|
|
local b = a("world")
|
|
|
|
)");
|
|
|
|
|
|
|
|
cgb.visit(block);
|
2022-07-01 00:52:43 +01:00
|
|
|
auto constraints = collectConstraints(NotNull(cgb.rootScope));
|
2022-06-03 23:15:45 +01:00
|
|
|
|
2022-09-02 00:14:03 +01:00
|
|
|
REQUIRE(3 == constraints.size());
|
2022-06-03 23:15:45 +01:00
|
|
|
|
|
|
|
ToStringOptions opts;
|
2022-06-17 02:05:14 +01:00
|
|
|
CHECK("string <: a" == toString(*constraints[0], opts));
|
2022-09-02 00:14:03 +01:00
|
|
|
CHECK("call a with { result = *blocked-tp-1* }" == toString(*constraints[1], opts));
|
|
|
|
CHECK("*blocked-tp-1* <: b" == toString(*constraints[2], opts));
|
2022-06-03 23:15:45 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE_FIXTURE(ConstraintGraphBuilderFixture, "local_function_definition")
|
|
|
|
{
|
|
|
|
AstStatBlock* block = parse(R"(
|
|
|
|
local function f(a)
|
|
|
|
return a
|
|
|
|
end
|
|
|
|
)");
|
|
|
|
|
|
|
|
cgb.visit(block);
|
2022-07-01 00:52:43 +01:00
|
|
|
auto constraints = collectConstraints(NotNull(cgb.rootScope));
|
2022-06-03 23:15:45 +01:00
|
|
|
|
|
|
|
REQUIRE(2 == constraints.size());
|
|
|
|
|
|
|
|
ToStringOptions opts;
|
2022-06-17 02:05:14 +01:00
|
|
|
CHECK("*blocked-1* ~ gen (a) -> (b...)" == toString(*constraints[0], opts));
|
|
|
|
CHECK("a <: b..." == toString(*constraints[1], opts));
|
2022-06-03 23:15:45 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE_FIXTURE(ConstraintGraphBuilderFixture, "recursive_function")
|
|
|
|
{
|
|
|
|
AstStatBlock* block = parse(R"(
|
|
|
|
local function f(a)
|
|
|
|
return f(a)
|
|
|
|
end
|
|
|
|
)");
|
|
|
|
|
|
|
|
cgb.visit(block);
|
2022-07-01 00:52:43 +01:00
|
|
|
auto constraints = collectConstraints(NotNull(cgb.rootScope));
|
2022-06-03 23:15:45 +01:00
|
|
|
|
2022-09-02 00:14:03 +01:00
|
|
|
REQUIRE(3 == constraints.size());
|
2022-06-03 23:15:45 +01:00
|
|
|
|
|
|
|
ToStringOptions opts;
|
2022-06-17 02:05:14 +01:00
|
|
|
CHECK("*blocked-1* ~ gen (a) -> (b...)" == toString(*constraints[0], opts));
|
2022-09-02 00:14:03 +01:00
|
|
|
CHECK("call (a) -> (b...) with { result = *blocked-tp-1* }" == toString(*constraints[1], opts));
|
|
|
|
CHECK("*blocked-tp-1* <: b..." == toString(*constraints[2], opts));
|
2022-06-03 23:15:45 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
TEST_SUITE_END();
|