| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 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 "content/public/browser/web_contents_user_data.h" | 5 #include "content/public/browser/web_contents_user_data.h" |
| 6 | 6 |
| 7 #include "base/memory/scoped_ptr.h" | 7 #include "base/memory/scoped_ptr.h" |
| 8 #include "content/public/browser/web_contents.h" | 8 #include "content/public/browser/web_contents.h" |
| 9 #include "content/public/test/test_renderer_host.h" | 9 #include "content/public/test/test_renderer_host.h" |
| 10 #include "content/public/test/web_contents_tester.h" | 10 #include "content/public/test/web_contents_tester.h" |
| 11 #include "testing/gtest/include/gtest/gtest.h" | 11 #include "testing/gtest/include/gtest/gtest.h" |
| 12 | 12 |
| 13 namespace content { |
| 14 |
| 13 class WebContentsAttachedClass1 | 15 class WebContentsAttachedClass1 |
| 14 : public content::WebContentsUserData<WebContentsAttachedClass1> { | 16 : public WebContentsUserData<WebContentsAttachedClass1> { |
| 15 public: | 17 public: |
| 16 virtual ~WebContentsAttachedClass1() {} | 18 virtual ~WebContentsAttachedClass1() {} |
| 17 private: | 19 private: |
| 18 explicit WebContentsAttachedClass1(content::WebContents* contents) {} | 20 explicit WebContentsAttachedClass1(WebContents* contents) {} |
| 19 friend class content::WebContentsUserData<WebContentsAttachedClass1>; | 21 friend class WebContentsUserData<WebContentsAttachedClass1>; |
| 20 }; | 22 }; |
| 21 | 23 |
| 22 class WebContentsAttachedClass2 | 24 class WebContentsAttachedClass2 |
| 23 : public content::WebContentsUserData<WebContentsAttachedClass2> { | 25 : public WebContentsUserData<WebContentsAttachedClass2> { |
| 24 public: | 26 public: |
| 25 virtual ~WebContentsAttachedClass2() {} | 27 virtual ~WebContentsAttachedClass2() {} |
| 26 private: | 28 private: |
| 27 explicit WebContentsAttachedClass2(content::WebContents* contents) {} | 29 explicit WebContentsAttachedClass2(WebContents* contents) {} |
| 28 friend class content::WebContentsUserData<WebContentsAttachedClass2>; | 30 friend class WebContentsUserData<WebContentsAttachedClass2>; |
| 29 }; | 31 }; |
| 30 | 32 |
| 31 DEFINE_WEB_CONTENTS_USER_DATA_KEY(WebContentsAttachedClass1) | 33 DEFINE_WEB_CONTENTS_USER_DATA_KEY(WebContentsAttachedClass1) |
| 32 DEFINE_WEB_CONTENTS_USER_DATA_KEY(WebContentsAttachedClass2) | 34 DEFINE_WEB_CONTENTS_USER_DATA_KEY(WebContentsAttachedClass2) |
| 33 | 35 |
| 34 typedef content::RenderViewHostTestHarness WebContentsUserDataTest; | 36 typedef RenderViewHostTestHarness WebContentsUserDataTest; |
| 35 | 37 |
| 36 TEST_F(WebContentsUserDataTest, OneInstanceTwoAttachments) { | 38 TEST_F(WebContentsUserDataTest, OneInstanceTwoAttachments) { |
| 37 content::WebContents* contents = web_contents(); | 39 WebContents* contents = web_contents(); |
| 38 WebContentsAttachedClass1* class1 = | 40 WebContentsAttachedClass1* class1 = |
| 39 WebContentsAttachedClass1::FromWebContents(contents); | 41 WebContentsAttachedClass1::FromWebContents(contents); |
| 40 ASSERT_EQ(NULL, class1); | 42 ASSERT_EQ(NULL, class1); |
| 41 WebContentsAttachedClass2* class2 = | 43 WebContentsAttachedClass2* class2 = |
| 42 WebContentsAttachedClass2::FromWebContents(contents); | 44 WebContentsAttachedClass2::FromWebContents(contents); |
| 43 ASSERT_EQ(NULL, class2); | 45 ASSERT_EQ(NULL, class2); |
| 44 | 46 |
| 45 WebContentsAttachedClass1::CreateForWebContents(contents); | 47 WebContentsAttachedClass1::CreateForWebContents(contents); |
| 46 class1 = WebContentsAttachedClass1::FromWebContents(contents); | 48 class1 = WebContentsAttachedClass1::FromWebContents(contents); |
| 47 ASSERT_TRUE(class1 != NULL); | 49 ASSERT_TRUE(class1 != NULL); |
| 48 class2 = WebContentsAttachedClass2::FromWebContents(contents); | 50 class2 = WebContentsAttachedClass2::FromWebContents(contents); |
| 49 ASSERT_EQ(NULL, class2); | 51 ASSERT_EQ(NULL, class2); |
| 50 | 52 |
| 51 WebContentsAttachedClass2::CreateForWebContents(contents); | 53 WebContentsAttachedClass2::CreateForWebContents(contents); |
| 52 WebContentsAttachedClass1* class1again = | 54 WebContentsAttachedClass1* class1again = |
| 53 WebContentsAttachedClass1::FromWebContents(contents); | 55 WebContentsAttachedClass1::FromWebContents(contents); |
| 54 ASSERT_TRUE(class1again != NULL); | 56 ASSERT_TRUE(class1again != NULL); |
| 55 class2 = WebContentsAttachedClass2::FromWebContents(contents); | 57 class2 = WebContentsAttachedClass2::FromWebContents(contents); |
| 56 ASSERT_TRUE(class2 != NULL); | 58 ASSERT_TRUE(class2 != NULL); |
| 57 ASSERT_EQ(class1, class1again); | 59 ASSERT_EQ(class1, class1again); |
| 58 ASSERT_NE(static_cast<void*>(class1), static_cast<void*>(class2)); | 60 ASSERT_NE(static_cast<void*>(class1), static_cast<void*>(class2)); |
| 59 } | 61 } |
| 60 | 62 |
| 61 TEST_F(WebContentsUserDataTest, TwoInstancesOneAttachment) { | 63 TEST_F(WebContentsUserDataTest, TwoInstancesOneAttachment) { |
| 62 content::WebContents* contents1 = web_contents(); | 64 WebContents* contents1 = web_contents(); |
| 63 scoped_ptr<content::WebContents> contents2( | 65 scoped_ptr<WebContents> contents2( |
| 64 content::WebContentsTester::CreateTestWebContents( | 66 WebContentsTester::CreateTestWebContents(browser_context(), NULL)); |
| 65 browser_context(), NULL)); | |
| 66 | 67 |
| 67 WebContentsAttachedClass1* one_class = | 68 WebContentsAttachedClass1* one_class = |
| 68 WebContentsAttachedClass1::FromWebContents(contents1); | 69 WebContentsAttachedClass1::FromWebContents(contents1); |
| 69 ASSERT_EQ(NULL, one_class); | 70 ASSERT_EQ(NULL, one_class); |
| 70 WebContentsAttachedClass1* two_class = | 71 WebContentsAttachedClass1* two_class = |
| 71 WebContentsAttachedClass1::FromWebContents(contents2.get()); | 72 WebContentsAttachedClass1::FromWebContents(contents2.get()); |
| 72 ASSERT_EQ(NULL, two_class); | 73 ASSERT_EQ(NULL, two_class); |
| 73 | 74 |
| 74 WebContentsAttachedClass1::CreateForWebContents(contents1); | 75 WebContentsAttachedClass1::CreateForWebContents(contents1); |
| 75 one_class = WebContentsAttachedClass1::FromWebContents(contents1); | 76 one_class = WebContentsAttachedClass1::FromWebContents(contents1); |
| 76 ASSERT_TRUE(one_class != NULL); | 77 ASSERT_TRUE(one_class != NULL); |
| 77 two_class = WebContentsAttachedClass1::FromWebContents(contents2.get()); | 78 two_class = WebContentsAttachedClass1::FromWebContents(contents2.get()); |
| 78 ASSERT_EQ(NULL, two_class); | 79 ASSERT_EQ(NULL, two_class); |
| 79 | 80 |
| 80 WebContentsAttachedClass1::CreateForWebContents(contents2.get()); | 81 WebContentsAttachedClass1::CreateForWebContents(contents2.get()); |
| 81 WebContentsAttachedClass1* one_class_again = | 82 WebContentsAttachedClass1* one_class_again = |
| 82 WebContentsAttachedClass1::FromWebContents(contents1); | 83 WebContentsAttachedClass1::FromWebContents(contents1); |
| 83 ASSERT_TRUE(one_class_again != NULL); | 84 ASSERT_TRUE(one_class_again != NULL); |
| 84 two_class = WebContentsAttachedClass1::FromWebContents(contents2.get()); | 85 two_class = WebContentsAttachedClass1::FromWebContents(contents2.get()); |
| 85 ASSERT_TRUE(two_class != NULL); | 86 ASSERT_TRUE(two_class != NULL); |
| 86 ASSERT_EQ(one_class, one_class_again); | 87 ASSERT_EQ(one_class, one_class_again); |
| 87 ASSERT_NE(one_class, two_class); | 88 ASSERT_NE(one_class, two_class); |
| 88 } | 89 } |
| 89 | 90 |
| 90 TEST_F(WebContentsUserDataTest, Idempotence) { | 91 TEST_F(WebContentsUserDataTest, Idempotence) { |
| 91 content::WebContents* contents = web_contents(); | 92 WebContents* contents = web_contents(); |
| 92 WebContentsAttachedClass1* clazz = | 93 WebContentsAttachedClass1* clazz = |
| 93 WebContentsAttachedClass1::FromWebContents(contents); | 94 WebContentsAttachedClass1::FromWebContents(contents); |
| 94 ASSERT_EQ(NULL, clazz); | 95 ASSERT_EQ(NULL, clazz); |
| 95 | 96 |
| 96 WebContentsAttachedClass1::CreateForWebContents(contents); | 97 WebContentsAttachedClass1::CreateForWebContents(contents); |
| 97 clazz = WebContentsAttachedClass1::FromWebContents(contents); | 98 clazz = WebContentsAttachedClass1::FromWebContents(contents); |
| 98 ASSERT_TRUE(clazz != NULL); | 99 ASSERT_TRUE(clazz != NULL); |
| 99 | 100 |
| 100 WebContentsAttachedClass1::CreateForWebContents(contents); | 101 WebContentsAttachedClass1::CreateForWebContents(contents); |
| 101 WebContentsAttachedClass1* class_again = | 102 WebContentsAttachedClass1* class_again = |
| 102 WebContentsAttachedClass1::FromWebContents(contents); | 103 WebContentsAttachedClass1::FromWebContents(contents); |
| 103 ASSERT_TRUE(class_again != NULL); | 104 ASSERT_TRUE(class_again != NULL); |
| 104 ASSERT_EQ(clazz, class_again); | 105 ASSERT_EQ(clazz, class_again); |
| 105 } | 106 } |
| 107 |
| 108 } // namespace content |
| OLD | NEW |