Make WebUI objects not derive from WebUI. WebUI objects own the controller. This is the ownership...

chrome/browser/ui/webui/sync_internals_ui_unittest.cc

 // Copyright (c) 2011 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.
 
 #include "chrome/browser/ui/webui/sync_internals_ui.h"
 
 #include <cstddef>
 #include <string>
 
 #include "base/message_loop.h"
@@ skipping 10 matching lines @@
 #include "content/test/test_browser_thread.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 using browser_sync::HasArgsAsList;
 using browser_sync::JsArgList;
 using browser_sync::JsEventDetails;
 using content::BrowserThread;
+using content::WebContents;
 using testing::_;
 using testing::Mock;
 using testing::NiceMock;
 using testing::Return;
 using testing::StrictMock;
 
-// Subclass of SyncInternalsUI to mock out ExecuteJavascript.
-class TestSyncInternalsUI : public SyncInternalsUI {
+// Subclass of WebUI to mock out ExecuteJavascript.
+class TestSyncWebUI: public WebUI {
  public:
-  explicit TestSyncInternalsUI(TabContents* contents)
-      : SyncInternalsUI(contents) {}
-  virtual ~TestSyncInternalsUI() {}
+  explicit TestSyncWebUI(WebContents* web_contents)
+      : WebUI(web_contents) {}
+  virtual ~TestSyncWebUI() {}
 
   MOCK_METHOD1(ExecuteJavascript, void(const string16&));
 };
 
 // Tests with non-NULL ProfileSyncService.
 class SyncInternalsUITestWithService : public ChromeRenderViewHostTestHarness {
  protected:
-  SyncInternalsUITestWithService() {}
+  SyncInternalsUITestWithService() : sync_internals_ui_(NULL) {}
 
   virtual ~SyncInternalsUITestWithService() {}
 
   virtual void SetUp() {
     NiceMock<ProfileMock>* profile_mock = new NiceMock<ProfileMock>();
     StrictMock<ProfileSyncServiceMock> profile_sync_service_mock;
     EXPECT_CALL(*profile_mock, GetProfileSyncService())
         .WillOnce(Return(&profile_sync_service_mock));
     browser_context_.reset(profile_mock);
 
     ChromeRenderViewHostTestHarness::SetUp();
 
     EXPECT_CALL(profile_sync_service_mock, GetJsController())
         .WillOnce(Return(mock_js_controller_.AsWeakPtr()));
 
     EXPECT_CALL(mock_js_controller_, AddJsEventHandler(_));
 
     {
-      // Needed by |test_sync_internals_ui_|'s constructor.  The
+      // Needed by |sync_internals_ui_|'s constructor.  The
       // message loop is provided by ChromeRenderViewHostTestHarness.
       content::TestBrowserThread ui_thread_(BrowserThread::UI,
                                             MessageLoopForUI::current());
-      // |test_sync_internals_ui_|'s constructor triggers all the
+      // |sync_internals_ui_|'s constructor triggers all the
       // expectations above.
-      test_sync_internals_ui_.reset(new TestSyncInternalsUI(contents()));
+      web_ui_.reset(new TestSyncWebUI(contents()));
+      sync_internals_ui_ = new SyncInternalsUI(web_ui_.get());
+      web_ui_->SetController(sync_internals_ui_);
     }
 
     Mock::VerifyAndClearExpectations(profile_mock);
     Mock::VerifyAndClearExpectations(&mock_js_controller_);
   }
 
   virtual void TearDown() {
     Mock::VerifyAndClearExpectations(&mock_js_controller_);
 
-    // Called by |test_sync_internals_ui_|'s destructor.
+    // Called by |sync_internals_ui_|'s destructor.
     EXPECT_CALL(mock_js_controller_,
-                RemoveJsEventHandler(test_sync_internals_ui_.get()));
-    test_sync_internals_ui_.reset();
+                RemoveJsEventHandler(sync_internals_ui_));
+    sync_internals_ui_ = NULL;
+    web_ui_.reset();
 
     ChromeRenderViewHostTestHarness::TearDown();
   }
 
   StrictMock<browser_sync::MockJsController> mock_js_controller_;
-  scoped_ptr<TestSyncInternalsUI> test_sync_internals_ui_;
+  scoped_ptr<TestSyncWebUI> web_ui_;
+  SyncInternalsUI* sync_internals_ui_;
 };
 
 TEST_F(SyncInternalsUITestWithService, HandleJsEvent) {
-  EXPECT_CALL(*test_sync_internals_ui_,
+  EXPECT_CALL(*web_ui_,
               ExecuteJavascript(
                   ASCIIToUTF16("chrome.sync.testMessage.fire({});")));
 
-  test_sync_internals_ui_->HandleJsEvent("testMessage", JsEventDetails());
+  sync_internals_ui_->HandleJsEvent("testMessage", JsEventDetails());
 }
 
 TEST_F(SyncInternalsUITestWithService, HandleJsReply) {
   EXPECT_CALL(
-      *test_sync_internals_ui_,
+      *web_ui_,
       ExecuteJavascript(
           ASCIIToUTF16("chrome.sync.testMessage.handleReply(5,true);")));
 
   ListValue args;
   args.Append(Value::CreateIntegerValue(5));
   args.Append(Value::CreateBooleanValue(true));
-  test_sync_internals_ui_->HandleJsReply("testMessage", JsArgList(&args));
+  sync_internals_ui_->HandleJsReply("testMessage", JsArgList(&args));
 }
 
 TEST_F(SyncInternalsUITestWithService, OnWebUISendBasic) {
   const std::string& name = "testName";
   ListValue args;
   args.Append(Value::CreateIntegerValue(10));
 
   EXPECT_CALL(mock_js_controller_,
               ProcessJsMessage(name, HasArgsAsList(args), _));
 
-  test_sync_internals_ui_->controller()->OverrideHandleWebUIMessage(
-      GURL(), name, args);
+  sync_internals_ui_->OverrideHandleWebUIMessage(GURL(), name, args);
 }
 
 // Tests with NULL ProfileSyncService.
 class SyncInternalsUITestWithoutService
     : public ChromeRenderViewHostTestHarness {
  protected:
-  SyncInternalsUITestWithoutService() {}
+  SyncInternalsUITestWithoutService() : sync_internals_ui_(NULL) {}
 
   virtual ~SyncInternalsUITestWithoutService() {}
 
   virtual void SetUp() {
     NiceMock<ProfileMock>* profile_mock = new NiceMock<ProfileMock>();
     EXPECT_CALL(*profile_mock, GetProfileSyncService())
         .WillOnce(Return(static_cast<ProfileSyncService*>(NULL)));
     browser_context_.reset(profile_mock);
 
     ChromeRenderViewHostTestHarness::SetUp();
 
     {
-      // Needed by |test_sync_internals_ui_|'s constructor.  The
+      // Needed by |sync_internals_ui_|'s constructor.  The
       // message loop is provided by ChromeRenderViewHostTestHarness.
       content::TestBrowserThread ui_thread_(BrowserThread::UI,
                                             MessageLoopForUI::current());
-      // |test_sync_internals_ui_|'s constructor triggers all the
+      // |sync_internals_ui_|'s constructor triggers all the
       // expectations above.
-      test_sync_internals_ui_.reset(new TestSyncInternalsUI(contents()));
+      web_ui_.reset(new TestSyncWebUI(contents()));
+      sync_internals_ui_ = new SyncInternalsUI(web_ui_.get());
+      web_ui_->SetController(sync_internals_ui_);
     }
 
     Mock::VerifyAndClearExpectations(profile_mock);
   }
 
-  scoped_ptr<TestSyncInternalsUI> test_sync_internals_ui_;
+  scoped_ptr<TestSyncWebUI> web_ui_;
+  SyncInternalsUI* sync_internals_ui_;
 };
 
 TEST_F(SyncInternalsUITestWithoutService, HandleJsEvent) {
-  EXPECT_CALL(*test_sync_internals_ui_,
+  EXPECT_CALL(*web_ui_,
               ExecuteJavascript(
                   ASCIIToUTF16("chrome.sync.testMessage.fire({});")));
 
-  test_sync_internals_ui_->HandleJsEvent("testMessage", JsEventDetails());
+  sync_internals_ui_->HandleJsEvent("testMessage", JsEventDetails());
 }
 
 TEST_F(SyncInternalsUITestWithoutService, HandleJsReply) {
   EXPECT_CALL(
-      *test_sync_internals_ui_,
+      *web_ui_,
       ExecuteJavascript(
           ASCIIToUTF16("chrome.sync.testMessage.handleReply(5,true);")));
 
   ListValue args;
   args.Append(Value::CreateIntegerValue(5));
   args.Append(Value::CreateBooleanValue(true));
-  test_sync_internals_ui_->HandleJsReply(
+  sync_internals_ui_->HandleJsReply(
       "testMessage", JsArgList(&args));
 }
 
 TEST_F(SyncInternalsUITestWithoutService, OnWebUISendBasic) {
   const std::string& name = "testName";
   ListValue args;
   args.Append(Value::CreateIntegerValue(5));
 
   // Should drop the message.
-  test_sync_internals_ui_->controller()->OverrideHandleWebUIMessage(
-      GURL(), name, args);
+  sync_internals_ui_->OverrideHandleWebUIMessage(GURL(), name, args);
 }
 
 // TODO(lipalani) - add a test case to test about:sync with a non null
 // service.
 TEST_F(SyncInternalsUITestWithoutService, OnWebUISendGetAboutInfo) {
   const char kAboutInfoCall[] =
       "chrome.sync.getAboutInfo.handleReply({\"summary\":\"SYNC DISABLED\"});";
-  EXPECT_CALL(*test_sync_internals_ui_,
+  EXPECT_CALL(*web_ui_,
               ExecuteJavascript(ASCIIToUTF16(kAboutInfoCall)));
 
   ListValue args;
-  test_sync_internals_ui_->controller()->OverrideHandleWebUIMessage(
+  sync_internals_ui_->OverrideHandleWebUIMessage(
       GURL(), "getAboutInfo", args);
 }
 
 }  // namespace