// 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 "components/optimization_guide/hints_processing_util.h"

#include "components/optimization_guide/proto/hint_cache.pb.h"
#include "components/optimization_guide/proto/hints.pb.h"
#include "components/optimization_guide/store_update_data.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"

namespace optimization_guide {

TEST(HintsProcessingUtilTest, FindPageHintForSubstringPagePattern) {
  proto::Hint hint1;

  // Page hint for "/one/"
  proto::PageHint* page_hint1 = hint1.add_page_hints();
  page_hint1->set_page_pattern("foo.org/*/one/");

  // Page hint for "two"
  proto::PageHint* page_hint2 = hint1.add_page_hints();
  page_hint2->set_page_pattern("two");

  // Page hint for "three.jpg"
  proto::PageHint* page_hint3 = hint1.add_page_hints();
  page_hint3->set_page_pattern("three.jpg");

  EXPECT_EQ(nullptr, FindPageHintForURL(GURL(""), &hint1));
  EXPECT_EQ(nullptr, FindPageHintForURL(GURL("https://www.foo.org/"), &hint1));
  EXPECT_EQ(nullptr,
            FindPageHintForURL(GURL("https://www.foo.org/one"), &hint1));

  EXPECT_EQ(nullptr,
            FindPageHintForURL(GURL("https://www.foo.org/one/"), &hint1));
  EXPECT_EQ(page_hint1,
            FindPageHintForURL(GURL("https://www.foo.org/pages/one/"), &hint1));
  EXPECT_EQ(page_hint1,
            FindPageHintForURL(GURL("https://www.foo.org/pages/subpages/one/"),
                               &hint1));
  EXPECT_EQ(page_hint1, FindPageHintForURL(
                            GURL("https://www.foo.org/pages/one/two"), &hint1));
  EXPECT_EQ(page_hint1,
            FindPageHintForURL(
                GURL("https://www.foo.org/pages/one/two/three.jpg"), &hint1));

  EXPECT_EQ(page_hint2,
            FindPageHintForURL(
                GURL("https://www.foo.org/pages/onetwo/three.jpg"), &hint1));
  EXPECT_EQ(page_hint2,
            FindPageHintForURL(GURL("https://www.foo.org/one/two/three.jpg"),
                               &hint1));
  EXPECT_EQ(page_hint2,
            FindPageHintForURL(GURL("https://one.two.org"), &hint1));

  EXPECT_EQ(page_hint3, FindPageHintForURL(
                            GURL("https://www.foo.org/bar/three.jpg"), &hint1));
}

TEST(HintsProcessingUtilTest, ConvertProtoEffectiveConnectionType) {
  EXPECT_EQ(
      ConvertProtoEffectiveConnectionType(
          proto::EffectiveConnectionType::EFFECTIVE_CONNECTION_TYPE_UNKNOWN),
      net::EffectiveConnectionType::EFFECTIVE_CONNECTION_TYPE_UNKNOWN);
  EXPECT_EQ(
      ConvertProtoEffectiveConnectionType(
          proto::EffectiveConnectionType::EFFECTIVE_CONNECTION_TYPE_OFFLINE),
      net::EffectiveConnectionType::EFFECTIVE_CONNECTION_TYPE_OFFLINE);
  EXPECT_EQ(
      ConvertProtoEffectiveConnectionType(
          proto::EffectiveConnectionType::EFFECTIVE_CONNECTION_TYPE_SLOW_2G),
      net::EffectiveConnectionType::EFFECTIVE_CONNECTION_TYPE_SLOW_2G);
  EXPECT_EQ(ConvertProtoEffectiveConnectionType(
                proto::EffectiveConnectionType::EFFECTIVE_CONNECTION_TYPE_2G),
            net::EffectiveConnectionType::EFFECTIVE_CONNECTION_TYPE_2G);
  EXPECT_EQ(ConvertProtoEffectiveConnectionType(
                proto::EffectiveConnectionType::EFFECTIVE_CONNECTION_TYPE_3G),
            net::EffectiveConnectionType::EFFECTIVE_CONNECTION_TYPE_3G);
  EXPECT_EQ(ConvertProtoEffectiveConnectionType(
                proto::EffectiveConnectionType::EFFECTIVE_CONNECTION_TYPE_4G),
            net::EffectiveConnectionType::EFFECTIVE_CONNECTION_TYPE_4G);
}

TEST(HintsProcessingUtilTest, IsValidURLForURLKeyedHints) {
  EXPECT_TRUE(IsValidURLForURLKeyedHint(GURL("https://blerg.com")));
  EXPECT_TRUE(IsValidURLForURLKeyedHint(GURL("http://blerg.com")));

  EXPECT_FALSE(IsValidURLForURLKeyedHint(GURL("file://blerg")));
  EXPECT_FALSE(IsValidURLForURLKeyedHint(GURL("chrome://blerg")));
  EXPECT_FALSE(IsValidURLForURLKeyedHint(
      GURL("https://username:password@www.example.com/")));
  EXPECT_FALSE(IsValidURLForURLKeyedHint(GURL("https://localhost:5000")));
  EXPECT_FALSE(IsValidURLForURLKeyedHint(GURL("https://192.168.1.1")));
}

}  // namespace optimization_guide