chrome/test/base/test_chrome_web_ui_controller_factory_browsertest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/test/base/in_process_browser_test.h"
   6 #include "chrome/test/base/test_chrome_web_ui_controller_factory.h"
   7 #include "chrome/test/base/ui_test_utils.h"
   8 #include "content/public/browser/web_ui_controller.h"
   9 #include "testing/gmock/include/gmock/gmock.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11 #include "url/gurl.h"
  12
  13 using content::WebContents;
  14 using content::WebUI;
  15 using content::WebUIController;
  16 using ::testing::_;
  17 using ::testing::Eq;
  18 using ::testing::StrictMock;
  19
  20 namespace {
  21
  22 // Returns a new WebUI object for the WebContents from |arg0|.
  23 ACTION(ReturnNewWebUI) {
  24   return new WebUIController(arg0);
  25 }
  26
  27 // Mock the TestChromeWebUIControllerFactory::WebUIProvider to prove that we are
  28 // called as expected.
  29 class MockWebUIProvider
  30     : public TestChromeWebUIControllerFactory::WebUIProvider {
  31  public:
  32   MOCK_METHOD2(NewWebUI, WebUIController*(content::WebUI* web_ui,
  33                                           const GURL& url));
  34 };
  35
  36 // Dummy URL location for us to override.
  37 const char kChromeTestChromeWebUIControllerFactory[] =
  38     "chrome://ChromeTestChromeWebUIControllerFactory/";
  39
  40 // Sets up and tears down the factory override for our url's host. It is
  41 // necessary to do this here, rather than in the test declaration, which is too
  42 // late to catch the possibility of an initial browse to about:blank mistakenly
  43 // going to this handler.
  44 class TestChromeWebUIControllerFactoryTest : public InProcessBrowserTest {
  45  public:
  46   void SetUpOnMainThread() override {
  47     content::WebUIControllerFactory::UnregisterFactoryForTesting(
  48         ChromeWebUIControllerFactory::GetInstance());
  49     test_factory_.reset(new TestChromeWebUIControllerFactory);
  50     content::WebUIControllerFactory::RegisterFactory(test_factory_.get());
  51     test_factory_->AddFactoryOverride(
  52         GURL(kChromeTestChromeWebUIControllerFactory).host(), &mock_provider_);
  53   }
  54
  55   void TearDownOnMainThread() override {
  56     test_factory_->RemoveFactoryOverride(
  57         GURL(kChromeTestChromeWebUIControllerFactory).host());
  58     content::WebUIControllerFactory::UnregisterFactoryForTesting(
  59         test_factory_.get());
  60
  61     test_factory_.reset();
  62   }
  63
  64  protected:
  65   StrictMock<MockWebUIProvider> mock_provider_;
  66   scoped_ptr<TestChromeWebUIControllerFactory> test_factory_;
  67 };
  68
  69 }  // namespace
  70
  71 // Test that browsing to our test url causes us to be called once.
  72 IN_PROC_BROWSER_TEST_F(TestChromeWebUIControllerFactoryTest,
  73                        TestWebUIProvider) {
  74   const GURL kChromeTestChromeWebUIControllerFactoryURL(
  75       kChromeTestChromeWebUIControllerFactory);
  76   EXPECT_CALL(mock_provider_,
  77               NewWebUI(_, Eq(kChromeTestChromeWebUIControllerFactoryURL)))
  78       .WillOnce(ReturnNewWebUI());
  79   ui_test_utils::NavigateToURL(browser(),
  80                                kChromeTestChromeWebUIControllerFactoryURL);
  81 }