Add unittest for mock redirect based responses

components/brave_shields/browser/adblock_stub_response_unittest.cc

@@ -0,0 +1,30 @@
+/* Copyright (c) 2019 The Brave Authors. All rights reserved.
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this file,
+ * You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "brave/components/brave_shields/browser/adblock_stub_response.h"
+#include "services/network/public/cpp/resource_request.h"
+#include "services/network/public/cpp/resource_response.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+TEST(AdBlockStubResponse, ScriptRedirectDataURL) {
+  std::string redirect =
+    "data:application/script,<script>alert('hi');</script>";
+  std::string data;
+  network::ResourceRequest request;
+  network::ResourceResponseHead resource_response;
+  brave_shields::MakeStubResponse(redirect, request, &resource_response, &data);
+  ASSERT_STREQ(data.c_str(), "<script>alert('hi');</script>");
+  ASSERT_STREQ(resource_response.mime_type.c_str(), "application/script");
+}
+
+TEST(AdBlockStubResponse, HTMLRedirectDataURL) {
+  std::string redirect = "data:text/html,<strong>π</strong>";
+  std::string data;
+  network::ResourceRequest request;
+  network::ResourceResponseHead resource_response;
+  brave_shields::MakeStubResponse(redirect, request, &resource_response, &data);
+  ASSERT_STREQ(data.c_str(), "<strong>π</strong>");
+  ASSERT_STREQ(resource_response.mime_type.c_str(), "text/html");
+}

test/BUILD.gn

@@ -92,6 +92,7 @@ test("brave_unit_tests") {
     "//brave/common/shield_exceptions_unittest.cc",
     "//brave/components/assist_ranker/ranker_model_loader_impl_unittest.cc",
     "//brave/components/brave_shields/browser/ad_block_regional_service_unittest.cc",
+    "//brave/components/brave_shields/browser/adblock_stub_response_unittest.cc",
     "//brave/components/brave_shields/browser/https_everywhere_recently_used_cache_unittest.cpp",
     "//brave/components/rappor/log_uploader_unittest.cc",
     "//brave/components/translate/core/browser/translate_language_list_unittest.cc",