[subresource_filter] Implement the "Smart" UI on Android

components/subresource_filter/content/browser/content_subresource_filter_driver_factory_unittest.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/subresource_filter/content/browser/content_subresource_filter_driver_factory.h"
 
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/metrics/field_trial.h"
 #include "base/test/histogram_tester.h"
@@ skipping 171 matching lines @@
   MockSubresourceFilterClient() {}
 
   ~MockSubresourceFilterClient() override = default;
 
   bool IsWhitelistedByContentSettings(const GURL& url) override {
     return false;
   }
 
   void WhitelistByContentSettings(const GURL& url) override {}
 
-  MOCK_METHOD1(ToggleNotificationVisibility, void(bool));
+  MOCK_METHOD2(ToggleNotificationVisibility, void(const GURL&, bool));
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockSubresourceFilterClient);
 };
 
 }  // namespace
 
 class ContentSubresourceFilterDriverFactoryTest
     : public content::RenderViewHostTestHarness {
  public:
@@ skipping 51 matching lines @@
       const std::vector<GURL>& navigation_chain,
       safe_browsing::SBThreatType threat_type,
       safe_browsing::ThreatPatternType threat_type_metadata,
       const content::Referrer& referrer,
       ui::PageTransition transition,
       RedirectChainMatchPattern expected_pattern,
       ActivationDecision expected_activation_decision) {
     const bool expected_activation =
         expected_activation_decision == ActivationDecision::ACTIVATED;
     base::HistogramTester tester;
-    EXPECT_CALL(*client(), ToggleNotificationVisibility(false)).Times(1);
+    EXPECT_CALL(*client(),
+                ToggleNotificationVisibility(navigation_chain.front(), false))
+        .Times(1);
 
     std::unique_ptr<content::NavigationSimulator> navigation_simulator =
         content::NavigationSimulator::CreateRendererInitiated(
             navigation_chain.front(), main_rfh());
     navigation_simulator->SetReferrer(referrer);
     navigation_simulator->SetTransition(transition);
     navigation_simulator->Start();
 
     if (blacklisted_urls.front()) {
       factory()->OnMainResourceMatchedSafeBrowsingBlacklist(
@@ skipping 37 matching lines @@
       // Check that we recorded only what is needed.
       EXPECT_EQ(1u, all_pattern);
       EXPECT_EQ(1u, all_chain_size);
     } else {
       EXPECT_EQ(0u, all_pattern);
       EXPECT_EQ(0u, all_chain_size);
     }
   }
 
   void NavigateAndCommitSubframe(const GURL& url, bool expected_activation) {
-    EXPECT_CALL(*client(), ToggleNotificationVisibility(::testing::_)).Times(0);
+    EXPECT_CALL(*client(),
+                ToggleNotificationVisibility(::testing::_, ::testing::_))
+        .Times(0);
 
     content::NavigationSimulator::NavigateAndCommitFromDocument(
         url, GetSubframeRFH());
     ExpectActivationSignalForFrame(GetSubframeRFH(), expected_activation);
     ::testing::Mock::VerifyAndClearExpectations(client());
   }
 
   void NavigateAndExpectActivation(
       const std::vector<bool>& blacklisted_urls,
       const std::vector<GURL>& navigation_chain,
@@ skipping 26 matching lines @@
   }
 
   void EmulateDidDisallowFirstSubresourceMessage() {
     factory()->OnMessageReceived(
         SubresourceFilterHostMsg_DidDisallowFirstSubresource(
             main_rfh()->GetRoutingID()),
         main_rfh());
   }
 
   void EmulateFailedNavigationAndExpectNoActivation(const GURL& url) {
-    EXPECT_CALL(*client(), ToggleNotificationVisibility(false)).Times(1);
+    EXPECT_CALL(*client(), ToggleNotificationVisibility(url, false)).Times(1);
 
     // With browser-side navigation enabled, ReadyToCommitNavigation is invoked
     // even for failed navigations. This is correctly simulated by
     // NavigationSimulator. Make sure no activation message is sent in this
     // case.
     content::NavigationSimulator::NavigateAndFailFromDocument(
         url, net::ERR_TIMED_OUT, main_rfh());
     ExpectActivationSignalForFrame(main_rfh(), false);
     ::testing::Mock::VerifyAndClearExpectations(client());
   }
 
   void EmulateInPageNavigation(
       const std::vector<bool>& blacklisted_urls,
       RedirectChainMatchPattern expected_pattern,
       ActivationDecision expected_activation_decision) {
     // This test examines the navigation with the following sequence of events:
     //   DidStartProvisional(main, "example.com")
     //   ReadyToCommitNavigation(“example.com”)
     //   DidCommitProvisional(main, "example.com")
     //   DidStartProvisional(sub, "example.com/login")
     //   DidCommitProvisional(sub, "example.com/login")
     //   DidCommitProvisional(main, "example.com#ref")
 
     NavigateAndExpectActivation(blacklisted_urls, {GURL(kExampleUrl)},
                                 expected_pattern, expected_activation_decision);
-    EXPECT_CALL(*client(), ToggleNotificationVisibility(::testing::_)).Times(0);
+    EXPECT_CALL(*client(),
+                ToggleNotificationVisibility(::testing::_, ::testing::_))
+        .Times(0);
     std::unique_ptr<content::NavigationSimulator> navigation_simulator =
         content::NavigationSimulator::CreateRendererInitiated(GURL(kExampleUrl),
                                                               main_rfh());
     navigation_simulator->CommitSameDocument();
     ExpectActivationSignalForFrame(main_rfh(), false);
     ::testing::Mock::VerifyAndClearExpectations(client());
   }
 
  private:
   static bool expected_measure_performance() {
@@ skipping 211 matching lines @@
 #endif
   }
 }
 
 TEST_F(ContentSubresourceFilterDriverFactoryTest, NotificationVisibility) {
   base::FieldTrialList field_trial_list(nullptr);
   testing::ScopedSubresourceFilterFeatureToggle scoped_feature_toggle(
       base::FeatureList::OVERRIDE_ENABLE_FEATURE, kActivationLevelEnabled,
       kActivationScopeAllSites);
 
-  NavigateAndExpectActivation({false}, {GURL(kExampleUrl)}, EMPTY,
+  GURL url(kExampleUrl);
+  NavigateAndExpectActivation({false}, {url}, EMPTY,
                               ActivationDecision::ACTIVATED);
-  EXPECT_CALL(*client(), ToggleNotificationVisibility(true)).Times(1);
+  EXPECT_CALL(*client(), ToggleNotificationVisibility(url, true)).Times(1);
   EmulateDidDisallowFirstSubresourceMessage();
 }
 
 TEST_F(ContentSubresourceFilterDriverFactoryTest,
        SuppressNotificationVisibility) {
   base::FieldTrialList field_trial_list(nullptr);
   testing::ScopedSubresourceFilterFeatureToggle scoped_feature_toggle(
       base::FeatureList::OVERRIDE_ENABLE_FEATURE, kActivationLevelEnabled,
       kActivationScopeAllSites, "" /* activation_lists */,
       "" /* performance_measurement_rate */,
       "true" /* suppress_notifications */);
 
   NavigateAndExpectActivation({false}, {GURL(kExampleUrl)}, EMPTY,
                               ActivationDecision::ACTIVATED);
-  EXPECT_CALL(*client(), ToggleNotificationVisibility(::testing::_)).Times(0);
+  EXPECT_CALL(*client(),
+              ToggleNotificationVisibility(::testing::_, ::testing::_))
+      .Times(0);
   EmulateDidDisallowFirstSubresourceMessage();
 }
 
 TEST_F(ContentSubresourceFilterDriverFactoryTest, WhitelistSiteOnReload) {
   const struct {
     content::Referrer referrer;
     ui::PageTransition transition;
     ActivationDecision expected_activation_decision;
   } kTestCases[] = {
       {content::Referrer(), ui::PAGE_TRANSITION_LINK,
@@ skipping 144 matching lines @@
     ActivationScopeTest,
     ContentSubresourceFilterDriverFactoryActivationScopeTest,
     ::testing::ValuesIn(kActivationScopeTestData));
 
 INSTANTIATE_TEST_CASE_P(
     ActivationLevelTest,
     ContentSubresourceFilterDriverFactoryActivationLevelTest,
     ::testing::ValuesIn(kActivationLevelTestData));
 
 }  // namespace subresource_filter