Enable websocket filtering via SubresourceFilter

components/subresource_filter/core/common/document_subresource_filter_unittest.cc

Index: components/subresource_filter/core/common/document_subresource_filter_unittest.cc
diff --git a/components/subresource_filter/core/common/document_subresource_filter_unittest.cc b/components/subresource_filter/core/common/document_subresource_filter_unittest.cc
index b7ea78249e1501b4a0226f002f7499c7903bb179..0030cbd105a14ce1032ef66eb79259c9f495bb1f 100644
--- a/components/subresource_filter/core/common/document_subresource_filter_unittest.cc
+++ b/components/subresource_filter/core/common/document_subresource_filter_unittest.cc
@@ -25,6 +25,7 @@ constexpr auto kSubdocumentType = proto::ELEMENT_TYPE_SUBDOCUMENT;
 
 constexpr const char kTestAlphaURL[] = "http://example.com/alpha";
 constexpr const char kTestAlphaDataURI[] = "data:text/plain,alpha";
+constexpr const char kTestAlphaWSURI[] = "ws://example.com/alpha";
 constexpr const char kTestBetaURL[] = "http://example.com/beta";
 
 constexpr const char kTestAlphaURLPathSuffix[] = "alpha";
@@ -70,6 +71,9 @@ TEST_F(DocumentSubresourceFilterTest, DryRun) {
             filter.GetLoadPolicy(GURL(kTestAlphaURL), kImageType));
   EXPECT_EQ(LoadPolicy::ALLOW,
             filter.GetLoadPolicy(GURL(kTestAlphaDataURI), kImageType));
+  EXPECT_EQ(
+      LoadPolicy::WOULD_DISALLOW,
+      filter.GetLoadPolicy(GURL(kTestAlphaWSURI), proto::ELEMENT_TYPE_OTHER));
   EXPECT_EQ(LoadPolicy::ALLOW,
             filter.GetLoadPolicy(GURL(kTestBetaURL), kImageType));
   EXPECT_EQ(LoadPolicy::WOULD_DISALLOW,
@@ -78,9 +82,9 @@ TEST_F(DocumentSubresourceFilterTest, DryRun) {
             filter.GetLoadPolicy(GURL(kTestBetaURL), kSubdocumentType));
 
   const auto& statistics = filter.statistics();
-  EXPECT_EQ(5, statistics.num_loads_total);
-  EXPECT_EQ(4, statistics.num_loads_evaluated);
-  EXPECT_EQ(2, statistics.num_loads_matching_rules);
+  EXPECT_EQ(6, statistics.num_loads_total);
+  EXPECT_EQ(5, statistics.num_loads_evaluated);
+  EXPECT_EQ(3, statistics.num_loads_matching_rules);
   EXPECT_EQ(0, statistics.num_loads_disallowed);
 }
 
@@ -95,6 +99,9 @@ TEST_F(DocumentSubresourceFilterTest, Enabled) {
               filter.GetLoadPolicy(GURL(kTestAlphaURL), kImageType));
     EXPECT_EQ(LoadPolicy::ALLOW,
               filter.GetLoadPolicy(GURL(kTestAlphaDataURI), kImageType));
+    EXPECT_EQ(
+        LoadPolicy::DISALLOW,
+        filter.GetLoadPolicy(GURL(kTestAlphaWSURI), proto::ELEMENT_TYPE_OTHER));
     EXPECT_EQ(LoadPolicy::ALLOW,
               filter.GetLoadPolicy(GURL(kTestBetaURL), kImageType));
     EXPECT_EQ(LoadPolicy::DISALLOW,
@@ -103,10 +110,10 @@ TEST_F(DocumentSubresourceFilterTest, Enabled) {
               filter.GetLoadPolicy(GURL(kTestBetaURL), kSubdocumentType));
 
     const auto& statistics = filter.statistics();
-    EXPECT_EQ(5, statistics.num_loads_total);
-    EXPECT_EQ(4, statistics.num_loads_evaluated);
-    EXPECT_EQ(2, statistics.num_loads_matching_rules);
-    EXPECT_EQ(2, statistics.num_loads_disallowed);
+    EXPECT_EQ(6, statistics.num_loads_total);
+    EXPECT_EQ(5, statistics.num_loads_evaluated);
+    EXPECT_EQ(3, statistics.num_loads_matching_rules);
+    EXPECT_EQ(3, statistics.num_loads_disallowed);
 
     if (!measure_performance) {
       EXPECT_TRUE(statistics.evaluation_total_cpu_duration.is_zero());