resolver: Add test for sem::Function::CallSites() Fixed: tint:835 Change-Id: I22bc5d478061afbc279a0e881e4c1eeb0183c68a Reviewed-on: https://dawn-review.googlesource.com/c/tint/+/67645 Kokoro: Kokoro <noreply+kokoro@google.com> Reviewed-by: Ben Clayton <bclayton@google.com>
diff --git a/src/resolver/resolver_test.cc b/src/resolver/resolver_test.cc index b460267..198109f 100644 --- a/src/resolver/resolver_test.cc +++ b/src/resolver/resolver_test.cc
@@ -930,6 +930,30 @@ EXPECT_TRUE(func_sem->ReturnType()->Is<sem::F32>()); } +TEST_F(ResolverTest, Function_CallSites) { + auto* foo = Func("foo", ast::VariableList{}, ty.void_(), {}); + + auto* call_1 = Call("foo"); + auto* call_2 = Call("foo"); + auto* bar = Func("bar", ast::VariableList{}, ty.void_(), + { + WrapInStatement(call_1), + WrapInStatement(call_2), + }); + + EXPECT_TRUE(r()->Resolve()) << r()->error(); + + auto* foo_sem = Sem().Get(foo); + ASSERT_NE(foo_sem, nullptr); + ASSERT_EQ(foo_sem->CallSites().size(), 2u); + EXPECT_EQ(foo_sem->CallSites()[0], call_1); + EXPECT_EQ(foo_sem->CallSites()[1], call_2); + + auto* bar_sem = Sem().Get(bar); + ASSERT_NE(bar_sem, nullptr); + EXPECT_EQ(bar_sem->CallSites().size(), 0u); +} + TEST_F(ResolverTest, Function_WorkgroupSize_NotSet) { // [[stage(compute), workgroup_size(1)]] // fn main() {}