Upstream Chrome on iOS source code [8/11].

ios/chrome/browser/ui/static_content/static_html_view_controller_unittest.mm

Index: ios/chrome/browser/ui/static_content/static_html_view_controller_unittest.mm
diff --git a/ios/chrome/browser/ui/static_content/static_html_view_controller_unittest.mm b/ios/chrome/browser/ui/static_content/static_html_view_controller_unittest.mm
new file mode 100644
index 0000000000000000000000000000000000000000..ea2506cdb05deb8d41707d82b879af1f232a5853
--- /dev/null
+++ b/ios/chrome/browser/ui/static_content/static_html_view_controller_unittest.mm
@@ -0,0 +1,251 @@
+// 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 "ios/chrome/browser/ui/static_content/static_html_view_controller.h"
+
+#include <memory>
+
+#import "base/mac/foundation_util.h"
+#include "base/mac/scoped_nsautorelease_pool.h"
+#include "base/mac/scoped_nsobject.h"
+#include "base/memory/ptr_util.h"
+#import "base/test/ios/wait_util.h"
+#include "ios/chrome/browser/browser_state/test_chrome_browser_state.h"
+#import "ios/chrome/browser/ui/url_loader.h"
+#include "ios/chrome/grit/ios_strings.h"
+#import "ios/testing/ocmock_complex_type_helper.h"
+#include "ios/web/public/referrer.h"
+#import "ios/web/public/test/js_test_util.h"
+#include "ios/web/public/test/scoped_testing_web_client.h"
+#import "ios/web/public/test/test_web_client.h"
+#include "ios/web/public/test/test_web_thread_bundle.h"
+#import "ios/web/public/web_state/ui/crw_native_content.h"
+#import "net/base/mac/url_conversions.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/gtest_mac.h"
+#include "testing/platform_test.h"
+#import "third_party/ocmock/OCMock/OCMock.h"
+#include "third_party/ocmock/gtest_support.h"
+#include "ui/base/l10n/l10n_util_mac.h"
+
+// Generator that returns the value of a localized identifier.
+@interface L10nHtmlGenerator : NSObject<HtmlGenerator> {
+ @private
+  int messageId_;
+}
+- (id)initWithMessageId:(int)messageId;
+@end
+
+@implementation L10nHtmlGenerator
+- (id)initWithMessageId:(int)messageId {
+  if ((self = [super init])) {
+    messageId_ = messageId;
+  }
+  return self;
+}
+- (void)generateHtml:(HtmlCallback)callback {
+  callback(l10n_util::GetNSString(messageId_));
+}
+@end
+
+@interface LoadTestMockLoader : OCMockComplexTypeHelper
+@end
+
+@implementation LoadTestMockLoader
+
+typedef void (^LoadURL_Referrer_transition_renderInitiated)(
+    const GURL&,
+    const web::Referrer&,
+    ui::PageTransition,
+    BOOL);
+
+- (void)loadURL:(const GURL&)url
+             referrer:(const web::Referrer&)referrer
+           transition:(ui::PageTransition)transition
+    rendererInitiated:(BOOL)rendererInitiated {
+  static_cast<LoadURL_Referrer_transition_renderInitiated>([self
+      blockForSelector:_cmd])(url, referrer, transition, rendererInitiated);
+}
+
+@end
+
+namespace {
+
+bool isRunLoopDry = true;
+
+// Wait until either all event in the run loop at the time of execution are
+// executed, or 5.0 seconds, whichever happens first. If |publishSentinel| is
+// false, the caller is responsible to set |isRunLoopDry| to true when it
+// considers the run loop dry.
+void DryRunLoop(bool publishSentinel) {
+  isRunLoopDry = false;
+  if (publishSentinel) {
+    dispatch_async(dispatch_get_main_queue(), ^{
+      isRunLoopDry = true;
+    });
+  }
+  NSDate* startDate = [NSDate date];
+  NSDate* endDate = [NSDate dateWithTimeIntervalSinceNow:5.0];
+
+  while (!isRunLoopDry && (-[startDate timeIntervalSinceNow]) < 5.0) {
+    [[NSRunLoop currentRunLoop] runMode:NSDefaultRunLoopMode
+                             beforeDate:endDate];
+  }
+}
+
+class StaticHtmlViewControllerTest : public PlatformTest {
+ public:
+  StaticHtmlViewControllerTest()
+      : web_client_(base::MakeUnique<web::TestWebClient>()) {}
+
+ protected:
+  void SetUp() override {
+    PlatformTest::SetUp();
+    chrome_browser_state_ = TestChromeBrowserState::Builder().Build();
+  }
+
+  web::TestWebThreadBundle thread_bundle_;
+  std::unique_ptr<TestChromeBrowserState> chrome_browser_state_;
+  // Private autorelease pool so objects are released as soon as possible.
+  base::mac::ScopedNSAutoreleasePool pool_;
+  web::ScopedTestingWebClient web_client_;
+};
+
+// Tests the creation of a StaticHtmlViewController displaying a resource file.
+TEST_F(StaticHtmlViewControllerTest, LoadResourceTest) {
+  base::scoped_nsobject<id> loader;
+  loader.reset([[LoadTestMockLoader alloc]
+      initWithRepresentedObject:[OCMockObject
+                                    mockForProtocol:@protocol(UrlLoader)]]);
+
+  id<CRWNativeContentDelegate> delegate =
+      [OCMockObject mockForProtocol:@protocol(CRWNativeContentDelegate)];
+  GURL referrer_url("chrome://foo");
+  web::Referrer referrer(referrer_url, web::ReferrerPolicyDefault);
+  base::scoped_nsobject<StaticHtmlViewController> content(
+      [[StaticHtmlViewController alloc]
+          initWithResource:@"terms_en.html"
+              browserState:chrome_browser_state_.get()]);
+  [content setLoader:loader referrer:referrer];
+  [content setDelegate:delegate];
+  [[(OCMockObject*)delegate expect]
+       nativeContent:content.get()
+      titleDidChange:[OCMArg checkWithBlock:^BOOL(id value) {
+        isRunLoopDry = true;
+        return [@"Google Chrome Terms of Service"
+            isEqualToString:(NSString*)value];
+      }]];
+  [content triggerPendingLoad];
+  DryRunLoop(false);
+  ASSERT_OCMOCK_VERIFY(loader.get());
+  ASSERT_OCMOCK_VERIFY((OCMockObject*)delegate);
+  base::scoped_nsobject<id> block(
+      [(id) ^ (const GURL& url, const web::Referrer& referrer,
+               ui::PageTransition transition, BOOL rendererInitiated) {
+        EXPECT_EQ(url, GURL());
+        EXPECT_EQ(referrer.url, referrer_url);
+        EXPECT_EQ(referrer.policy, web::ReferrerPolicyDefault);
+        EXPECT_TRUE(
+            PageTransitionCoreTypeIs(transition, ui::PAGE_TRANSITION_LINK));
+        EXPECT_TRUE(rendererInitiated);
+      } copy]);
+
+  [loader onSelector:@selector(loadURL:referrer:transition:rendererInitiated:)
+      callBlockExpectation:block];
+
+  DryRunLoop(true);
+  ASSERT_OCMOCK_VERIFY(loader.get());
+  ASSERT_OCMOCK_VERIFY((OCMockObject*)delegate);
+}
+
+// Tests the creation of a StaticHtmlViewController displaying a local file.
+TEST_F(StaticHtmlViewControllerTest, LoadFileURLTest) {
+  base::scoped_nsobject<id> loader;
+  loader.reset([[LoadTestMockLoader alloc]
+      initWithRepresentedObject:[OCMockObject
+                                    mockForProtocol:@protocol(UrlLoader)]]);
+
+  id<CRWNativeContentDelegate> delegate =
+      [OCMockObject mockForProtocol:@protocol(CRWNativeContentDelegate)];
+  GURL referrer_url("chrome://foo");
+  web::Referrer referrer(referrer_url, web::ReferrerPolicyDefault);
+  NSURL* fileURL = [NSURL
+      fileURLWithPath:[[NSBundle mainBundle] pathForResource:@"terms_en.html"
+                                                      ofType:nil
+                                                 inDirectory:nil]];
+  base::scoped_nsobject<StaticHtmlViewController> content(
+      [[StaticHtmlViewController alloc]
+                  initWithFileURL:net::GURLWithNSURL(fileURL)
+          allowingReadAccessToURL:net::GURLWithNSURL(
+                                      [fileURL URLByDeletingLastPathComponent])
+                     browserState:chrome_browser_state_.get()]);
+  [content setLoader:loader referrer:referrer];
+  [content setDelegate:delegate];
+  [[(OCMockObject*)delegate expect]
+       nativeContent:content.get()
+      titleDidChange:[OCMArg checkWithBlock:^BOOL(id value) {
+        isRunLoopDry = true;
+        return [@"Google Chrome Terms of Service"
+            isEqualToString:(NSString*)value];
+      }]];
+  [content triggerPendingLoad];
+  DryRunLoop(false);
+  ASSERT_OCMOCK_VERIFY(loader.get());
+  ASSERT_OCMOCK_VERIFY((OCMockObject*)delegate);
+  base::scoped_nsobject<id> block(
+      [(id) ^ (const GURL& url, const web::Referrer& referrer,
+               ui::PageTransition transition, BOOL rendererInitiated) {
+        EXPECT_EQ(url, GURL());
+        EXPECT_EQ(referrer.url, referrer_url);
+        EXPECT_EQ(referrer.policy, web::ReferrerPolicyDefault);
+        EXPECT_TRUE(
+            PageTransitionCoreTypeIs(transition, ui::PAGE_TRANSITION_LINK));
+        EXPECT_TRUE(rendererInitiated);
+      } copy]);
+
+  [loader onSelector:@selector(loadURL:referrer:transition:rendererInitiated:)
+      callBlockExpectation:block];
+
+  DryRunLoop(true);
+  ASSERT_OCMOCK_VERIFY(loader.get());
+  ASSERT_OCMOCK_VERIFY((OCMockObject*)delegate);
+}
+
+// Tests that -[StaticHtmlViewController webView] returns a non-nil view.
+TEST_F(StaticHtmlViewControllerTest, WebViewNonNil) {
+  base::scoped_nsobject<L10nHtmlGenerator> generator(
+      [[L10nHtmlGenerator alloc] initWithMessageId:IDS_IOS_TOOLS_MENU]);
+  base::scoped_nsobject<StaticHtmlViewController> staticHtmlViewController(
+      [[StaticHtmlViewController alloc]
+          initWithGenerator:generator
+               browserState:chrome_browser_state_.get()]);
+  EXPECT_TRUE([staticHtmlViewController webView]);
+}
+
+// Tests the generated HTML is localized.
+TEST_F(StaticHtmlViewControllerTest, L10NTest) {
+  base::scoped_nsobject<L10nHtmlGenerator> generator(
+      [[L10nHtmlGenerator alloc] initWithMessageId:IDS_IOS_TOOLS_MENU]);
+  base::scoped_nsobject<StaticHtmlViewController> content(
+      [[StaticHtmlViewController alloc]
+          initWithGenerator:generator
+               browserState:chrome_browser_state_.get()]);
+  id<UrlLoader> loader = [OCMockObject mockForProtocol:@protocol(UrlLoader)];
+  [content setLoader:loader
+            referrer:web::Referrer(GURL("chrome://foo"),
+                                   web::ReferrerPolicyDefault)];
+  [content triggerPendingLoad];
+  ASSERT_OCMOCK_VERIFY((OCMockObject*)loader);
+  __block id string_in_page = nil;
+  base::test::ios::WaitUntilCondition(^bool {
+    string_in_page =
+        web::ExecuteJavaScript([content webView], @"document.body.innerHTML");
+    return ![string_in_page isEqual:@""];
+  });
+  EXPECT_TRUE([string_in_page isKindOfClass:[NSString class]]);
+  NSString* to_find = l10n_util::GetNSString(IDS_IOS_TOOLS_MENU);
+  EXPECT_TRUE([string_in_page rangeOfString:to_find].length);
+}
+
+}  // namespace