brave_translate_redirect_network_delegate_helper_unittest.cc

/* 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/browser/net/brave_translate_redirect_network_delegate_helper.h"

#include <memory>
#include <string>
#include <vector>

#include "brave/browser/net/url_context.h"
#include "brave/common/translate_network_constants.h"
#include "chrome/test/base/chrome_render_view_host_test_harness.h"
#include "net/traffic_annotation/network_traffic_annotation_test_helper.h"
#include "net/url_request/url_request_test_util.h"
#include "url/gurl.h"

namespace {

class BraveTranslateRedirectNetworkDelegateHelperTest
    : public testing::Test {
 public:
  BraveTranslateRedirectNetworkDelegateHelperTest()
      : context_(new net::TestURLRequestContext(true)) {}
  ~BraveTranslateRedirectNetworkDelegateHelperTest() override {}
  void SetUp() override { context_->Init(); }
  net::TestURLRequestContext* context() { return context_.get(); }

 private:
  content::BrowserTaskEnvironment task_environment_;
  std::unique_ptr<net::TestURLRequestContext> context_;
};

TEST_F(BraveTranslateRedirectNetworkDelegateHelperTest,
    RedirectTranslateScriptRequestInitiateByGTranslate) {
  std::string translate_host("https://translate.googleapis.com");
  std::vector<std::string> paths({
      "/element/TE_20181015_01/e/js/element/element_main.js",
      "/translate_static/js/element/main.js"
  });
  std::for_each(paths.begin(), paths.end(),
      [this, &translate_host](std::string path_string) {
    net::TestDelegate test_delegate;
    GURL url(translate_host + path_string);

    std::unique_ptr<net::URLRequest> request = context()->CreateRequest(
        url, net::IDLE, &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
    request->set_initiator(url::Origin::Create(GURL(kTranslateInitiatorURL)));

    std::shared_ptr<brave::BraveRequestInfo> before_url_context(
        new brave::BraveRequestInfo());
    before_url_context->initiator_url = GURL(kTranslateInitiatorURL);
    before_url_context->request_url = url;
    brave::ResponseCallback callback;

    GURL expected_url(kBraveTranslateServer + path_string);
    int ret = OnBeforeURLRequest_TranslateRedirectWork(callback,
        before_url_context);
    EXPECT_EQ(before_url_context->new_url_spec, expected_url);
    EXPECT_EQ(ret, net::OK);
  });
}

TEST_F(BraveTranslateRedirectNetworkDelegateHelperTest,
    NoRedirectTranslateScriptRequestNotInitiateByGTranslate) {
  std::string translate_host("https://translate.googleapis.com");
  std::vector<std::string> paths({
      "/element/TE_20181015_01/e/js/element/element_main.js",
      "/translate_static/js/element/main.js"
  });
  std::for_each(paths.begin(), paths.end(),
      [this, &translate_host](std::string path_string) {
    net::TestDelegate test_delegate;

    GURL url(translate_host + path_string);
    std::unique_ptr<net::URLRequest> request = context()->CreateRequest(
        url, net::IDLE, &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS);

    std::shared_ptr<brave::BraveRequestInfo> before_url_context(
        new brave::BraveRequestInfo());
    before_url_context->request_url = url;
    brave::ResponseCallback callback;

    int ret = OnBeforeURLRequest_TranslateRedirectWork(callback,
        before_url_context);
    EXPECT_EQ(before_url_context->new_url_spec, "");
    EXPECT_EQ(ret, net::OK);
  });
}

TEST_F(BraveTranslateRedirectNetworkDelegateHelperTest,
    RedirectTranslateResourceRequest) {
  std::string translate_host("https://translate.googleapis.com");
  std::string gstatic_host("https://www.gstatic.com");
  std::vector<GURL> urls({
      GURL(translate_host + "/translate_static/css/translateelement.css"),
      GURL(gstatic_host + "/images/branding/product/1x/translate_24dp.png"),
      GURL(gstatic_host + "/images/branding/product/2x/translate_24dp.png")
  });
  std::for_each(urls.begin(), urls.end(),
      [this](GURL url) {
    net::TestDelegate test_delegate;

    std::unique_ptr<net::URLRequest> request = context()->CreateRequest(
        url, net::IDLE, &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS);

    std::shared_ptr<brave::BraveRequestInfo> before_url_context(
        new brave::BraveRequestInfo());
    before_url_context->request_url = url;
    brave::ResponseCallback callback;

    GURL expected_url(kBraveTranslateServer + url.path());
    int ret = OnBeforeURLRequest_TranslateRedirectWork(callback,
        before_url_context);
    EXPECT_EQ(before_url_context->new_url_spec, expected_url);
    EXPECT_EQ(ret, net::OK);
  });
}

TEST_F(BraveTranslateRedirectNetworkDelegateHelperTest,
    RedirectTranslateRequestInitiateByGTranslate) {
  net::TestDelegate test_delegate;
  GURL url(
      "https://translate.googleapis.com/translate_a/t?"
      "anno=3&client=te_lib&format=html&v=1.0&key=DUMMY_KEY&"
      "logld=vTE_20181015_01&sl=auto&tl=en&sp=nmt&tc=1&sr=1&"
      "tk=568026.932804&mode=1");

  std::unique_ptr<net::URLRequest> request = context()->CreateRequest(
      url, net::IDLE, &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
  request->set_initiator(url::Origin::Create(GURL(kTranslateInitiatorURL)));

  std::shared_ptr<brave::BraveRequestInfo> before_url_context(
      new brave::BraveRequestInfo());
  before_url_context->initiator_url = GURL(kTranslateInitiatorURL);
  before_url_context->request_url = url;
  brave::ResponseCallback callback;

  GURL expected_url(kBraveTranslateEndpoint + std::string("?") + url.query());
  int ret = OnBeforeURLRequest_TranslateRedirectWork(callback,
      before_url_context);
  EXPECT_EQ(before_url_context->new_url_spec, expected_url);
  EXPECT_EQ(ret, net::OK);
}

TEST_F(BraveTranslateRedirectNetworkDelegateHelperTest,
    NoRedirectTranslateRequestNotInitiateByGTranslate) {
  net::TestDelegate test_delegate;
  GURL url(
      "https://translate.googleapis.com/translate_a/t?"
      "anno=3&client=te_lib&format=html&v=1.0&key=DUMMY_KEY&"
      "logld=vTE_20181015_01&sl=auto&tl=en&sp=nmt&tc=1&sr=1&"
      "tk=568026.932804&mode=1");

  std::unique_ptr<net::URLRequest> request = context()->CreateRequest(
      url, net::IDLE, &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS);

  std::shared_ptr<brave::BraveRequestInfo> before_url_context(
      new brave::BraveRequestInfo());
  before_url_context->request_url = url;
  brave::ResponseCallback callback;

  GURL expected_url(kBraveTranslateServer + url.path());
  int ret = OnBeforeURLRequest_TranslateRedirectWork(callback,
      before_url_context);
  EXPECT_EQ(before_url_context->new_url_spec, "");
  EXPECT_EQ(ret, net::OK);
}

TEST_F(BraveTranslateRedirectNetworkDelegateHelperTest,
    AbortTranslateGen204RequestFromTELib) {
  net::TestDelegate test_delegate;

  GURL url(
      "https://translate.google.com/gen204"
      "?nca=te_li&client=te_lib&logld=vTE_20181015_01");

  std::unique_ptr<net::URLRequest> request = context()->CreateRequest(
      url, net::IDLE, &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS);

  std::shared_ptr<brave::BraveRequestInfo> before_url_context(
      new brave::BraveRequestInfo());
  before_url_context->request_url = url;
  brave::ResponseCallback callback;

  int ret = OnBeforeURLRequest_TranslateRedirectWork(callback,
      before_url_context);

  EXPECT_EQ(before_url_context->new_url_spec, "");
  EXPECT_EQ(ret, net::ERR_ABORTED);
}

TEST_F(BraveTranslateRedirectNetworkDelegateHelperTest,
  NoAbortTranslateGen204RequestNotFromTELib) {
  net::TestDelegate test_delegate;

  GURL url(
      "https://translate.google.com/gen204"
      "?nca=te_li&client=webapp&logld=vTE_20181015_01");

  std::unique_ptr<net::URLRequest> request = context()->CreateRequest(
      url, net::IDLE, &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS);

  std::shared_ptr<brave::BraveRequestInfo> before_url_context(
      new brave::BraveRequestInfo());
  before_url_context->request_url = url;
  brave::ResponseCallback callback;

  int ret = OnBeforeURLRequest_TranslateRedirectWork(callback,
      before_url_context);
  EXPECT_EQ(before_url_context->new_url_spec, "");
  EXPECT_EQ(ret, net::OK);
}

}  // namespace