1 files changed, 117 insertions, 0 deletions
diff --git a/chromium/chrome/renderer/subresource_redirect/subresource_redirect_experiments_unittest.cc b/chromium/chrome/renderer/subresource_redirect/subresource_redirect_experiments_unittest.cc
new file mode 100644
index 00000000000..99c66e146a5
--- /dev/null
+++ b/chromium/chrome/renderer/subresource_redirect/subresource_redirect_experiments_unittest.cc
@@ -0,0 +1,117 @@
+// Copyright 2019 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 "chrome/renderer/subresource_redirect/subresource_redirect_experiments.h"
+
+#include "base/test/scoped_feature_list.h"
+#include "chrome/common/chrome_features.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace subresource_redirect {
+
+namespace {
+
+TEST(SubresourceRedirectExperimentsTest, TestDefaultShouldIncludeMediaSuffix) {
+  base::test::ScopedFeatureList scoped_feature_list;
+  scoped_feature_list.InitAndEnableFeature(
+      features::kSubresourceRedirectIncludedMediaSuffixes);
+
+  EXPECT_FALSE(ShouldIncludeMediaSuffix(GURL("http://chromium.org/path/")));
+
+  std::vector<std::string> default_suffixes = {".jpg", ".jpeg", ".png", ".svg",
+                                               ".webp"};
+  for (const std::string& suffix : default_suffixes) {
+    GURL url("http://chromium.org/image" + suffix);
+    EXPECT_TRUE(ShouldIncludeMediaSuffix(url));
+  }
+}
+
+TEST(SubresourceRedirectExperimentsTest, TestShouldIncludeMediaSuffix) {
+  struct TestCase {
+    std::string msg;
+    bool enable_feature;
+    std::string varaiation_value;
+    std::vector<std::string> urls;
+    bool want_return;
+  };
+  const TestCase kTestCases[]{
+      {
+          .msg = "Feature disabled, should always return true",
+          .enable_feature = false,
+          .varaiation_value = "",
+          .urls = {"http://chromium.org/image.jpg"},
+          .want_return = true,
+      },
+      {
+          .msg = "Default values are overridden by variations",
+          .enable_feature = true,
+          .varaiation_value = ".html",
+          .urls = {"http://chromium.org/image.jpeg",
+                   "http://chromium.org/image.png",
+                   "http://chromium.org/image.jpg",
+                   "http://chromium.org/image.svg"},
+          .want_return = false,
+      },
+      {
+          .msg = "Variation value whitespace should be trimmed",
+          .enable_feature = true,
+          .varaiation_value = " .svg , \t .png\n",
+          .urls = {"http://chromium.org/image.svg",
+                   "http://chromium.org/image.png"},
+          .want_return = true,
+      },
+      {
+          .msg = "Variation value empty values should be excluded",
+          .enable_feature = true,
+          .varaiation_value = ".svg,,.png,",
+          .urls = {"http://chromium.org/image.svg",
+                   "http://chromium.org/image.png"},
+          .want_return = true,
+      },
+      {
+          .msg = "URLs should be compared case insensitive",
+          .enable_feature = true,
+          .varaiation_value = ".svg,.png,",
+          .urls = {"http://chromium.org/image.SvG",
+                   "http://chromium.org/image.PNG"},
+          .want_return = true,
+      },
+      {
+          .msg = "Query params and fragments don't matter",
+          .enable_feature = true,
+          .varaiation_value = ".svg,.png,",
+          .urls = {"http://chromium.org/image.svg?hello=world",
+                   "http://chromium.org/image.png#test"},
+          .want_return = true,
+      },
+      {
+          .msg = "Query params and fragments shouldn't be considered",
+          .enable_feature = true,
+          .varaiation_value = ".svg,.png,",
+          .urls = {"http://chromium.org/?image=image.svg",
+                   "http://chromium.org/#image.png"},
+          .want_return = false,
+      },
+  };
+  for (const TestCase& test_case : kTestCases) {
+    SCOPED_TRACE(test_case.msg);
+
+    base::test::ScopedFeatureList scoped_feature_list;
+    if (test_case.enable_feature) {
+      scoped_feature_list.InitAndEnableFeatureWithParameters(
+          features::kSubresourceRedirectIncludedMediaSuffixes,
+          {{"included_path_suffixes", test_case.varaiation_value}});
+    } else {
+      scoped_feature_list.InitAndDisableFeature(
+          features::kSubresourceRedirectIncludedMediaSuffixes);
+    }
+
+    for (const std::string& url : test_case.urls) {
+      EXPECT_EQ(test_case.want_return, ShouldIncludeMediaSuffix(GURL(url)));
+    }
+  }
+}
+
+}  // namespace
+}  // namespace subresource_redirect