Skip to content

Commit 32f7f00

Browse files
[memprof] Use gtest matchers at more places (#119050)
These gtest matchers reduce the number of times we mention the variables under examined.
1 parent 00090ac commit 32f7f00

File tree

2 files changed

+22
-32
lines changed

2 files changed

+22
-32
lines changed

llvm/unittests/ProfileData/InstrProfTest.cpp

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626

2727
using namespace llvm;
2828
using ::llvm::memprof::LineLocation;
29+
using ::testing::ElementsAre;
2930
using ::testing::EndsWith;
3031
using ::testing::IsSubsetOf;
3132
using ::testing::Pair;
@@ -564,19 +565,16 @@ TEST_F(InstrProfTest, test_caller_callee_pairs) {
564565

565566
auto It = Pairs.find(0x123);
566567
ASSERT_NE(It, Pairs.end());
567-
ASSERT_THAT(It->second, SizeIs(2));
568-
EXPECT_THAT(It->second[0], Pair(LineLocation(1, 2), 0x234U));
569-
EXPECT_THAT(It->second[1], Pair(LineLocation(5, 6), 0x345U));
568+
EXPECT_THAT(It->second, ElementsAre(Pair(LineLocation(1, 2), 0x234U),
569+
Pair(LineLocation(5, 6), 0x345U)));
570570

571571
It = Pairs.find(0x234);
572572
ASSERT_NE(It, Pairs.end());
573-
ASSERT_THAT(It->second, SizeIs(1));
574-
EXPECT_THAT(It->second[0], Pair(LineLocation(3, 4), 0U));
573+
EXPECT_THAT(It->second, ElementsAre(Pair(LineLocation(3, 4), 0U)));
575574

576575
It = Pairs.find(0x345);
577576
ASSERT_NE(It, Pairs.end());
578-
ASSERT_THAT(It->second, SizeIs(1));
579-
EXPECT_THAT(It->second[0], Pair(LineLocation(7, 8), 0U));
577+
EXPECT_THAT(It->second, ElementsAre(Pair(LineLocation(7, 8), 0U)));
580578
}
581579

582580
TEST_F(InstrProfTest, test_memprof_getrecord_error) {

llvm/unittests/ProfileData/MemProfTest.cpp

Lines changed: 17 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -222,31 +222,28 @@ TEST(MemProf, FillsValue) {
222222
EXPECT_THAT(Bar.AllocSites[0].CallStack[3],
223223
FrameContains("abc", 5U, 30U, false));
224224

225-
ASSERT_THAT(Bar.CallSites, SizeIs(1));
226-
ASSERT_THAT(Bar.CallSites[0], SizeIs(2));
227-
EXPECT_THAT(Bar.CallSites[0][0], FrameContains("foo", 5U, 30U, true));
228-
EXPECT_THAT(Bar.CallSites[0][1], FrameContains("bar", 51U, 20U, false));
225+
EXPECT_THAT(Bar.CallSites,
226+
ElementsAre(ElementsAre(FrameContains("foo", 5U, 30U, true),
227+
FrameContains("bar", 51U, 20U, false))));
229228

230229
// Check the memprof record for xyz.
231230
const llvm::GlobalValue::GUID XyzId = IndexedMemProfRecord::getGUID("xyz");
232231
ASSERT_TRUE(Records.contains(XyzId));
233232
const MemProfRecord &Xyz = Records[XyzId];
234-
ASSERT_THAT(Xyz.CallSites, SizeIs(1));
235-
ASSERT_THAT(Xyz.CallSites[0], SizeIs(2));
236233
// Expect the entire frame even though in practice we only need the first
237234
// entry here.
238-
EXPECT_THAT(Xyz.CallSites[0][0], FrameContains("xyz", 5U, 30U, true));
239-
EXPECT_THAT(Xyz.CallSites[0][1], FrameContains("abc", 5U, 30U, false));
235+
EXPECT_THAT(Xyz.CallSites,
236+
ElementsAre(ElementsAre(FrameContains("xyz", 5U, 30U, true),
237+
FrameContains("abc", 5U, 30U, false))));
240238

241239
// Check the memprof record for abc.
242240
const llvm::GlobalValue::GUID AbcId = IndexedMemProfRecord::getGUID("abc");
243241
ASSERT_TRUE(Records.contains(AbcId));
244242
const MemProfRecord &Abc = Records[AbcId];
245243
EXPECT_TRUE(Abc.AllocSites.empty());
246-
ASSERT_THAT(Abc.CallSites, SizeIs(1));
247-
ASSERT_THAT(Abc.CallSites[0], SizeIs(2));
248-
EXPECT_THAT(Abc.CallSites[0][0], FrameContains("xyz", 5U, 30U, true));
249-
EXPECT_THAT(Abc.CallSites[0][1], FrameContains("abc", 5U, 30U, false));
244+
EXPECT_THAT(Abc.CallSites,
245+
ElementsAre(ElementsAre(FrameContains("xyz", 5U, 30U, true),
246+
FrameContains("abc", 5U, 30U, false))));
250247
}
251248

252249
TEST(MemProf, PortableWrapper) {
@@ -420,9 +417,8 @@ TEST(MemProf, SymbolizationFilter) {
420417

421418
ASSERT_THAT(Records, SizeIs(1));
422419
ASSERT_THAT(Records[0].AllocSites, SizeIs(1));
423-
ASSERT_THAT(Records[0].AllocSites[0].CallStack, SizeIs(1));
424-
EXPECT_THAT(Records[0].AllocSites[0].CallStack[0],
425-
FrameContains("foo", 5U, 30U, false));
420+
EXPECT_THAT(Records[0].AllocSites[0].CallStack,
421+
ElementsAre(FrameContains("foo", 5U, 30U, false)));
426422
}
427423

428424
TEST(MemProf, BaseMemProfReader) {
@@ -453,11 +449,9 @@ TEST(MemProf, BaseMemProfReader) {
453449

454450
ASSERT_THAT(Records, SizeIs(1));
455451
ASSERT_THAT(Records[0].AllocSites, SizeIs(1));
456-
ASSERT_THAT(Records[0].AllocSites[0].CallStack, SizeIs(2));
457-
EXPECT_THAT(Records[0].AllocSites[0].CallStack[0],
458-
FrameContains("foo", 20U, 5U, true));
459-
EXPECT_THAT(Records[0].AllocSites[0].CallStack[1],
460-
FrameContains("bar", 10U, 2U, false));
452+
EXPECT_THAT(Records[0].AllocSites[0].CallStack,
453+
ElementsAre(FrameContains("foo", 20U, 5U, true),
454+
FrameContains("bar", 10U, 2U, false)));
461455
}
462456

463457
TEST(MemProf, BaseMemProfReaderWithCSIdMap) {
@@ -490,11 +484,9 @@ TEST(MemProf, BaseMemProfReaderWithCSIdMap) {
490484

491485
ASSERT_THAT(Records, SizeIs(1));
492486
ASSERT_THAT(Records[0].AllocSites, SizeIs(1));
493-
ASSERT_THAT(Records[0].AllocSites[0].CallStack, SizeIs(2));
494-
EXPECT_THAT(Records[0].AllocSites[0].CallStack[0],
495-
FrameContains("foo", 20U, 5U, true));
496-
EXPECT_THAT(Records[0].AllocSites[0].CallStack[1],
497-
FrameContains("bar", 10U, 2U, false));
487+
EXPECT_THAT(Records[0].AllocSites[0].CallStack,
488+
ElementsAre(FrameContains("foo", 20U, 5U, true),
489+
FrameContains("bar", 10U, 2U, false)));
498490
}
499491

500492
TEST(MemProf, IndexedMemProfRecordToMemProfRecord) {

0 commit comments

Comments
 (0)