| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "extensions/browser/guest_view/guest_view_manager.h" | 5 #include "extensions/browser/guest_view/guest_view_manager.h" |
| 6 | 6 |
| 7 #include "content/public/browser/notification_service.h" | 7 #include "content/public/browser/notification_service.h" |
| 8 #include "content/public/test/test_browser_context.h" | 8 #include "content/public/test/test_browser_context.h" |
| 9 #include "content/public/test/test_browser_thread_bundle.h" | 9 #include "content/public/test/test_browser_thread_bundle.h" |
| 10 #include "content/public/test/web_contents_tester.h" | 10 #include "content/public/test/web_contents_tester.h" |
| 11 #include "extensions/browser/extensions_test.h" | 11 #include "extensions/browser/extensions_test.h" |
| 12 #include "extensions/browser/guest_view/extensions_guest_view_manager_delegate.h
" |
| 13 #include "extensions/browser/guest_view/guest_view_manager.h" |
| 14 #include "extensions/browser/guest_view/guest_view_manager_delegate.h" |
| 15 #include "extensions/browser/guest_view/test_guest_view_manager.h" |
| 12 | 16 |
| 13 using content::WebContents; | 17 using content::WebContents; |
| 14 using content::WebContentsTester; | 18 using content::WebContentsTester; |
| 19 using guestview::GuestViewManagerDelegate; |
| 15 | 20 |
| 16 namespace extensions { | 21 namespace extensions { |
| 17 | 22 |
| 18 namespace guestview { | |
| 19 | |
| 20 // This class allows us to access some private variables in | |
| 21 // GuestViewManager. | |
| 22 class TestGuestViewManager : public GuestViewManager { | |
| 23 public: | |
| 24 explicit TestGuestViewManager(content::BrowserContext* context) | |
| 25 : GuestViewManager(context) {} | |
| 26 | |
| 27 int last_instance_id_removed_for_testing() { | |
| 28 return last_instance_id_removed_; | |
| 29 } | |
| 30 | |
| 31 size_t GetRemovedInstanceIdSize() { return removed_instance_ids_.size(); } | |
| 32 | |
| 33 private: | |
| 34 using GuestViewManager::last_instance_id_removed_; | |
| 35 using GuestViewManager::removed_instance_ids_; | |
| 36 | |
| 37 DISALLOW_COPY_AND_ASSIGN(TestGuestViewManager); | |
| 38 }; | |
| 39 | |
| 40 } // namespace guestview | |
| 41 | |
| 42 namespace { | 23 namespace { |
| 43 | 24 |
| 44 class GuestViewManagerTest : public extensions::ExtensionsTest { | 25 class GuestViewManagerTest : public extensions::ExtensionsTest { |
| 45 public: | 26 public: |
| 46 GuestViewManagerTest() : | 27 GuestViewManagerTest() : |
| 47 notification_service_(content::NotificationService::Create()) {} | 28 notification_service_(content::NotificationService::Create()) {} |
| 48 ~GuestViewManagerTest() override {} | 29 ~GuestViewManagerTest() override {} |
| 49 | 30 |
| 50 scoped_ptr<WebContents> CreateWebContents() { | 31 scoped_ptr<WebContents> CreateWebContents() { |
| 51 return scoped_ptr<WebContents>( | 32 return scoped_ptr<WebContents>( |
| 52 WebContentsTester::CreateTestWebContents(&browser_context_, NULL)); | 33 WebContentsTester::CreateTestWebContents(&browser_context_, NULL)); |
| 53 } | 34 } |
| 54 | 35 |
| 55 private: | 36 private: |
| 56 scoped_ptr<content::NotificationService> notification_service_; | 37 scoped_ptr<content::NotificationService> notification_service_; |
| 57 content::TestBrowserThreadBundle thread_bundle_; | 38 content::TestBrowserThreadBundle thread_bundle_; |
| 58 content::TestBrowserContext browser_context_; | 39 content::TestBrowserContext browser_context_; |
| 59 | 40 |
| 60 DISALLOW_COPY_AND_ASSIGN(GuestViewManagerTest); | 41 DISALLOW_COPY_AND_ASSIGN(GuestViewManagerTest); |
| 61 }; | 42 }; |
| 62 | 43 |
| 63 } // namespace | 44 } // namespace |
| 64 | 45 |
| 65 TEST_F(GuestViewManagerTest, AddRemove) { | 46 TEST_F(GuestViewManagerTest, AddRemove) { |
| 66 content::TestBrowserContext browser_context; | 47 content::TestBrowserContext browser_context; |
| 67 scoped_ptr<guestview::TestGuestViewManager> manager( | 48 scoped_ptr<GuestViewManagerDelegate> delegate( |
| 68 new guestview::TestGuestViewManager(&browser_context)); | 49 new ExtensionsGuestViewManagerDelegate(&browser_context)); |
| 50 scoped_ptr<TestGuestViewManager> manager( |
| 51 new extensions::TestGuestViewManager(&browser_context, delegate.Pass())); |
| 69 | 52 |
| 70 scoped_ptr<WebContents> web_contents1(CreateWebContents()); | 53 scoped_ptr<WebContents> web_contents1(CreateWebContents()); |
| 71 scoped_ptr<WebContents> web_contents2(CreateWebContents()); | 54 scoped_ptr<WebContents> web_contents2(CreateWebContents()); |
| 72 scoped_ptr<WebContents> web_contents3(CreateWebContents()); | 55 scoped_ptr<WebContents> web_contents3(CreateWebContents()); |
| 73 | 56 |
| 74 EXPECT_EQ(0, manager->last_instance_id_removed_for_testing()); | 57 EXPECT_EQ(0, manager->last_instance_id_removed()); |
| 75 | 58 |
| 76 EXPECT_TRUE(manager->CanUseGuestInstanceID(1)); | 59 EXPECT_TRUE(manager->CanUseGuestInstanceID(1)); |
| 77 EXPECT_TRUE(manager->CanUseGuestInstanceID(2)); | 60 EXPECT_TRUE(manager->CanUseGuestInstanceID(2)); |
| 78 EXPECT_TRUE(manager->CanUseGuestInstanceID(3)); | 61 EXPECT_TRUE(manager->CanUseGuestInstanceID(3)); |
| 79 | 62 |
| 80 manager->AddGuest(1, web_contents1.get()); | 63 manager->AddGuest(1, web_contents1.get()); |
| 81 manager->AddGuest(2, web_contents2.get()); | 64 manager->AddGuest(2, web_contents2.get()); |
| 82 manager->RemoveGuest(2); | 65 manager->RemoveGuest(2); |
| 83 | 66 |
| 84 // Since we removed 2, it would be an invalid ID. | 67 // Since we removed 2, it would be an invalid ID. |
| 85 EXPECT_TRUE(manager->CanUseGuestInstanceID(1)); | 68 EXPECT_TRUE(manager->CanUseGuestInstanceID(1)); |
| 86 EXPECT_FALSE(manager->CanUseGuestInstanceID(2)); | 69 EXPECT_FALSE(manager->CanUseGuestInstanceID(2)); |
| 87 EXPECT_TRUE(manager->CanUseGuestInstanceID(3)); | 70 EXPECT_TRUE(manager->CanUseGuestInstanceID(3)); |
| 88 | 71 |
| 89 EXPECT_EQ(0, manager->last_instance_id_removed_for_testing()); | 72 EXPECT_EQ(0, manager->last_instance_id_removed()); |
| 90 | 73 |
| 91 EXPECT_TRUE(manager->CanUseGuestInstanceID(3)); | 74 EXPECT_TRUE(manager->CanUseGuestInstanceID(3)); |
| 92 | 75 |
| 93 manager->AddGuest(3, web_contents3.get()); | 76 manager->AddGuest(3, web_contents3.get()); |
| 94 manager->RemoveGuest(1); | 77 manager->RemoveGuest(1); |
| 95 EXPECT_FALSE(manager->CanUseGuestInstanceID(1)); | 78 EXPECT_FALSE(manager->CanUseGuestInstanceID(1)); |
| 96 EXPECT_FALSE(manager->CanUseGuestInstanceID(2)); | 79 EXPECT_FALSE(manager->CanUseGuestInstanceID(2)); |
| 97 | 80 |
| 98 EXPECT_EQ(2, manager->last_instance_id_removed_for_testing()); | 81 EXPECT_EQ(2, manager->last_instance_id_removed()); |
| 99 manager->RemoveGuest(3); | 82 manager->RemoveGuest(3); |
| 100 EXPECT_EQ(3, manager->last_instance_id_removed_for_testing()); | 83 EXPECT_EQ(3, manager->last_instance_id_removed()); |
| 101 | 84 |
| 102 EXPECT_FALSE(manager->CanUseGuestInstanceID(1)); | 85 EXPECT_FALSE(manager->CanUseGuestInstanceID(1)); |
| 103 EXPECT_FALSE(manager->CanUseGuestInstanceID(2)); | 86 EXPECT_FALSE(manager->CanUseGuestInstanceID(2)); |
| 104 EXPECT_FALSE(manager->CanUseGuestInstanceID(3)); | 87 EXPECT_FALSE(manager->CanUseGuestInstanceID(3)); |
| 105 | 88 |
| 106 EXPECT_EQ(0u, manager->GetRemovedInstanceIdSize()); | 89 EXPECT_EQ(0, manager->GetNumRemovedInstanceIDs()); |
| 107 } | 90 } |
| 108 | 91 |
| 109 } // namespace extensions | 92 } // namespace extensions |
| OLD | NEW |