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

chrome/browser/ui/webui/print_preview/print_preview_ui_unittest.cc

 // Copyright (c) 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.
 
 #include <vector>
 
 #include "base/memory/ref_counted_memory.h"
 #include "chrome/browser/printing/print_preview_tab_controller.h"
 #include "chrome/browser/printing/print_preview_unit_test_base.h"
 #include "chrome/browser/printing/print_view_manager.h"
@@ skipping 40 matching lines @@
   ASSERT_TRUE(controller);
 
   initiator_tab->print_view_manager()->PrintPreviewNow();
   TabContentsWrapper* preview_tab =
       controller->GetOrCreatePreviewTab(initiator_tab);
 
   EXPECT_NE(initiator_tab, preview_tab);
   EXPECT_EQ(1, browser()->tab_count());
   EXPECT_EQ(1U, GetConstrainedWindowCount(initiator_tab));
 
-  PrintPreviewUI* preview_ui = reinterpret_cast<PrintPreviewUI*>(
-      preview_tab->web_contents()->GetWebUI());
+  PrintPreviewUI* preview_ui = static_cast<PrintPreviewUI*>(
+      preview_tab->web_contents()->GetWebUI()->GetController());
   ASSERT_TRUE(preview_ui != NULL);
 
   scoped_refptr<RefCountedBytes> data;
   preview_ui->GetPrintPreviewDataForIndex(
       printing::COMPLETE_PREVIEW_DOCUMENT_INDEX,
       &data);
   EXPECT_EQ(NULL, data.get());
 
   std::vector<unsigned char> preview_data(blob1, blob1 + sizeof(blob1));
   scoped_refptr<RefCountedBytes> dummy_data(new RefCountedBytes(preview_data));
@@ skipping 32 matching lines @@
   ASSERT_TRUE(controller);
 
   initiator_tab->print_view_manager()->PrintPreviewNow();
   TabContentsWrapper* preview_tab =
       controller->GetOrCreatePreviewTab(initiator_tab);
 
   EXPECT_NE(initiator_tab, preview_tab);
   EXPECT_EQ(1, browser()->tab_count());
   EXPECT_EQ(1U, GetConstrainedWindowCount(initiator_tab));
 
-  PrintPreviewUI* preview_ui = reinterpret_cast<PrintPreviewUI*>(
-      preview_tab->web_contents()->GetWebUI());
+  PrintPreviewUI* preview_ui = static_cast<PrintPreviewUI*>(
+      preview_tab->web_contents()->GetWebUI()->GetController());
   ASSERT_TRUE(preview_ui != NULL);
 
   scoped_refptr<RefCountedBytes> data;
   preview_ui->GetPrintPreviewDataForIndex(printing::FIRST_PAGE_INDEX, &data);
   EXPECT_EQ(NULL, data.get());
 
   std::vector<unsigned char> preview_data(blob1, blob1 + sizeof(blob1));
   scoped_refptr<RefCountedBytes> dummy_data(new RefCountedBytes(preview_data));
 
   preview_ui->SetPrintPreviewDataForIndex(printing::FIRST_PAGE_INDEX,
@@ skipping 39 matching lines @@
   ASSERT_TRUE(controller);
 
   initiator_tab->print_view_manager()->PrintPreviewNow();
   TabContentsWrapper* preview_tab =
       controller->GetOrCreatePreviewTab(initiator_tab);
 
   EXPECT_NE(initiator_tab, preview_tab);
   EXPECT_EQ(1, browser()->tab_count());
   EXPECT_EQ(1U, GetConstrainedWindowCount(initiator_tab));
 
-  PrintPreviewUI* preview_ui = reinterpret_cast<PrintPreviewUI*>(
-      preview_tab->web_contents()->GetWebUI());
+  PrintPreviewUI* preview_ui = static_cast<PrintPreviewUI*>(
+      preview_tab->web_contents()->GetWebUI()->GetController());
   ASSERT_TRUE(preview_ui != NULL);
 
   // Test with invalid |preview_ui_addr|.
   bool cancel = false;
   preview_ui->GetCurrentPrintPreviewStatus("invalid", 0, &cancel);
   EXPECT_TRUE(cancel);
 
   const int kFirstRequestId = 1000;
   const int kSecondRequestId = 1001;
   const std::string preview_ui_addr = preview_ui->GetPrintPreviewUIAddress();
@@ skipping 15 matching lines @@
   cancel = false;
   preview_ui->GetCurrentPrintPreviewStatus(preview_ui_addr, kFirstRequestId,
                                            &cancel);
   EXPECT_TRUE(cancel);
 
   cancel = true;
   preview_ui->GetCurrentPrintPreviewStatus(preview_ui_addr, kSecondRequestId,
                                            &cancel);
   EXPECT_FALSE(cancel);
 }