Index: chrome/browser/ssl/ssl_browser_tests.cc |
diff --git a/chrome/browser/ssl/ssl_browser_tests.cc b/chrome/browser/ssl/ssl_browser_tests.cc |
index d8f3a8f5d574adf6e028103b2bcd36c15c9544ea..8c2df7cef945de18984990a39ee0ce579a5b6f0d 100644 |
--- a/chrome/browser/ssl/ssl_browser_tests.cc |
+++ b/chrome/browser/ssl/ssl_browser_tests.cc |
@@ -32,9 +32,9 @@ |
#include "chrome/browser/ssl/cert_report_helper.h" |
#include "chrome/browser/ssl/cert_verifier_browser_test.h" |
#include "chrome/browser/ssl/certificate_reporting_test_utils.h" |
-#include "chrome/browser/ssl/chrome_security_state_model_client.h" |
#include "chrome/browser/ssl/chrome_ssl_host_state_delegate.h" |
#include "chrome/browser/ssl/common_name_mismatch_handler.h" |
+#include "chrome/browser/ssl/security_state_tab_helper.h" |
#include "chrome/browser/ssl/ssl_blocking_page.h" |
#include "chrome/browser/ssl/ssl_error_handler.h" |
#include "chrome/browser/ui/browser.h" |
@@ -54,8 +54,8 @@ |
#include "components/prefs/pref_service.h" |
#include "components/security_interstitials/core/controller_client.h" |
#include "components/security_interstitials/core/metrics_helper.h" |
-#include "components/security_state/security_state_model.h" |
-#include "components/security_state/switches.h" |
+#include "components/security_state/core/security_state_model.h" |
+#include "components/security_state/core/switches.h" |
#include "components/ssl_errors/error_classification.h" |
#include "components/variations/variations_associated_data.h" |
#include "components/web_modal/web_contents_modal_dialog_manager.h" |
@@ -162,10 +162,9 @@ namespace SecurityStyle { |
void Check( |
WebContents* tab, |
security_state::SecurityStateModel::SecurityLevel expected_security_level) { |
- ChromeSecurityStateModelClient* model_client = |
- ChromeSecurityStateModelClient::FromWebContents(tab); |
+ SecurityStateTabHelper* helper = SecurityStateTabHelper::FromWebContents(tab); |
security_state::SecurityStateModel::SecurityInfo security_info; |
- model_client->GetSecurityInfo(&security_info); |
+ helper->GetSecurityInfo(&security_info); |
EXPECT_EQ(expected_security_level, security_info.security_level); |
} |
@@ -1201,13 +1200,13 @@ IN_PROC_BROWSER_TEST_F(SSLUITest, MarkFileAsNonSecure) { |
browser()->tab_strip_model()->GetActiveWebContents(); |
ASSERT_TRUE(contents); |
- ChromeSecurityStateModelClient* model_client = |
- ChromeSecurityStateModelClient::FromWebContents(contents); |
- ASSERT_TRUE(model_client); |
+ SecurityStateTabHelper* helper = |
+ SecurityStateTabHelper::FromWebContents(contents); |
+ ASSERT_TRUE(helper); |
ui_test_utils::NavigateToURL(browser(), GURL("file:///")); |
security_state::SecurityStateModel::SecurityInfo security_info; |
- model_client->GetSecurityInfo(&security_info); |
+ helper->GetSecurityInfo(&security_info); |
EXPECT_EQ(security_state::SecurityStateModel::NONE, |
security_info.security_level); |
} |
@@ -1221,13 +1220,13 @@ IN_PROC_BROWSER_TEST_F(SSLUITest, MarkAboutAsNonSecure) { |
browser()->tab_strip_model()->GetActiveWebContents(); |
ASSERT_TRUE(contents); |
- ChromeSecurityStateModelClient* model_client = |
- ChromeSecurityStateModelClient::FromWebContents(contents); |
- ASSERT_TRUE(model_client); |
+ SecurityStateTabHelper* helper = |
+ SecurityStateTabHelper::FromWebContents(contents); |
+ ASSERT_TRUE(helper); |
ui_test_utils::NavigateToURL(browser(), GURL("about:blank")); |
security_state::SecurityStateModel::SecurityInfo security_info; |
- model_client->GetSecurityInfo(&security_info); |
+ helper->GetSecurityInfo(&security_info); |
EXPECT_EQ(security_state::SecurityStateModel::NONE, |
security_info.security_level); |
} |
@@ -1241,13 +1240,13 @@ IN_PROC_BROWSER_TEST_F(SSLUITest, MarkDataAsNonSecure) { |
browser()->tab_strip_model()->GetActiveWebContents(); |
ASSERT_TRUE(contents); |
- ChromeSecurityStateModelClient* model_client = |
- ChromeSecurityStateModelClient::FromWebContents(contents); |
- ASSERT_TRUE(model_client); |
+ SecurityStateTabHelper* helper = |
+ SecurityStateTabHelper::FromWebContents(contents); |
+ ASSERT_TRUE(helper); |
ui_test_utils::NavigateToURL(browser(), GURL("data:text/plain,hello")); |
security_state::SecurityStateModel::SecurityInfo security_info; |
- model_client->GetSecurityInfo(&security_info); |
+ helper->GetSecurityInfo(&security_info); |
EXPECT_EQ(security_state::SecurityStateModel::NONE, |
security_info.security_level); |
} |
@@ -1261,15 +1260,15 @@ IN_PROC_BROWSER_TEST_F(SSLUITest, MarkBlobAsNonSecure) { |
browser()->tab_strip_model()->GetActiveWebContents(); |
ASSERT_TRUE(contents); |
- ChromeSecurityStateModelClient* model_client = |
- ChromeSecurityStateModelClient::FromWebContents(contents); |
- ASSERT_TRUE(model_client); |
+ SecurityStateTabHelper* helper = |
+ SecurityStateTabHelper::FromWebContents(contents); |
+ ASSERT_TRUE(helper); |
ui_test_utils::NavigateToURL( |
browser(), |
GURL("blob:chrome://newtab/49a463bb-fac8-476c-97bf-5d7076c3ea1a")); |
security_state::SecurityStateModel::SecurityInfo security_info; |
- model_client->GetSecurityInfo(&security_info); |
+ helper->GetSecurityInfo(&security_info); |
EXPECT_EQ(security_state::SecurityStateModel::NONE, |
security_info.security_level); |
} |
@@ -2459,11 +2458,10 @@ IN_PROC_BROWSER_TEST_F(SSLUITest, TestUnsafeContentsInWorkerWithUserException) { |
CheckAuthenticationBrokenState(tab, net::CERT_STATUS_COMMON_NAME_INVALID, |
AuthState::NONE); |
- ChromeSecurityStateModelClient* client = |
- ChromeSecurityStateModelClient::FromWebContents(tab); |
- ASSERT_TRUE(client); |
+ SecurityStateTabHelper* helper = SecurityStateTabHelper::FromWebContents(tab); |
+ ASSERT_TRUE(helper); |
security_state::SecurityStateModel::SecurityInfo security_info; |
- client->GetSecurityInfo(&security_info); |
+ helper->GetSecurityInfo(&security_info); |
EXPECT_EQ(security_state::SecurityStateModel::CONTENT_STATUS_NONE, |
security_info.mixed_content_status); |
EXPECT_EQ(security_state::SecurityStateModel::CONTENT_STATUS_NONE, |
@@ -2480,7 +2478,7 @@ IN_PROC_BROWSER_TEST_F(SSLUITest, TestUnsafeContentsInWorkerWithUserException) { |
CheckWorkerLoadResult(tab, true); // Worker loads insecure content |
CheckAuthenticationBrokenState(tab, CertError::NONE, AuthState::NONE); |
- client->GetSecurityInfo(&security_info); |
+ helper->GetSecurityInfo(&security_info); |
EXPECT_EQ(security_state::SecurityStateModel::CONTENT_STATUS_NONE, |
security_info.mixed_content_status); |
EXPECT_EQ(security_state::SecurityStateModel::CONTENT_STATUS_RAN, |
@@ -2495,11 +2493,10 @@ IN_PROC_BROWSER_TEST_F(SSLUITest, TestUnsafeContentsWithUserException) { |
"/ssl/page_with_unsafe_contents.html")); |
CheckAuthenticationBrokenState(tab, CertError::NONE, AuthState::NONE); |
- ChromeSecurityStateModelClient* client = |
- ChromeSecurityStateModelClient::FromWebContents(tab); |
- ASSERT_TRUE(client); |
+ SecurityStateTabHelper* helper = SecurityStateTabHelper::FromWebContents(tab); |
+ ASSERT_TRUE(helper); |
security_state::SecurityStateModel::SecurityInfo security_info; |
- client->GetSecurityInfo(&security_info); |
+ helper->GetSecurityInfo(&security_info); |
EXPECT_EQ(security_state::SecurityStateModel::CONTENT_STATUS_NONE, |
security_info.mixed_content_status); |
EXPECT_EQ( |
@@ -2534,7 +2531,7 @@ IN_PROC_BROWSER_TEST_F(SSLUITest, TestUnsafeContentsWithUserException) { |
CheckAuthenticationBrokenState(tab, net::CERT_STATUS_COMMON_NAME_INVALID, |
AuthState::NONE); |
- client->GetSecurityInfo(&security_info); |
+ helper->GetSecurityInfo(&security_info); |
EXPECT_EQ(security_state::SecurityStateModel::CONTENT_STATUS_NONE, |
security_info.mixed_content_status); |
EXPECT_EQ( |
@@ -2548,11 +2545,10 @@ IN_PROC_BROWSER_TEST_F(SSLUITest, TestUnsafeImageWithUserException) { |
ASSERT_NO_FATAL_FAILURE( |
SetUpUnsafeContentsWithUserException("/ssl/page_with_unsafe_image.html")); |
- ChromeSecurityStateModelClient* client = |
- ChromeSecurityStateModelClient::FromWebContents(tab); |
- ASSERT_TRUE(client); |
+ SecurityStateTabHelper* helper = SecurityStateTabHelper::FromWebContents(tab); |
+ ASSERT_TRUE(helper); |
security_state::SecurityStateModel::SecurityInfo security_info; |
- client->GetSecurityInfo(&security_info); |
+ helper->GetSecurityInfo(&security_info); |
EXPECT_EQ(security_state::SecurityStateModel::CONTENT_STATUS_NONE, |
security_info.mixed_content_status); |
EXPECT_EQ(security_state::SecurityStateModel::CONTENT_STATUS_DISPLAYED, |