diff options
Diffstat (limited to 'llvm/unittests/ProfileData/SampleProfTest.cpp')
| -rw-r--r-- | llvm/unittests/ProfileData/SampleProfTest.cpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/llvm/unittests/ProfileData/SampleProfTest.cpp b/llvm/unittests/ProfileData/SampleProfTest.cpp index 764bded2f03..412f36f5c9c 100644 --- a/llvm/unittests/ProfileData/SampleProfTest.cpp +++ b/llvm/unittests/ProfileData/SampleProfTest.cpp @@ -77,6 +77,9 @@ struct SampleProfTest : ::testing::Test { BarSamples.addTotalSamples(20301); BarSamples.addHeadSamples(1437); BarSamples.addBodySamples(1, 0, 1437); + BarSamples.addCalledTargetSamples(1, 0, "_M_construct<char *>", 1000); + BarSamples.addCalledTargetSamples( + 1, 0, "string_view<std::allocator<char> >", 437); StringMap<FunctionSamples> Profiles; Profiles[FooName] = std::move(FooSamples); @@ -104,6 +107,11 @@ struct SampleProfTest : ::testing::Test { FunctionSamples &ReadBarSamples = ReadProfiles[BarName]; ASSERT_EQ(20301u, ReadBarSamples.getTotalSamples()); ASSERT_EQ(1437u, ReadBarSamples.getHeadSamples()); + ErrorOr<SampleRecord::CallTargetMap> CTMap = + ReadBarSamples.findCallTargetMapAt(1, 0); + ASSERT_FALSE(CTMap.getError()); + ASSERT_EQ(1000u, CTMap.get()["_M_construct<char *>"]); + ASSERT_EQ(437u, CTMap.get()["string_view<std::allocator<char> >"]); auto VerifySummary = [](ProfileSummary &Summary) mutable { ASSERT_EQ(ProfileSummary::PSK_Sample, Summary.getKind()); |
