[ios] TabCollectionItem

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

 // Copyright 2017 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #import "ios/clean/chrome/browser/ui/tab_grid/tab_grid_mediator.h"
 
 #include "base/memory/ptr_util.h"
 #include "ios/chrome/browser/web_state_list/fake_web_state_list_delegate.h"
 #include "ios/chrome/browser/web_state_list/web_state_list.h"
 #import "ios/chrome/browser/web_state_list/web_state_list_observer_bridge.h"
@@ skipping 23 matching lines @@
   void SetUpEmptyWebStateList() {
     web_state_list_ = base::MakeUnique<WebStateList>(&web_state_list_delegate_);
   }
 
   void InsertWebStateAt(int index) {
     auto web_state = base::MakeUnique<web::TestWebState>();
     web_state_list_->InsertWebState(index, std::move(web_state));
   }
 
   void SetConsumer() {
-    consumer_ = [OCMockObject mockForProtocol:@protocol(TabGridConsumer)];
+    consumer_ = OCMProtocolMock(@protocol(TabGridConsumer));
     mediator_.consumer = consumer_;
   }
 
   TabGridMediator* mediator_;
   std::unique_ptr<WebStateList> web_state_list_;
   FakeWebStateListDelegate web_state_list_delegate_;
   id consumer_;
 };
 
 // Tests that the noTabsOverlay is removed when a web state is inserted when
 // 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) {
+  SetConsumer();
   InsertWebStateAt(0);
-  SetConsumer();
-  [[consumer_ expect] deleteItemAtIndex:0];
-  [[consumer_ expect] addNoTabsOverlay];
   web_state_list_->CloseWebStateAt(0);
-  EXPECT_OCMOCK_VERIFY(consumer_);
+  [[consumer_ verify] addNoTabsOverlay];
 }
 
 }  // namespace