Chromium Code Reviews| Index: components/subresource_filter/content/browser/content_subresource_filter_driver_factory_unittest.cc |
| diff --git a/components/subresource_filter/content/browser/content_subresource_filter_driver_factory_unittest.cc b/components/subresource_filter/content/browser/content_subresource_filter_driver_factory_unittest.cc |
| index 814d137987e1f613c2f1928a601738c7e278dfc4..bf02f836e9983663ca604fc5da51ab7804fd1feb 100644 |
| --- a/components/subresource_filter/content/browser/content_subresource_filter_driver_factory_unittest.cc |
| +++ b/components/subresource_filter/content/browser/content_subresource_filter_driver_factory_unittest.cc |
| @@ -189,10 +189,10 @@ class MockSubresourceFilterClient : public SubresourceFilterClient { |
| ~MockSubresourceFilterClient() override = default; |
| - bool ShouldSuppressActivation(content::NavigationHandle* handle) override { |
| - return handle->IsInMainFrame() && |
| - whitelisted_hosts_.find(handle->GetURL().host()) != |
| - whitelisted_hosts_.end(); |
| + bool OnPageActivationComputed(content::NavigationHandle* handle, |
| + bool activated) override { |
| + DCHECK(handle->IsInMainFrame()); |
| + return whitelisted_hosts_.count(handle->GetURL().host()); |
| } |
| void WhitelistByContentSettings(const GURL& url) override {} |
| @@ -544,9 +544,10 @@ TEST_F(ContentSubresourceFilterDriverFactoryTest, |
| NavigateAndExpectActivation({true}, {url}, NO_REDIRECTS_HIT, |
| ActivationDecision::ACTIVATION_DISABLED); |
| + // Whitelisting occurs last, so the decision should still be DISABLED. |
| factory()->client()->WhitelistInCurrentWebContents(url); |
| NavigateAndExpectActivation({true}, {url}, NO_REDIRECTS_HIT, |
| - ActivationDecision::URL_WHITELISTED); |
| + ActivationDecision::ACTIVATION_DISABLED); |
| } |
| TEST_F(ContentSubresourceFilterDriverFactoryTest, NoActivationWhenNoMatch) { |
| @@ -797,9 +798,13 @@ TEST_P(ContentSubresourceFilterDriverFactoryActivationLevelTest, |
| NavigateAndExpectActivation({true}, {url}, NO_REDIRECTS_HIT, |
| test_data.expected_activation_decision); |
| factory()->client()->WhitelistInCurrentWebContents(url); |
| - NavigateAndExpectActivation({true}, {GURL(kExampleUrlWithParams)}, |
| - NO_REDIRECTS_HIT, |
| - ActivationDecision::URL_WHITELISTED); |
| + |
| + // Whitelisting is only applied when the page will otherwise activate. |
|
engedy
2017/05/11 09:13:14
I started deprioritizing ActivationDecision::WHITE
Charlie Harrison
2017/05/11 12:07:15
woohoo!
|
| + ActivationDecision decision = |
| + test_data.activation_level == ActivationLevel::DISABLED |
| + ? test_data.expected_activation_decision |
| + : ActivationDecision::URL_WHITELISTED; |
| + NavigateAndExpectActivation({true}, {url}, NO_REDIRECTS_HIT, decision); |
| } |
| TEST_P(ContentSubresourceFilterDriverFactoryThreatTypeTest, |