@@ -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
252249TEST (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
428424TEST (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
463457TEST (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
500492TEST (MemProf, IndexedMemProfRecordToMemProfRecord) {
0 commit comments