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 |