2021-10-29 21:25:12 +01:00
|
|
|
// This file is part of the Luau programming language and is licensed under MIT License; see LICENSE.txt for details
|
2021-11-05 02:07:18 +00:00
|
|
|
#include "Luau/Scope.h"
|
2021-10-29 21:25:12 +01:00
|
|
|
#include "Luau/TypeInfer.h"
|
|
|
|
#include "Luau/TypeVar.h"
|
|
|
|
|
|
|
|
#include "Fixture.h"
|
|
|
|
|
|
|
|
#include "doctest.h"
|
|
|
|
|
2021-11-05 02:42:00 +00:00
|
|
|
LUAU_FASTFLAG(LuauQuantifyInPlace2);
|
|
|
|
|
2021-10-29 21:25:12 +01:00
|
|
|
using namespace Luau;
|
|
|
|
|
2022-01-06 22:10:07 +00:00
|
|
|
LUAU_FASTFLAG(LuauUseCommittingTxnLog)
|
|
|
|
|
2021-10-29 21:25:12 +01:00
|
|
|
struct TryUnifyFixture : Fixture
|
|
|
|
{
|
|
|
|
TypeArena arena;
|
|
|
|
ScopePtr globalScope{new Scope{arena.addTypePack({TypeId{}})}};
|
|
|
|
InternalErrorReporter iceHandler;
|
2021-11-05 02:42:00 +00:00
|
|
|
UnifierSharedState unifierState{&iceHandler};
|
2022-02-24 23:15:41 +00:00
|
|
|
Unifier state{&arena, Mode::Strict, Location{}, Variance::Covariant, unifierState};
|
2021-10-29 21:25:12 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
TEST_SUITE_BEGIN("TryUnifyTests");
|
|
|
|
|
|
|
|
TEST_CASE_FIXTURE(TryUnifyFixture, "primitives_unify")
|
|
|
|
{
|
|
|
|
TypeVar numberOne{TypeVariant{PrimitiveTypeVar{PrimitiveTypeVar::Number}}};
|
|
|
|
TypeVar numberTwo = numberOne;
|
|
|
|
|
2022-01-06 22:10:07 +00:00
|
|
|
state.tryUnify(&numberTwo, &numberOne);
|
2021-10-29 21:25:12 +01:00
|
|
|
|
|
|
|
CHECK(state.errors.empty());
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE_FIXTURE(TryUnifyFixture, "compatible_functions_are_unified")
|
|
|
|
{
|
|
|
|
TypeVar functionOne{
|
|
|
|
TypeVariant{FunctionTypeVar(arena.addTypePack({arena.freshType(globalScope->level)}), arena.addTypePack({typeChecker.numberType}))}};
|
|
|
|
|
|
|
|
TypeVar functionTwo{TypeVariant{
|
|
|
|
FunctionTypeVar(arena.addTypePack({arena.freshType(globalScope->level)}), arena.addTypePack({arena.freshType(globalScope->level)}))}};
|
|
|
|
|
2022-01-06 22:10:07 +00:00
|
|
|
state.tryUnify(&functionTwo, &functionOne);
|
2021-10-29 21:25:12 +01:00
|
|
|
CHECK(state.errors.empty());
|
|
|
|
|
2022-01-06 22:10:07 +00:00
|
|
|
if (FFlag::LuauUseCommittingTxnLog)
|
|
|
|
state.log.commit();
|
|
|
|
|
2021-10-29 21:25:12 +01:00
|
|
|
CHECK_EQ(functionOne, functionTwo);
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE_FIXTURE(TryUnifyFixture, "incompatible_functions_are_preserved")
|
|
|
|
{
|
|
|
|
TypePackVar argPackOne{TypePack{{arena.freshType(globalScope->level)}, std::nullopt}};
|
|
|
|
TypeVar functionOne{
|
|
|
|
TypeVariant{FunctionTypeVar(arena.addTypePack({arena.freshType(globalScope->level)}), arena.addTypePack({typeChecker.numberType}))}};
|
|
|
|
|
|
|
|
TypeVar functionOneSaved = functionOne;
|
|
|
|
|
|
|
|
TypePackVar argPackTwo{TypePack{{arena.freshType(globalScope->level)}, std::nullopt}};
|
|
|
|
TypeVar functionTwo{
|
|
|
|
TypeVariant{FunctionTypeVar(arena.addTypePack({arena.freshType(globalScope->level)}), arena.addTypePack({typeChecker.stringType}))}};
|
|
|
|
|
|
|
|
TypeVar functionTwoSaved = functionTwo;
|
|
|
|
|
2022-01-06 22:10:07 +00:00
|
|
|
state.tryUnify(&functionTwo, &functionOne);
|
2021-10-29 21:25:12 +01:00
|
|
|
CHECK(!state.errors.empty());
|
|
|
|
|
|
|
|
CHECK_EQ(functionOne, functionOneSaved);
|
|
|
|
CHECK_EQ(functionTwo, functionTwoSaved);
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE_FIXTURE(TryUnifyFixture, "tables_can_be_unified")
|
|
|
|
{
|
|
|
|
TypeVar tableOne{TypeVariant{
|
|
|
|
TableTypeVar{{{"foo", {arena.freshType(globalScope->level)}}}, std::nullopt, globalScope->level, TableState::Unsealed},
|
|
|
|
}};
|
|
|
|
|
|
|
|
TypeVar tableTwo{TypeVariant{
|
|
|
|
TableTypeVar{{{"foo", {arena.freshType(globalScope->level)}}}, std::nullopt, globalScope->level, TableState::Unsealed},
|
|
|
|
}};
|
|
|
|
|
|
|
|
CHECK_NE(*getMutable<TableTypeVar>(&tableOne)->props["foo"].type, *getMutable<TableTypeVar>(&tableTwo)->props["foo"].type);
|
|
|
|
|
2022-01-06 22:10:07 +00:00
|
|
|
state.tryUnify(&tableTwo, &tableOne);
|
2021-10-29 21:25:12 +01:00
|
|
|
|
|
|
|
CHECK(state.errors.empty());
|
|
|
|
|
2022-01-06 22:10:07 +00:00
|
|
|
if (FFlag::LuauUseCommittingTxnLog)
|
|
|
|
state.log.commit();
|
|
|
|
|
2021-10-29 21:25:12 +01:00
|
|
|
CHECK_EQ(*getMutable<TableTypeVar>(&tableOne)->props["foo"].type, *getMutable<TableTypeVar>(&tableTwo)->props["foo"].type);
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE_FIXTURE(TryUnifyFixture, "incompatible_tables_are_preserved")
|
|
|
|
{
|
|
|
|
TypeVar tableOne{TypeVariant{
|
|
|
|
TableTypeVar{{{"foo", {arena.freshType(globalScope->level)}}, {"bar", {typeChecker.numberType}}}, std::nullopt, globalScope->level,
|
|
|
|
TableState::Unsealed},
|
|
|
|
}};
|
|
|
|
|
|
|
|
TypeVar tableTwo{TypeVariant{
|
|
|
|
TableTypeVar{{{"foo", {arena.freshType(globalScope->level)}}, {"bar", {typeChecker.stringType}}}, std::nullopt, globalScope->level,
|
|
|
|
TableState::Unsealed},
|
|
|
|
}};
|
|
|
|
|
|
|
|
CHECK_NE(*getMutable<TableTypeVar>(&tableOne)->props["foo"].type, *getMutable<TableTypeVar>(&tableTwo)->props["foo"].type);
|
|
|
|
|
2022-01-06 22:10:07 +00:00
|
|
|
state.tryUnify(&tableTwo, &tableOne);
|
2021-10-29 21:25:12 +01:00
|
|
|
|
|
|
|
CHECK_EQ(1, state.errors.size());
|
|
|
|
|
2022-01-06 22:10:07 +00:00
|
|
|
if (!FFlag::LuauUseCommittingTxnLog)
|
|
|
|
state.DEPRECATED_log.rollback();
|
2021-10-29 21:25:12 +01:00
|
|
|
|
|
|
|
CHECK_NE(*getMutable<TableTypeVar>(&tableOne)->props["foo"].type, *getMutable<TableTypeVar>(&tableTwo)->props["foo"].type);
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE_FIXTURE(TryUnifyFixture, "members_of_failed_typepack_unification_are_unified_with_errorType")
|
|
|
|
{
|
|
|
|
CheckResult result = check(R"(
|
|
|
|
function f(arg: number) end
|
|
|
|
local a
|
|
|
|
local b
|
|
|
|
f(a, b)
|
|
|
|
)");
|
|
|
|
|
|
|
|
LUAU_REQUIRE_ERROR_COUNT(1, result);
|
|
|
|
|
2021-11-18 22:21:07 +00:00
|
|
|
CHECK_EQ("a", toString(requireType("a")));
|
|
|
|
CHECK_EQ("*unknown*", toString(requireType("b")));
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE_FIXTURE(TryUnifyFixture, "result_of_failed_typepack_unification_is_constrained")
|
|
|
|
{
|
|
|
|
ScopedFastFlag sff{"LuauErrorRecoveryType", true};
|
|
|
|
|
|
|
|
CheckResult result = check(R"(
|
|
|
|
function f(arg: number) return arg end
|
|
|
|
local a
|
|
|
|
local b
|
|
|
|
local c = f(a, b)
|
|
|
|
)");
|
|
|
|
|
|
|
|
LUAU_REQUIRE_ERROR_COUNT(1, result);
|
2021-10-29 21:25:12 +01:00
|
|
|
|
2021-11-18 22:21:07 +00:00
|
|
|
CHECK_EQ("a", toString(requireType("a")));
|
|
|
|
CHECK_EQ("*unknown*", toString(requireType("b")));
|
|
|
|
CHECK_EQ("number", toString(requireType("c")));
|
2021-10-29 21:25:12 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE_FIXTURE(TryUnifyFixture, "typepack_unification_should_trim_free_tails")
|
|
|
|
{
|
|
|
|
CheckResult result = check(R"(
|
|
|
|
--!strict
|
|
|
|
local function f(v: number)
|
|
|
|
if v % 2 == 0 then
|
|
|
|
return true
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
return function()
|
|
|
|
return (f(1))
|
|
|
|
end
|
|
|
|
)");
|
|
|
|
|
|
|
|
LUAU_REQUIRE_ERROR_COUNT(1, result);
|
2021-11-05 02:42:00 +00:00
|
|
|
if (FFlag::LuauQuantifyInPlace2)
|
|
|
|
CHECK_EQ("(number) -> boolean", toString(requireType("f")));
|
|
|
|
else
|
|
|
|
CHECK_EQ("(number) -> (boolean)", toString(requireType("f")));
|
2021-10-29 21:25:12 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE_FIXTURE(TryUnifyFixture, "variadic_type_pack_unification")
|
|
|
|
{
|
|
|
|
TypePackVar testPack{TypePack{{typeChecker.numberType, typeChecker.stringType}, std::nullopt}};
|
|
|
|
TypePackVar variadicPack{VariadicTypePack{typeChecker.numberType}};
|
|
|
|
|
2022-01-06 22:10:07 +00:00
|
|
|
state.tryUnify(&testPack, &variadicPack);
|
2021-10-29 21:25:12 +01:00
|
|
|
CHECK(!state.errors.empty());
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE_FIXTURE(TryUnifyFixture, "variadic_tails_respect_progress")
|
|
|
|
{
|
|
|
|
TypePackVar variadicPack{VariadicTypePack{typeChecker.booleanType}};
|
|
|
|
TypePackVar a{TypePack{{typeChecker.numberType, typeChecker.stringType, typeChecker.booleanType, typeChecker.booleanType}}};
|
|
|
|
TypePackVar b{TypePack{{typeChecker.numberType, typeChecker.stringType}, &variadicPack}};
|
|
|
|
|
2022-01-06 22:10:07 +00:00
|
|
|
state.tryUnify(&b, &a);
|
2021-10-29 21:25:12 +01:00
|
|
|
CHECK(state.errors.empty());
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE_FIXTURE(TryUnifyFixture, "variadics_should_use_reversed_properly")
|
|
|
|
{
|
|
|
|
CheckResult result = check(R"(
|
|
|
|
--!strict
|
|
|
|
local function f<T>(...: T): ...T
|
|
|
|
return ...
|
|
|
|
end
|
|
|
|
|
|
|
|
local x: string = f(1)
|
|
|
|
)");
|
|
|
|
|
|
|
|
LUAU_REQUIRE_ERROR_COUNT(1, result);
|
|
|
|
TypeMismatch* tm = get<TypeMismatch>(result.errors[0]);
|
|
|
|
REQUIRE(tm);
|
|
|
|
CHECK_EQ(toString(tm->givenType), "number");
|
|
|
|
CHECK_EQ(toString(tm->wantedType), "string");
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE_FIXTURE(TryUnifyFixture, "cli_41095_concat_log_in_sealed_table_unification")
|
|
|
|
{
|
|
|
|
CheckResult result = check(R"(
|
|
|
|
--!strict
|
|
|
|
table.insert()
|
|
|
|
)");
|
|
|
|
|
|
|
|
LUAU_REQUIRE_ERROR_COUNT(2, result);
|
|
|
|
CHECK_EQ(toString(result.errors[0]), "No overload for function accepts 0 arguments.");
|
|
|
|
CHECK_EQ(toString(result.errors[1]), "Available overloads: ({a}, a) -> (); and ({a}, number, a) -> ()");
|
|
|
|
}
|
|
|
|
|
2022-01-06 22:10:07 +00:00
|
|
|
TEST_CASE("undo_new_prop_on_unsealed_table")
|
2021-12-10 21:17:10 +00:00
|
|
|
{
|
|
|
|
ScopedFastFlag flags[] = {
|
|
|
|
{"LuauTableSubtypingVariance2", true},
|
2022-01-06 22:10:07 +00:00
|
|
|
// This test makes no sense with a committing TxnLog.
|
|
|
|
{"LuauUseCommittingTxnLog", false},
|
2021-12-10 21:17:10 +00:00
|
|
|
};
|
|
|
|
// I am not sure how to make this happen in Luau code.
|
|
|
|
|
2022-01-06 22:10:07 +00:00
|
|
|
TryUnifyFixture fix;
|
|
|
|
|
|
|
|
TypeId unsealedTable = fix.arena.addType(TableTypeVar{TableState::Unsealed, TypeLevel{}});
|
|
|
|
TypeId sealedTable =
|
|
|
|
fix.arena.addType(TableTypeVar{{{"prop", Property{getSingletonTypes().numberType}}}, std::nullopt, TypeLevel{}, TableState::Sealed});
|
2021-12-10 21:17:10 +00:00
|
|
|
|
|
|
|
const TableTypeVar* ttv = get<TableTypeVar>(unsealedTable);
|
|
|
|
REQUIRE(ttv);
|
|
|
|
|
2022-01-06 22:10:07 +00:00
|
|
|
fix.state.tryUnify(sealedTable, unsealedTable);
|
2021-12-10 21:17:10 +00:00
|
|
|
|
|
|
|
// To be honest, it's really quite spooky here that we're amending an unsealed table in this case.
|
|
|
|
CHECK(!ttv->props.empty());
|
|
|
|
|
2022-01-06 22:10:07 +00:00
|
|
|
fix.state.DEPRECATED_log.rollback();
|
2021-12-10 21:17:10 +00:00
|
|
|
|
|
|
|
CHECK(ttv->props.empty());
|
|
|
|
}
|
|
|
|
|
2022-01-06 22:10:07 +00:00
|
|
|
TEST_CASE_FIXTURE(TryUnifyFixture, "free_tail_is_grown_properly")
|
|
|
|
{
|
|
|
|
TypePackId threeNumbers = arena.addTypePack(TypePack{{typeChecker.numberType, typeChecker.numberType, typeChecker.numberType}, std::nullopt});
|
|
|
|
TypePackId numberAndFreeTail = arena.addTypePack(TypePack{{typeChecker.numberType}, arena.addTypePack(TypePackVar{FreeTypePack{TypeLevel{}}})});
|
|
|
|
|
|
|
|
ErrorVec unifyErrors = state.canUnify(numberAndFreeTail, threeNumbers);
|
|
|
|
CHECK(unifyErrors.size() == 0);
|
|
|
|
}
|
|
|
|
|
2022-02-03 23:09:37 +00:00
|
|
|
TEST_CASE_FIXTURE(TryUnifyFixture, "recursive_metatable_getmatchtag")
|
|
|
|
{
|
|
|
|
TypeVar redirect{FreeTypeVar{TypeLevel{}}};
|
|
|
|
TypeVar table{TableTypeVar{}};
|
|
|
|
TypeVar metatable{MetatableTypeVar{&redirect, &table}};
|
|
|
|
redirect = BoundTypeVar{&metatable}; // Now we have a metatable that is recursive on the table type
|
|
|
|
TypeVar variant{UnionTypeVar{{&metatable, typeChecker.numberType}}};
|
|
|
|
|
|
|
|
state.tryUnify(&metatable, &variant);
|
|
|
|
}
|
|
|
|
|
2022-02-11 18:43:14 +00:00
|
|
|
TEST_CASE_FIXTURE(TryUnifyFixture, "cli_50320_follow_in_any_unification")
|
|
|
|
{
|
|
|
|
ScopedFastFlag sffs[] = {
|
|
|
|
{"LuauUseCommittingTxnLog", true},
|
|
|
|
{"LuauFollowWithCommittingTxnLogInAnyUnification", true},
|
|
|
|
};
|
|
|
|
|
|
|
|
TypePackVar free{FreeTypePack{TypeLevel{}}};
|
|
|
|
TypePackVar target{TypePack{}};
|
|
|
|
|
|
|
|
TypeVar func{FunctionTypeVar{&free, &free}};
|
|
|
|
|
|
|
|
state.tryUnify(&free, &target);
|
|
|
|
// Shouldn't assert or error.
|
|
|
|
state.tryUnify(&func, typeChecker.anyType);
|
|
|
|
}
|
|
|
|
|
2021-10-29 21:25:12 +01:00
|
|
|
TEST_SUITE_END();
|