[ios] TabCollectionItem

ios/clean/chrome/browser/ui/tab_grid/tab_grid_mediator_unittest.mm

Index: ios/clean/chrome/browser/ui/tab_grid/tab_grid_mediator_unittest.mm
diff --git a/ios/clean/chrome/browser/ui/tab_grid/tab_grid_mediator_unittest.mm b/ios/clean/chrome/browser/ui/tab_grid/tab_grid_mediator_unittest.mm
index 792a6cac41a7139625f74fd48ff95402a0aae516..2f514f0e290a05c2eedf87648e5b3219818ced5c 100644
--- a/ios/clean/chrome/browser/ui/tab_grid/tab_grid_mediator_unittest.mm
+++ b/ios/clean/chrome/browser/ui/tab_grid/tab_grid_mediator_unittest.mm
@@ -41,7 +41,7 @@ class TabGridMediatorTest : public PlatformTest {
   }
 
   void SetConsumer() {
-    consumer_ = [OCMockObject mockForProtocol:@protocol(TabGridConsumer)];
+    consumer_ = OCMProtocolMock(@protocol(TabGridConsumer));
     mediator_.consumer = consumer_;
   }
 
@@ -55,20 +55,16 @@ class TabGridMediatorTest : public PlatformTest {
 // the list is empty.
 TEST_F(TabGridMediatorTest, TestRemoveNoTabsOverlay) {
   SetConsumer();
-  [[consumer_ expect] insertItemAtIndex:0];
-  [[consumer_ expect] removeNoTabsOverlay];
   InsertWebStateAt(0);
-  EXPECT_OCMOCK_VERIFY(consumer_);
+  [[consumer_ verify] removeNoTabsOverlay];
 }
 
 // Tests that the noTabsOverlay is added when the web state list becomes empty.
 TEST_F(TabGridMediatorTest, TestAddNoTabsOverlay) {
-  InsertWebStateAt(0);
   SetConsumer();
-  [[consumer_ expect] deleteItemAtIndex:0];
-  [[consumer_ expect] addNoTabsOverlay];
+  InsertWebStateAt(0);
   web_state_list_->CloseWebStateAt(0);
-  EXPECT_OCMOCK_VERIFY(consumer_);
+  [[consumer_ verify] addNoTabsOverlay];
 }
 
 }  // namespace