Index: ios/chrome/browser/web/js_findinpage_manager_unittest.mm |
diff --git a/ios/chrome/browser/web/js_findinpage_manager_unittest.mm b/ios/chrome/browser/web/js_findinpage_manager_unittest.mm |
new file mode 100644 |
index 0000000000000000000000000000000000000000..cd914d7e22427a86c4980321fbd844640b24d6d4 |
--- /dev/null |
+++ b/ios/chrome/browser/web/js_findinpage_manager_unittest.mm |
@@ -0,0 +1,79 @@ |
+// Copyright 2014 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 <Foundation/Foundation.h> |
+ |
+#import "base/test/ios/wait_util.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/js/crw_js_injection_receiver.h" |
+#import "ios/web/public/web_state/web_state.h" |
+#import "testing/gtest_mac.h" |
+ |
+// Test fixture to test Find In Page JS. |
+class JsFindinpageManagerTest : public ChromeWebTest { |
+ protected: |
+ // Loads the given HTML and initializes the findInPage JS scripts. |
+ void LoadHtml(NSString* html) { |
+ ChromeWebTest::LoadHtml(html); |
+ manager_ = |
+ static_cast<JsFindinpageManager*>([web_state()->GetJSInjectionReceiver() |
+ instanceOfClass:[JsFindinpageManager class]]); |
+ [manager_ inject]; |
+ } |
+ JsFindinpageManager* manager_; |
+}; |
+ |
+// Tests that findString script reports a match when appropriate. |
+TEST_F(JsFindinpageManagerTest, FindInPageSucceeds) { |
+ LoadHtml(@"<html><body><p>Target phrase</p></body></html>"); |
+ __block BOOL completion_handler_block_was_called = NO; |
+ id completion_handler_block = ^(BOOL success, CGPoint scrollPosition) { |
+ ASSERT_TRUE(success); |
+ // 'scrollPosition' is updated if the string is found. |
+ EXPECT_NE(FLT_MAX, scrollPosition.x); |
+ completion_handler_block_was_called = YES; |
+ }; |
+ [manager_ findString:@"Target phrase" |
+ completionHandler:completion_handler_block]; |
+ base::test::ios::WaitUntilCondition(^bool() { |
+ return completion_handler_block_was_called; |
+ }); |
+} |
+ |
+// Tests that findString script does not report a match when appropriate. |
+TEST_F(JsFindinpageManagerTest, FindInPageFails) { |
+ LoadHtml(@"<html><body><p>Target phrase</p></body></html>"); |
+ __block BOOL completion_handler_block_was_called = NO; |
+ id completion_handler_block = ^(BOOL success, CGPoint scrollPosition) { |
+ // findString should return YES even if the target phrase is not found. |
+ ASSERT_TRUE(success); |
+ // 'point' is *not* updated if the string is not found. |
+ EXPECT_TRUE(CGPointEqualToPoint(CGPointZero, scrollPosition)); |
+ completion_handler_block_was_called = YES; |
+ }; |
+ [manager_ findString:@"Non-included phrase" |
+ completionHandler:completion_handler_block]; |
+ base::test::ios::WaitUntilCondition(^bool() { |
+ return completion_handler_block_was_called; |
+ }); |
+} |
+ |
+// Attepting to break out of the script and inject new script fails. |
+TEST_F(JsFindinpageManagerTest, InjectionTest) { |
+ LoadHtml(@"<html><body><p>Target phrase</p></body></html>"); |
+ __block BOOL completion_handler_block_was_called = NO; |
+ id completion_handler_block = ^(BOOL success, CGPoint scrollPosition) { |
+ [manager_ executeJavaScript:@"token" |
+ completionHandler:^(NSString* result, NSError*) { |
+ EXPECT_NSNE(@YES, result); |
+ completion_handler_block_was_called = YES; |
+ }]; |
+ }; |
+ [manager_ findString:@"');token=true;('" |
+ completionHandler:completion_handler_block]; |
+ base::test::ios::WaitUntilCondition(^bool() { |
+ return completion_handler_block_was_called; |
+ }); |
+} |