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

ios/chrome/browser/ui/chrome_web_view_factory_unittest.mm

Index: ios/chrome/browser/ui/chrome_web_view_factory_unittest.mm
diff --git a/ios/chrome/browser/ui/chrome_web_view_factory_unittest.mm b/ios/chrome/browser/ui/chrome_web_view_factory_unittest.mm
new file mode 100644
index 0000000000000000000000000000000000000000..92e7ede8765a6c1f55342e10c94f16b92c5e7f22
--- /dev/null
+++ b/ios/chrome/browser/ui/chrome_web_view_factory_unittest.mm
@@ -0,0 +1,68 @@
+// 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/chrome_web_view_factory.h"
+
+#include "base/mac/scoped_nsobject.h"
+#include "base/memory/ptr_util.h"
+#include "base/message_loop/message_loop.h"
+#include "base/run_loop.h"
+#include "ios/chrome/browser/browser_state/test_chrome_browser_state_isolated_context.h"
+#import "ios/chrome/browser/web/chrome_web_client.h"
+#include "ios/web/net/request_group_util.h"
+#include "ios/web/net/request_tracker_impl.h"
+#include "ios/web/public/test/scoped_testing_web_client.h"
+#include "ios/web/public/test/test_web_thread.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"
+#import "third_party/ocmock/gtest_support.h"
+
+using web::RequestTrackerImpl;
+
+namespace {
+
+class ChromeWebViewFactoryTest : public PlatformTest {
+ public:
+  ChromeWebViewFactoryTest()
+      : ui_thread_(web::WebThread::UI, &message_loop_),
+        io_thread_(web::WebThread::IO, &message_loop_),
+        web_client_(base::MakeUnique<ChromeWebClient>()) {}
+
+ protected:
+  base::MessageLoop message_loop_;
+  web::TestWebThread ui_thread_;
+  web::TestWebThread io_thread_;
+  web::ScopedTestingWebClient web_client_;
+  TestChromeBrowserStateWithIsolatedContext chrome_browser_state_;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(ChromeWebViewFactoryTest);
+};
+
+TEST_F(ChromeWebViewFactoryTest, TestTrackerForExternal) {
+  [ChromeWebViewFactory setBrowserStateToUseForExternal:&chrome_browser_state_];
+  base::scoped_nsobject<UIWebView> webView([ChromeWebViewFactory
+      newExternalWebView:chrome_browser_state_.SharingService()]);
+  // Check that the tracker is registered
+  RequestTrackerImpl* tracker = RequestTrackerImpl::GetTrackerForRequestGroupID(
+      ChromeWebView::kExternalRequestGroupID);
+  EXPECT_TRUE(tracker) << "Tracker not registered.";
+  // External should use the request context for the sharing service, and not
+  // the main one.
+  net::URLRequestContextGetter* expected_getter = [ChromeWebViewFactory
+      requestContextForExternalService:chrome_browser_state_.SharingService()];
+  EXPECT_EQ(expected_getter->GetURLRequestContext(),
+            tracker->GetRequestContext());
+  EXPECT_FALSE(chrome_browser_state_.MainContextCalled());
+  // Unregister the tracker.
+  [ChromeWebViewFactory externalSessionFinished];
+  base::RunLoop().RunUntilIdle();
+  EXPECT_FALSE(RequestTrackerImpl::GetTrackerForRequestGroupID(
+      ChromeWebView::kExternalRequestGroupID));
+  [ChromeWebViewFactory setBrowserStateToUseForExternal:nullptr];
+}
+
+}  // namespace