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 "components/guest_view/browser/guest_view_manager.h"
6
7 #include <memory>
8 #include <utility>
9
10 #include "base/macros.h"
11 #include "components/guest_view/browser/guest_view_manager_delegate.h"
12 #include "components/guest_view/browser/test_guest_view_manager.h"
13 #include "content/public/test/test_renderer_host.h"
14 #include "content/public/test/web_contents_tester.h"
15 #include "testing/gtest/include/gtest/gtest.h"
16
17 using content::WebContents;
18 using content::WebContentsTester;
19
20 namespace guest_view {
21
22 namespace {
23
24 class GuestViewManagerTest : public content::RenderViewHostTestHarness {
25 public:
GuestViewManagerTest()26 GuestViewManagerTest() {}
~GuestViewManagerTest()27 ~GuestViewManagerTest() override {}
28
CreateWebContents()29 std::unique_ptr<WebContents> CreateWebContents() {
30 return WebContentsTester::CreateTestWebContents(browser_context(), nullptr);
31 }
32
33 private:
34 DISALLOW_COPY_AND_ASSIGN(GuestViewManagerTest);
35 };
36
37 } // namespace
38
TEST_F(GuestViewManagerTest,AddRemove)39 TEST_F(GuestViewManagerTest, AddRemove) {
40 std::unique_ptr<GuestViewManagerDelegate> delegate(
41 new GuestViewManagerDelegate());
42 std::unique_ptr<TestGuestViewManager> manager(
43 new TestGuestViewManager(browser_context(), std::move(delegate)));
44
45 std::unique_ptr<WebContents> web_contents1(CreateWebContents());
46 std::unique_ptr<WebContents> web_contents2(CreateWebContents());
47 std::unique_ptr<WebContents> web_contents3(CreateWebContents());
48
49 EXPECT_EQ(0, manager->last_instance_id_removed());
50
51 EXPECT_TRUE(manager->CanUseGuestInstanceID(1));
52 EXPECT_TRUE(manager->CanUseGuestInstanceID(2));
53 EXPECT_TRUE(manager->CanUseGuestInstanceID(3));
54
55 manager->AddGuest(1, web_contents1.get());
56 manager->AddGuest(2, web_contents2.get());
57 manager->RemoveGuest(2);
58
59 // Since we removed 2, it would be an invalid ID.
60 EXPECT_TRUE(manager->CanUseGuestInstanceID(1));
61 EXPECT_FALSE(manager->CanUseGuestInstanceID(2));
62 EXPECT_TRUE(manager->CanUseGuestInstanceID(3));
63
64 EXPECT_EQ(0, manager->last_instance_id_removed());
65
66 EXPECT_TRUE(manager->CanUseGuestInstanceID(3));
67
68 manager->AddGuest(3, web_contents3.get());
69 manager->RemoveGuest(1);
70 EXPECT_FALSE(manager->CanUseGuestInstanceID(1));
71 EXPECT_FALSE(manager->CanUseGuestInstanceID(2));
72
73 EXPECT_EQ(2, manager->last_instance_id_removed());
74 manager->RemoveGuest(3);
75 EXPECT_EQ(3, manager->last_instance_id_removed());
76
77 EXPECT_FALSE(manager->CanUseGuestInstanceID(1));
78 EXPECT_FALSE(manager->CanUseGuestInstanceID(2));
79 EXPECT_FALSE(manager->CanUseGuestInstanceID(3));
80
81 EXPECT_EQ(0u, manager->GetNumRemovedInstanceIDs());
82 }
83
84 } // namespace guest_view
85