OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 "chrome/browser/interstitials/security_interstitial_page_test_utils.h" | 5 #include "chrome/browser/interstitials/security_interstitial_page_test_utils.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "base/prefs/pref_service.h" | 9 #include "base/prefs/pref_service.h" |
10 #include "base/strings/stringprintf.h" | 10 #include "base/strings/stringprintf.h" |
11 #include "chrome/browser/interstitials/security_interstitial_page.h" | 11 #include "chrome/browser/interstitials/security_interstitial_page.h" |
12 #include "chrome/browser/profiles/profile.h" | 12 #include "chrome/browser/profiles/profile.h" |
13 #include "chrome/browser/ui/browser.h" | 13 #include "chrome/browser/ui/browser.h" |
14 #include "chrome/browser/ui/tabs/tab_strip_model.h" | 14 #include "chrome/browser/ui/tabs/tab_strip_model.h" |
15 #include "chrome/common/pref_names.h" | 15 #include "chrome/common/pref_names.h" |
| 16 #include "components/security_interstitials/core/controller_client.h" |
16 #include "content/public/browser/interstitial_page.h" | 17 #include "content/public/browser/interstitial_page.h" |
17 #include "content/public/browser/web_contents.h" | 18 #include "content/public/browser/web_contents.h" |
18 #include "content/public/test/browser_test_utils.h" | 19 #include "content/public/test/browser_test_utils.h" |
19 #include "testing/gtest/include/gtest/gtest.h" | 20 #include "testing/gtest/include/gtest/gtest.h" |
20 #include "url/gurl.h" | 21 #include "url/gurl.h" |
21 | 22 |
22 namespace chrome_browser_interstitials { | 23 namespace chrome_browser_interstitials { |
23 | 24 |
24 bool IsInterstitialDisplayingText( | 25 bool IsInterstitialDisplayingText( |
25 const content::InterstitialPage* const interstitial, | 26 const content::InterstitialPage* const interstitial, |
26 const std::string& text) { | 27 const std::string& text) { |
27 // It's valid for |text| to contain "\'", but simply look for "'" instead | 28 // It's valid for |text| to contain "\'", but simply look for "'" instead |
28 // since this function is used for searching host names and a predefined | 29 // since this function is used for searching host names and a predefined |
29 // string. | 30 // string. |
30 DCHECK(text.find("\'") == std::string::npos); | 31 DCHECK(text.find("\'") == std::string::npos); |
31 std::string command = base::StringPrintf( | 32 std::string command = base::StringPrintf( |
32 "var hasText = document.body.textContent.indexOf('%s') >= 0;" | 33 "var hasText = document.body.textContent.indexOf('%s') >= 0;" |
33 "window.domAutomationController.send(hasText ? %d : %d);", | 34 "window.domAutomationController.send(hasText ? %d : %d);", |
34 text.c_str(), SecurityInterstitialPage::CMD_TEXT_FOUND, | 35 text.c_str(), security_interstitials::CMD_TEXT_FOUND, |
35 SecurityInterstitialPage::CMD_TEXT_NOT_FOUND); | 36 security_interstitials::CMD_TEXT_NOT_FOUND); |
36 int result = 0; | 37 int result = 0; |
37 EXPECT_TRUE(content::ExecuteScriptAndExtractInt(interstitial->GetMainFrame(), | 38 EXPECT_TRUE(content::ExecuteScriptAndExtractInt(interstitial->GetMainFrame(), |
38 command, &result)); | 39 command, &result)); |
39 return result == SecurityInterstitialPage::CMD_TEXT_FOUND; | 40 return result == security_interstitials::CMD_TEXT_FOUND; |
40 } | 41 } |
41 | 42 |
42 void SecurityInterstitialIDNTest::SetUpOnMainThread() { | 43 void SecurityInterstitialIDNTest::SetUpOnMainThread() { |
43 // Clear AcceptLanguages to force punycode decoding. | 44 // Clear AcceptLanguages to force punycode decoding. |
44 browser()->profile()->GetPrefs()->SetString(prefs::kAcceptLanguages, | 45 browser()->profile()->GetPrefs()->SetString(prefs::kAcceptLanguages, |
45 std::string()); | 46 std::string()); |
46 } | 47 } |
47 | 48 |
48 testing::AssertionResult SecurityInterstitialIDNTest::VerifyIDNDecoded() const { | 49 testing::AssertionResult SecurityInterstitialIDNTest::VerifyIDNDecoded() const { |
49 const char kHostname[] = "xn--d1abbgf6aiiy.xn--p1ai"; | 50 const char kHostname[] = "xn--d1abbgf6aiiy.xn--p1ai"; |
(...skipping 14 matching lines...) Expand all Loading... |
64 if (!WaitForRenderFrameReady(contents->GetInterstitialPage()->GetMainFrame())) | 65 if (!WaitForRenderFrameReady(contents->GetInterstitialPage()->GetMainFrame())) |
65 return testing::AssertionFailure() << "Render frame not ready"; | 66 return testing::AssertionFailure() << "Render frame not ready"; |
66 | 67 |
67 if (IsInterstitialDisplayingText(contents->GetInterstitialPage(), | 68 if (IsInterstitialDisplayingText(contents->GetInterstitialPage(), |
68 kHostnameJSUnicode)) { | 69 kHostnameJSUnicode)) { |
69 return testing::AssertionSuccess(); | 70 return testing::AssertionSuccess(); |
70 } | 71 } |
71 return testing::AssertionFailure() << "Interstitial not displaying text"; | 72 return testing::AssertionFailure() << "Interstitial not displaying text"; |
72 } | 73 } |
73 } // namespace chrome_browser_interstitials | 74 } // namespace chrome_browser_interstitials |
OLD | NEW |