| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 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/renderer/peripheral_content_heuristic.h" | 5 #include "content/renderer/peripheral_content_heuristic.h" |
| 6 | 6 |
| 7 #include "testing/gtest/include/gtest/gtest.h" | 7 #include "testing/gtest/include/gtest/gtest.h" |
| 8 #include "ui/gfx/geometry/size.h" | |
| 9 #include "url/gurl.h" | 8 #include "url/gurl.h" |
| 10 #include "url/origin.h" | 9 #include "url/origin.h" |
| 11 | 10 |
| 12 namespace content { | 11 namespace content { |
| 13 | 12 |
| 14 namespace { | 13 namespace { |
| 15 | 14 |
| 16 const char kSameOrigin[] = "http://same.com"; | 15 const char kSameOrigin[] = "http://same.com"; |
| 17 const char kOtherOrigin[] = "http://other.com"; | 16 const char kOtherOrigin[] = "http://other.com"; |
| 18 | 17 |
| 19 } // namespaces | 18 } // namespaces |
| 20 | 19 |
| 21 TEST(PeripheralContentHeuristic, AllowSameOrigin) { | 20 TEST(PeripheralContentHeuristic, AllowSameOrigin) { |
| 22 EXPECT_EQ(RenderFrame::CONTENT_STATUS_ESSENTIAL_SAME_ORIGIN, | 21 EXPECT_EQ( |
| 23 PeripheralContentHeuristic::GetPeripheralStatus( | 22 PeripheralContentHeuristic::HEURISTIC_DECISION_ESSENTIAL_SAME_ORIGIN, |
| 24 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), | 23 PeripheralContentHeuristic::GetPeripheralStatus( |
| 25 url::Origin(GURL(kSameOrigin)), gfx::Size(100, 100))); | 24 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), |
| 26 EXPECT_EQ(RenderFrame::CONTENT_STATUS_ESSENTIAL_SAME_ORIGIN, | 25 url::Origin(GURL(kSameOrigin)), 100, 100)); |
| 27 PeripheralContentHeuristic::GetPeripheralStatus( | 26 EXPECT_EQ( |
| 28 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), | 27 PeripheralContentHeuristic::HEURISTIC_DECISION_ESSENTIAL_SAME_ORIGIN, |
| 29 url::Origin(GURL(kSameOrigin)), gfx::Size(1000, 1000))); | 28 PeripheralContentHeuristic::GetPeripheralStatus( |
| 29 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), |
| 30 url::Origin(GURL(kSameOrigin)), 1000, 1000)); |
| 30 } | 31 } |
| 31 | 32 |
| 32 TEST(PeripheralContentHeuristic, DisallowCrossOriginUnlessLarge) { | 33 TEST(PeripheralContentHeuristic, DisallowCrossOriginUnlessLarge) { |
| 33 EXPECT_EQ(RenderFrame::CONTENT_STATUS_PERIPHERAL, | 34 EXPECT_EQ(PeripheralContentHeuristic::HEURISTIC_DECISION_PERIPHERAL, |
| 34 PeripheralContentHeuristic::GetPeripheralStatus( | 35 PeripheralContentHeuristic::GetPeripheralStatus( |
| 35 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), | 36 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), |
| 36 url::Origin(GURL(kOtherOrigin)), gfx::Size(100, 100))); | 37 url::Origin(GURL(kOtherOrigin)), 100, 100)); |
| 37 EXPECT_EQ(RenderFrame::CONTENT_STATUS_ESSENTIAL_CROSS_ORIGIN_BIG, | 38 EXPECT_EQ( |
| 38 PeripheralContentHeuristic::GetPeripheralStatus( | 39 PeripheralContentHeuristic::HEURISTIC_DECISION_ESSENTIAL_CROSS_ORIGIN_BIG, |
| 39 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), | 40 PeripheralContentHeuristic::GetPeripheralStatus( |
| 40 url::Origin(GURL(kOtherOrigin)), gfx::Size(1000, 1000))); | 41 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), |
| 42 url::Origin(GURL(kOtherOrigin)), 1000, 1000)); |
| 41 } | 43 } |
| 42 | 44 |
| 43 TEST(PeripheralContentHeuristic, AlwaysAllowTinyContent) { | 45 TEST(PeripheralContentHeuristic, AlwaysAllowTinyContent) { |
| 44 EXPECT_EQ(RenderFrame::CONTENT_STATUS_ESSENTIAL_SAME_ORIGIN, | 46 EXPECT_EQ( |
| 47 PeripheralContentHeuristic::HEURISTIC_DECISION_ESSENTIAL_SAME_ORIGIN, |
| 48 PeripheralContentHeuristic::GetPeripheralStatus( |
| 49 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), |
| 50 url::Origin(GURL(kSameOrigin)), 1, 1)); |
| 51 EXPECT_EQ(PeripheralContentHeuristic:: |
| 52 HEURISTIC_DECISION_ESSENTIAL_CROSS_ORIGIN_TINY, |
| 45 PeripheralContentHeuristic::GetPeripheralStatus( | 53 PeripheralContentHeuristic::GetPeripheralStatus( |
| 46 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), | 54 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), |
| 47 url::Origin(GURL(kSameOrigin)), gfx::Size(1, 1))); | 55 url::Origin(GURL(kOtherOrigin)), 1, 1)); |
| 48 EXPECT_EQ(RenderFrame::CONTENT_STATUS_ESSENTIAL_CROSS_ORIGIN_TINY, | 56 EXPECT_EQ(PeripheralContentHeuristic:: |
| 57 HEURISTIC_DECISION_ESSENTIAL_CROSS_ORIGIN_TINY, |
| 49 PeripheralContentHeuristic::GetPeripheralStatus( | 58 PeripheralContentHeuristic::GetPeripheralStatus( |
| 50 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), | 59 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), |
| 51 url::Origin(GURL(kOtherOrigin)), gfx::Size(1, 1))); | 60 url::Origin(GURL(kOtherOrigin)), 5, 5)); |
| 52 EXPECT_EQ(RenderFrame::CONTENT_STATUS_ESSENTIAL_CROSS_ORIGIN_TINY, | 61 EXPECT_EQ(PeripheralContentHeuristic::HEURISTIC_DECISION_PERIPHERAL, |
| 53 PeripheralContentHeuristic::GetPeripheralStatus( | 62 PeripheralContentHeuristic::GetPeripheralStatus( |
| 54 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), | 63 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), |
| 55 url::Origin(GURL(kOtherOrigin)), gfx::Size(5, 5))); | 64 url::Origin(GURL(kOtherOrigin)), 10, 10)); |
| 56 EXPECT_EQ(RenderFrame::CONTENT_STATUS_PERIPHERAL, | |
| 57 PeripheralContentHeuristic::GetPeripheralStatus( | |
| 58 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), | |
| 59 url::Origin(GURL(kOtherOrigin)), gfx::Size(10, 10))); | |
| 60 } | 65 } |
| 61 | 66 |
| 62 TEST(PeripheralContentHeuristic, TemporaryOriginWhitelist) { | 67 TEST(PeripheralContentHeuristic, TemporaryOriginWhitelist) { |
| 63 EXPECT_EQ(RenderFrame::CONTENT_STATUS_PERIPHERAL, | 68 EXPECT_EQ(PeripheralContentHeuristic::HEURISTIC_DECISION_PERIPHERAL, |
| 64 PeripheralContentHeuristic::GetPeripheralStatus( | 69 PeripheralContentHeuristic::GetPeripheralStatus( |
| 65 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), | 70 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), |
| 66 url::Origin(GURL(kOtherOrigin)), gfx::Size(100, 100))); | 71 url::Origin(GURL(kOtherOrigin)), 100, 100)); |
| 67 | 72 |
| 68 std::set<url::Origin> whitelist; | 73 std::set<url::Origin> whitelist; |
| 69 whitelist.insert(url::Origin(GURL(kOtherOrigin))); | 74 whitelist.insert(url::Origin(GURL(kOtherOrigin))); |
| 70 | 75 |
| 71 EXPECT_EQ(RenderFrame::CONTENT_STATUS_ESSENTIAL_CROSS_ORIGIN_WHITELISTED, | 76 EXPECT_EQ(PeripheralContentHeuristic:: |
| 77 HEURISTIC_DECISION_ESSENTIAL_CROSS_ORIGIN_WHITELISTED, |
| 72 PeripheralContentHeuristic::GetPeripheralStatus( | 78 PeripheralContentHeuristic::GetPeripheralStatus( |
| 73 whitelist, url::Origin(GURL(kSameOrigin)), | 79 whitelist, url::Origin(GURL(kSameOrigin)), |
| 74 url::Origin(GURL(kOtherOrigin)), gfx::Size(100, 100))); | 80 url::Origin(GURL(kOtherOrigin)), 100, 100)); |
| 75 } | |
| 76 | |
| 77 TEST(PeripheralContentHeuristic, UndefinedSize) { | |
| 78 EXPECT_EQ(RenderFrame::CONTENT_STATUS_ESSENTIAL_SAME_ORIGIN, | |
| 79 PeripheralContentHeuristic::GetPeripheralStatus( | |
| 80 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), | |
| 81 url::Origin(GURL(kSameOrigin)), gfx::Size())); | |
| 82 | |
| 83 std::set<url::Origin> whitelist; | |
| 84 whitelist.insert(url::Origin(GURL(kOtherOrigin))); | |
| 85 EXPECT_EQ(RenderFrame::CONTENT_STATUS_ESSENTIAL_CROSS_ORIGIN_WHITELISTED, | |
| 86 PeripheralContentHeuristic::GetPeripheralStatus( | |
| 87 whitelist, url::Origin(GURL(kSameOrigin)), | |
| 88 url::Origin(GURL(kOtherOrigin)), gfx::Size())); | |
| 89 | |
| 90 EXPECT_EQ(RenderFrame::CONTENT_STATUS_ESSENTIAL_UNKNOWN_SIZE, | |
| 91 PeripheralContentHeuristic::GetPeripheralStatus( | |
| 92 std::set<url::Origin>(), url::Origin(GURL(kSameOrigin)), | |
| 93 url::Origin(GURL(kOtherOrigin)), gfx::Size())); | |
| 94 } | 81 } |
| 95 | 82 |
| 96 } // namespace content | 83 } // namespace content |
| OLD | NEW |