OLD | NEW |
| (Empty) |
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 | |
3 // found in the LICENSE file. | |
4 | |
5 #include "extensions/browser/guest_view/guest_view_manager.h" | |
6 | |
7 #include "content/public/browser/notification_service.h" | |
8 #include "content/public/test/test_browser_context.h" | |
9 #include "content/public/test/test_browser_thread_bundle.h" | |
10 #include "content/public/test/web_contents_tester.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" | |
16 | |
17 using content::WebContents; | |
18 using content::WebContentsTester; | |
19 using guestview::GuestViewManagerDelegate; | |
20 | |
21 namespace extensions { | |
22 | |
23 namespace { | |
24 | |
25 class GuestViewManagerTest : public extensions::ExtensionsTest { | |
26 public: | |
27 GuestViewManagerTest() : | |
28 notification_service_(content::NotificationService::Create()) {} | |
29 ~GuestViewManagerTest() override {} | |
30 | |
31 scoped_ptr<WebContents> CreateWebContents() { | |
32 return scoped_ptr<WebContents>( | |
33 WebContentsTester::CreateTestWebContents(&browser_context_, NULL)); | |
34 } | |
35 | |
36 private: | |
37 scoped_ptr<content::NotificationService> notification_service_; | |
38 content::TestBrowserThreadBundle thread_bundle_; | |
39 content::TestBrowserContext browser_context_; | |
40 | |
41 DISALLOW_COPY_AND_ASSIGN(GuestViewManagerTest); | |
42 }; | |
43 | |
44 } // namespace | |
45 | |
46 TEST_F(GuestViewManagerTest, AddRemove) { | |
47 content::TestBrowserContext browser_context; | |
48 scoped_ptr<GuestViewManagerDelegate> delegate( | |
49 new ExtensionsGuestViewManagerDelegate(&browser_context)); | |
50 scoped_ptr<TestGuestViewManager> manager( | |
51 new extensions::TestGuestViewManager(&browser_context, delegate.Pass())); | |
52 | |
53 scoped_ptr<WebContents> web_contents1(CreateWebContents()); | |
54 scoped_ptr<WebContents> web_contents2(CreateWebContents()); | |
55 scoped_ptr<WebContents> web_contents3(CreateWebContents()); | |
56 | |
57 EXPECT_EQ(0, manager->last_instance_id_removed()); | |
58 | |
59 EXPECT_TRUE(manager->CanUseGuestInstanceID(1)); | |
60 EXPECT_TRUE(manager->CanUseGuestInstanceID(2)); | |
61 EXPECT_TRUE(manager->CanUseGuestInstanceID(3)); | |
62 | |
63 manager->AddGuest(1, web_contents1.get()); | |
64 manager->AddGuest(2, web_contents2.get()); | |
65 manager->RemoveGuest(2); | |
66 | |
67 // Since we removed 2, it would be an invalid ID. | |
68 EXPECT_TRUE(manager->CanUseGuestInstanceID(1)); | |
69 EXPECT_FALSE(manager->CanUseGuestInstanceID(2)); | |
70 EXPECT_TRUE(manager->CanUseGuestInstanceID(3)); | |
71 | |
72 EXPECT_EQ(0, manager->last_instance_id_removed()); | |
73 | |
74 EXPECT_TRUE(manager->CanUseGuestInstanceID(3)); | |
75 | |
76 manager->AddGuest(3, web_contents3.get()); | |
77 manager->RemoveGuest(1); | |
78 EXPECT_FALSE(manager->CanUseGuestInstanceID(1)); | |
79 EXPECT_FALSE(manager->CanUseGuestInstanceID(2)); | |
80 | |
81 EXPECT_EQ(2, manager->last_instance_id_removed()); | |
82 manager->RemoveGuest(3); | |
83 EXPECT_EQ(3, manager->last_instance_id_removed()); | |
84 | |
85 EXPECT_FALSE(manager->CanUseGuestInstanceID(1)); | |
86 EXPECT_FALSE(manager->CanUseGuestInstanceID(2)); | |
87 EXPECT_FALSE(manager->CanUseGuestInstanceID(3)); | |
88 | |
89 EXPECT_EQ(0, manager->GetNumRemovedInstanceIDs()); | |
90 } | |
91 | |
92 } // namespace extensions | |
OLD | NEW |