| 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 |