[ios] Simplifies the Find In Page JS unittest.

ios/chrome/browser/find_in_page/find_in_page_js_unittest.mm

 // Copyright 2012 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 <UIKit/UIKit.h>
 
+#include "base/mac/foundation_util.h"
 #include "base/mac/scoped_nsobject.h"
 #include "base/strings/sys_string_conversions.h"
-#import "ios/chrome/browser/find_in_page/find_in_page_controller.h"
+#import "ios/chrome/browser/find_in_page/find_in_page_model.h"
+#import "ios/chrome/browser/find_in_page/js_findinpage_manager.h"
 #import "ios/chrome/browser/web/chrome_web_test.h"
+#import "ios/web/public/web_state/crw_web_view_proxy.h"
+#import "ios/web/public/web_state/js/crw_js_injection_receiver.h"
 #import "ios/web/public/web_state/web_state.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/gtest_mac.h"
-#import "third_party/ocmock/OCMock/OCMock.h"
 
-// Unit tests for the models/resources/find_in_page.js JavaScript file.
+// Unit tests for the find_in_page.js JavaScript file.
 
 namespace {
 
 // JavaScript invocation to search for 'foo' (for 1000 milliseconds).
 NSString* kJavaScriptToSearchForFoo =
     @"__gCrWeb.findInPage.highlightWord('foo', false, 1000)";
 
 // Other JavaScript functions invoked by the tests.
 NSString* kJavaScriptIncrementIndex = @"__gCrWeb.findInPage.incrementIndex()";
 NSString* kJavaScriptDecrementIndex = @"__gCrWeb.findInPage.decrementIndex()";
@@ skipping 29 matching lines @@
                                     "  <span style='display:none'>foo</span>"
                                     "  <span style='display:none'>foo</span>"
                                     "</body></html>";
 
 // Test fixture to test Find In Page JS.
 class FindInPageJsTest : public ChromeWebTest {
  public:
   // Loads the given HTML, then loads the |findInPage| JavaScript.
   void LoadHtml(NSString* html) {
     ChromeWebTest::LoadHtml(html);
-    [findInPageController_ initFindInPage];
+
+    // Inject and initialize the find in page javascript.
+    [findInPageJsManager_ inject];
+    CGRect frame = [web_state()->GetWebViewProxy() bounds];
+    [findInPageJsManager_ setWidth:frame.size.width height:frame.size.height];
   }
 
   // Runs the given JavaScript and asserts that the result matches the given
   // |expected_value|.
   void AssertJavaScriptValue(NSString* script, int expected_value) {
     id result = ExecuteJavaScript(script);
     EXPECT_TRUE(result) << " in script: " << base::SysNSStringToUTF8(script);
     EXPECT_EQ(expected_value, [result intValue])
         << " in script: " << base::SysNSStringToUTF8(script);
   }
@@ skipping 10 matching lines @@
 
     // Search for 'foo'. Performing the search sets the index to point to the
     // first visible occurence of 'foo'.
     ExecuteJavaScript(kJavaScriptToSearchForFoo);
     AssertJavaScriptValue(kJavaScriptIndex, 1);
     AssertJavaScriptValue(kJavaScriptSpansLength, kNumberOfFoosInHtml);
   }
 
   void SetUp() override {
     ChromeWebTest::SetUp();
-    mockDelegate_.reset([[OCMockObject
-        niceMockForProtocol:@protocol(FindInPageControllerDelegate)] retain]);
-    findInPageController_.reset([[FindInPageController alloc]
-        initWithWebState:web_state()
-                delegate:mockDelegate_]);
+    findInPageModel_.reset([[FindInPageModel alloc] init]);
+    findInPageJsManager_.reset([base::mac::ObjCCastStrict<JsFindinpageManager>(
+        [web_state()->GetJSInjectionReceiver()
+            instanceOfClass:[JsFindinpageManager class]]) retain]);
+    findInPageJsManager_.get().findInPageModel = findInPageModel_;
   }
 
-  base::scoped_nsobject<FindInPageController> findInPageController_;
-  base::scoped_nsobject<id> mockDelegate_;
+  base::scoped_nsobject<FindInPageModel> findInPageModel_;
+  base::scoped_nsobject<JsFindinpageManager> findInPageJsManager_;
 };
 
 // Performs a search, then calls |incrementIndex| to loop through the
 // matches, ensuring that when the end is reached the index wraps back to zero.
 TEST_F(FindInPageJsTest, IncrementIndex) {
   SearchForFoo();
 
   // Increment index until it hits the max index.
   for (int i = 2; i < kNumberOfFoosInHtml; i++) {
     ExecuteJavaScript(kJavaScriptIncrementIndex);
@@ skipping 94 matching lines @@
   // point to the first visible occurence of the non-Ascii.
   NSString* result = ExecuteJavaScript([NSString
       stringWithFormat:@"__gCrWeb.findInPage.highlightWord('%@', false, 1000)",
                        kNonAscii]);
   DCHECK(result);
   AssertJavaScriptValue(kJavaScriptIndex, 0);
   AssertJavaScriptValue(kJavaScriptSpansLength, 1);
 }
 
 }  // namespace