@@ -222,31 +222,28 @@ TEST(MemProf, FillsValue) {
222
222
EXPECT_THAT (Bar.AllocSites [0 ].CallStack [3 ],
223
223
FrameContains (" abc" , 5U , 30U , false ));
224
224
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 ))));
229
228
230
229
// Check the memprof record for xyz.
231
230
const llvm::GlobalValue::GUID XyzId = IndexedMemProfRecord::getGUID (" xyz" );
232
231
ASSERT_TRUE (Records.contains (XyzId));
233
232
const MemProfRecord &Xyz = Records[XyzId];
234
- ASSERT_THAT (Xyz.CallSites , SizeIs (1 ));
235
- ASSERT_THAT (Xyz.CallSites [0 ], SizeIs (2 ));
236
233
// Expect the entire frame even though in practice we only need the first
237
234
// 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 ))));
240
238
241
239
// Check the memprof record for abc.
242
240
const llvm::GlobalValue::GUID AbcId = IndexedMemProfRecord::getGUID (" abc" );
243
241
ASSERT_TRUE (Records.contains (AbcId));
244
242
const MemProfRecord &Abc = Records[AbcId];
245
243
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 ))));
250
247
}
251
248
252
249
TEST (MemProf, PortableWrapper) {
@@ -420,9 +417,8 @@ TEST(MemProf, SymbolizationFilter) {
420
417
421
418
ASSERT_THAT (Records, SizeIs (1 ));
422
419
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 )));
426
422
}
427
423
428
424
TEST (MemProf, BaseMemProfReader) {
@@ -453,11 +449,9 @@ TEST(MemProf, BaseMemProfReader) {
453
449
454
450
ASSERT_THAT (Records, SizeIs (1 ));
455
451
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 )));
461
455
}
462
456
463
457
TEST (MemProf, BaseMemProfReaderWithCSIdMap) {
@@ -490,11 +484,9 @@ TEST(MemProf, BaseMemProfReaderWithCSIdMap) {
490
484
491
485
ASSERT_THAT (Records, SizeIs (1 ));
492
486
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 )));
498
490
}
499
491
500
492
TEST (MemProf, IndexedMemProfRecordToMemProfRecord) {
0 commit comments