[on-hold][subresource_filter] Add ActivationTracker to track all activated contents

chrome/browser/subresource_filter/subresource_filter_browsertest.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 <map>
 #include <memory>
 #include <string>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/command_line.h"
 #include "base/files/file_path.h"
 #include "base/location.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/path_service.h"
 #include "base/strings/string_piece.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
 #include "base/test/histogram_tester.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/content_settings/host_content_settings_map_factory.h"
 #include "chrome/browser/content_settings/tab_specific_content_settings.h"
 #include "chrome/browser/metrics/subprocess_metrics_provider.h"
 #include "chrome/browser/page_load_metrics/observers/subresource_filter_metrics_observer.h"
 #include "chrome/browser/safe_browsing/test_safe_browsing_service.h"
 #include "chrome/browser/safe_browsing/v4_test_utils.h"
 #include "chrome/browser/subresource_filter/chrome_subresource_filter_client.h"
+#include "chrome/browser/subresource_filter/subresource_filter_activation_tracker.h"
+#include "chrome/browser/subresource_filter/subresource_filter_profile_context.h"
+#include "chrome/browser/subresource_filter/subresource_filter_profile_context_factory.h"
 #include "chrome/browser/subresource_filter/test_ruleset_publisher.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_commands.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "components/content_settings/core/browser/host_content_settings_map.h"
 #include "components/content_settings/core/common/content_settings.h"
@@ skipping 439 matching lines @@
 
   void SetUpActivationFeature() override {
     ToggleFeatures(base::MakeUnique<ScopedSubresourceFilterFeatureToggle>(
         base::FeatureList::OVERRIDE_ENABLE_FEATURE, kActivationLevelEnabled,
         kActivationScopeActivationList, kActivationListSubresourceFilter, "0",
         "" /* suppress_notifications */, "false"));
   }
 };
 #endif
 
+class SubresourceFilterDryRunBrowserTest
+    : public SubresourceFilterBrowserTestImpl {
+ public:
+  SubresourceFilterDryRunBrowserTest()
+      : SubresourceFilterBrowserTestImpl(false, false) {}
+
+  void SetUpActivationFeature() override {
+    ToggleFeatures(base::MakeUnique<ScopedSubresourceFilterFeatureToggle>(
+        base::FeatureList::OVERRIDE_ENABLE_FEATURE, kActivationLevelDryRun,
+        kActivationScopeActivationList, kActivationListSubresourceFilter, "0",
+        "" /* suppress_notifications */, "false"));
+  }
+};
+
 // Tests -----------------------------------------------------------------------
 
 IN_PROC_BROWSER_TEST_F(SubresourceFilterBrowserTest, MainFrameActivation) {
   GURL url(GetTestUrl("subresource_filter/frame_with_included_script.html"));
   ConfigureAsPhishingURL(url);
   ASSERT_NO_FATAL_FAILURE(SetRulesetToDisallowURLsWithPathSuffix(
       "suffix-that-does-not-match-anything"));
   ui_test_utils::NavigateToURL(browser(), url);
   EXPECT_TRUE(WasParsedScriptElementLoaded(web_contents()->GetMainFrame()));
 
@@ skipping 513 matching lines @@
   EXPECT_TRUE(WasParsedScriptElementLoaded(web_contents()->GetMainFrame()));
 
   // A cross site blacklisted navigation should stay activated, however.
   GURL a_url(embedded_test_server()->GetURL(
       "a.com", "/subresource_filter/frame_with_included_script.html"));
   ConfigureAsPhishingURL(a_url);
   ui_test_utils::NavigateToURL(browser(), a_url);
   EXPECT_FALSE(WasParsedScriptElementLoaded(web_contents()->GetMainFrame()));
 }
 
