Index: chrome/browser/apps/guest_view/extension_view/extension_view_browsertest.cc |
diff --git a/chrome/browser/apps/guest_view/extension_view/extension_view_browsertest.cc b/chrome/browser/apps/guest_view/extension_view/extension_view_browsertest.cc |
index f5392208b29dc1df0de21aa1a5181e6d07f22bac..8c694a8d4f5fa8b4f0b35c012cc906793795913d 100644 |
--- a/chrome/browser/apps/guest_view/extension_view/extension_view_browsertest.cc |
+++ b/chrome/browser/apps/guest_view/extension_view/extension_view_browsertest.cc |
@@ -6,77 +6,21 @@ |
#include "chrome/browser/apps/app_browsertest_util.h" |
#include "chrome/test/base/ui_test_utils.h" |
#include "content/public/test/browser_test_utils.h" |
+#include "extensions/browser/guest_view/extensions_guest_view_manager_delegate.h" |
#include "extensions/browser/guest_view/guest_view_manager.h" |
#include "extensions/browser/guest_view/guest_view_manager_factory.h" |
+#include "extensions/browser/guest_view/test_guest_view_manager.h" |
#include "extensions/test/extension_test_message_listener.h" |
-namespace { |
- |
-class TestGuestViewManager : public extensions::GuestViewManager { |
- public: |
- explicit TestGuestViewManager(content::BrowserContext* context) |
- : extensions::GuestViewManager(context), web_contents_(NULL) {} |
- |
- content::WebContents* WaitForGuestCreated() { |
- if (web_contents_) |
- return web_contents_; |
- |
- message_loop_runner_ = new content::MessageLoopRunner; |
- message_loop_runner_->Run(); |
- return web_contents_; |
- } |
- |
- private: |
- // GuestViewManager override. |
- void AddGuest(int guest_instance_id, |
- content::WebContents* guest_web_contents) override { |
- extensions::GuestViewManager::AddGuest(guest_instance_id, |
- guest_web_contents); |
- web_contents_ = guest_web_contents; |
- |
- if (message_loop_runner_.get()) |
- message_loop_runner_->Quit(); |
- } |
- |
- content::WebContents* web_contents_; |
- scoped_refptr<content::MessageLoopRunner> message_loop_runner_; |
-}; |
- |
-// Test factory for creating test instances of GuestViewManager. |
-class TestGuestViewManagerFactory : public extensions::GuestViewManagerFactory { |
- public: |
- TestGuestViewManagerFactory() : test_guest_view_manager_(NULL) {} |
- |
- ~TestGuestViewManagerFactory() override {} |
- |
- extensions::GuestViewManager* CreateGuestViewManager( |
- content::BrowserContext* context) override { |
- return GetManager(context); |
- } |
- |
- TestGuestViewManager* GetManager(content::BrowserContext* context) { |
- if (!test_guest_view_manager_) { |
- test_guest_view_manager_ = new TestGuestViewManager(context); |
- } |
- return test_guest_view_manager_; |
- } |
- |
- private: |
- TestGuestViewManager* test_guest_view_manager_; |
- |
- DISALLOW_COPY_AND_ASSIGN(TestGuestViewManagerFactory); |
-}; |
- |
-} // namespace |
- |
class ExtensionViewTest : public extensions::PlatformAppBrowserTest { |
public: |
ExtensionViewTest() { |
extensions::GuestViewManager::set_factory_for_testing(&factory_); |
} |
- TestGuestViewManager* GetGuestViewManager() { |
- return factory_.GetManager(browser()->profile()); |
+ extensions::TestGuestViewManager* GetGuestViewManager() { |
+ return static_cast<extensions::TestGuestViewManager*>( |
Fady Samuel
2015/04/24 01:31:23
Implement Create a TestGuestViewManager if one doe
Fady Samuel
2015/04/24 04:37:48
Done.
|
+ extensions::GuestViewManager::FromBrowserContext(browser()->profile())); |
} |
void TestHelper(const std::string& test_name, |
@@ -111,7 +55,7 @@ class ExtensionViewTest : public extensions::PlatformAppBrowserTest { |
extensions::PlatformAppBrowserTest::SetUpCommandLine(command_line); |
} |
- TestGuestViewManagerFactory factory_; |
+ extensions::TestGuestViewManagerFactory factory_; |
}; |
// Tests that <extensionview> can be created and added to the DOM. |