OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 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 "base/compiler_specific.h" | |
6 #include "base/memory/scoped_ptr.h" | |
7 #include "base/message_loop.h" | |
8 #include "content/browser/tab_contents/tab_contents.h" | |
9 #include "content/public/browser/web_contents_delegate.h" | |
10 #include "content/test/test_browser_context.h" | |
11 #include "testing/gtest/include/gtest/gtest.h" | |
12 | |
13 namespace { | |
14 | |
15 class MockWebContentsDelegate : public content::WebContentsDelegate { | |
16 public: | |
17 virtual ~MockWebContentsDelegate() {} | |
18 }; | |
19 | |
20 TEST(WebContentsDelegateTest, UnregisterInDestructor) { | |
21 MessageLoop loop(MessageLoop::TYPE_UI); | |
22 TestBrowserContext browser_context; | |
23 | |
24 scoped_ptr<TabContents> contents_a( | |
25 new TabContents(&browser_context, NULL, MSG_ROUTING_NONE, NULL, NULL)); | |
26 scoped_ptr<TabContents> contents_b( | |
27 new TabContents(&browser_context, NULL, MSG_ROUTING_NONE, NULL, NULL)); | |
28 EXPECT_TRUE(contents_a->GetDelegate() == NULL); | |
29 EXPECT_TRUE(contents_b->GetDelegate() == NULL); | |
30 | |
31 scoped_ptr<MockWebContentsDelegate> delegate(new MockWebContentsDelegate()); | |
32 | |
33 // Setting a delegate should work correctly. | |
34 contents_a->SetDelegate(delegate.get()); | |
35 EXPECT_EQ(delegate.get(), contents_a->GetDelegate()); | |
36 EXPECT_TRUE(contents_b->GetDelegate() == NULL); | |
37 | |
38 // A delegate can be a delegate to multiple TabContents. | |
39 contents_b->SetDelegate(delegate.get()); | |
40 EXPECT_EQ(delegate.get(), contents_a->GetDelegate()); | |
41 EXPECT_EQ(delegate.get(), contents_b->GetDelegate()); | |
42 | |
43 // Setting the same delegate multiple times should work correctly. | |
44 contents_b->SetDelegate(delegate.get()); | |
45 EXPECT_EQ(delegate.get(), contents_a->GetDelegate()); | |
46 EXPECT_EQ(delegate.get(), contents_b->GetDelegate()); | |
47 | |
48 // Setting delegate to NULL should work correctly. | |
49 contents_b->SetDelegate(NULL); | |
50 EXPECT_EQ(delegate.get(), contents_a->GetDelegate()); | |
51 EXPECT_TRUE(contents_b->GetDelegate() == NULL); | |
52 | |
53 // Destroying the delegate while it is still the delegate | |
54 // for a TabContents should unregister it. | |
55 contents_b->SetDelegate(delegate.get()); | |
56 EXPECT_EQ(delegate.get(), contents_a->GetDelegate()); | |
57 EXPECT_EQ(delegate.get(), contents_b->GetDelegate()); | |
58 delegate.reset(NULL); | |
59 EXPECT_TRUE(contents_a->GetDelegate() == NULL); | |
60 EXPECT_TRUE(contents_b->GetDelegate() == NULL); | |
61 | |
62 // Destroy the tab contents and run the message loop to prevent leaks. | |
63 contents_a.reset(NULL); | |
64 contents_b.reset(NULL); | |
65 loop.RunAllPending(); | |
66 } | |
67 | |
68 } // namespace | |
OLD | NEW |