+IN_PROC_BROWSER_TEST_F(SubresourceFilterBrowserTest, ActivationTracking) {
+  SetRulesetWithRules(std::vector<proto::UrlRule>());
+  SubresourceFilterProfileContext* context =
+      SubresourceFilterProfileContextFactory::GetForProfile(
+          browser()->profile());
+
+  const GURL kActivatedUrl = GetTestUrl(kTestFrameSetPath);
+  ConfigureAsPhishingURL(kActivatedUrl);
+
+  EXPECT_FALSE(context->activation_tracker()->HasActivatedWebContents());
+  ui_test_utils::NavigateToURL(browser(), kActivatedUrl);
+  EXPECT_TRUE(context->activation_tracker()->HasActivatedWebContents());
+
+  // Add another tab without activation.
+  AddTabAtIndex(0, GURL("about:blank"), ui::PAGE_TRANSITION_LINK);
+  EXPECT_TRUE(context->activation_tracker()->HasActivatedWebContents());
+  ui_test_utils::NavigateToURL(browser(), GURL("https://example.test"));
+  EXPECT_TRUE(context->activation_tracker()->HasActivatedWebContents());
+
+  // Add another tab with activation, but close it.
+  AddTabAtIndex(0, kActivatedUrl, ui::PAGE_TRANSITION_LINK);
+  EXPECT_TRUE(context->activation_tracker()->HasActivatedWebContents());
+  browser()->tab_strip_model()->CloseSelectedTabs();
+  EXPECT_TRUE(context->activation_tracker()->HasActivatedWebContents());
+
+  // Go back to the original tab and navigate away.
+  browser()->tab_strip_model()->ActivateTabAt(1, true /* user_gesture */);
+  ui_test_utils::NavigateToURL(browser(), GURL("https://example.test"));
+  EXPECT_FALSE(context->activation_tracker()->HasActivatedWebContents());
+
+  // Navigate once again, and close all the tabs.
+  ui_test_utils::NavigateToURL(browser(), kActivatedUrl);
+  EXPECT_TRUE(context->activation_tracker()->HasActivatedWebContents());
+
+  browser()->tab_strip_model()->CloseAllTabs();
+  EXPECT_FALSE(context->activation_tracker()->HasActivatedWebContents());
+}
+
+// Dry run activation should not be tracked by that activation tracker.
+IN_PROC_BROWSER_TEST_F(SubresourceFilterDryRunBrowserTest,
+                       DryRun_NoActivationTracking) {
+  SetRulesetWithRules(std::vector<proto::UrlRule>());
+  SubresourceFilterProfileContext* context =
+      SubresourceFilterProfileContextFactory::GetForProfile(
+          browser()->profile());
+
+  const GURL kActivatedUrl = GetTestUrl(kTestFrameSetPath);
+  ConfigureAsPhishingURL(kActivatedUrl);
+
+  EXPECT_FALSE(context->activation_tracker()->HasActivatedWebContents());
+  ui_test_utils::NavigateToURL(browser(), kActivatedUrl);
+  EXPECT_FALSE(context->activation_tracker()->HasActivatedWebContents());
+
+  // Add another tab with activation, but close it.
+  AddTabAtIndex(0, kActivatedUrl, ui::PAGE_TRANSITION_LINK);
+  EXPECT_FALSE(context->activation_tracker()->HasActivatedWebContents());
+  browser()->tab_strip_model()->CloseSelectedTabs();
+  EXPECT_FALSE(context->activation_tracker()->HasActivatedWebContents());
+}
+
 IN_PROC_BROWSER_TEST_P(SubresourceFilterWebSocketBrowserTest, BlockWebSocket) {
   GURL url(GetTestUrl(
       base::StringPrintf("subresource_filter/page_with_websocket.html?%s",
                          GetParam() == IN_WORKER ? "inWorker" : "")));
   GURL websocket_url(GetWebSocketUrl("echo-with-no-extension"));
   ConfigureAsPhishingURL(url);
   ASSERT_NO_FATAL_FAILURE(
       SetRulesetToDisallowURLsWithPathSuffix("echo-with-no-extension"));
   ui_test_utils::NavigateToURL(browser(), url);
   CreateWebSocketAndExpectResult(websocket_url,
@@ skipping 349 matching lines @@
                                    std::string(kSubresourceFilterOnlySuffix)),
               ::testing::IsEmpty());
 
   EXPECT_THAT(tester.GetAllSamples(std::string(kNavigationChainSize) +
                                    std::string(kSubresourceFilterOnlySuffix)),
               ::testing::IsEmpty());
 }
 #endif
 
 }  // namespace subresource_filter