components/previews/content/hint_cache_unittest.cc

// Copyright 2018 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/previews/content/hint_cache.h"

#include <string>
#include <vector>

#include "base/bind.h"
#include "base/macros.h"
#include "components/previews/core/previews_experiments.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"

namespace previews {

namespace {

class HintCacheTest : public testing::Test {
 public:
  HintCacheTest() {}

  ~HintCacheTest() override {}

  void LoadCallback(const optimization_guide::proto::Hint& hint) {
    loaded_hint_ = std::make_unique<optimization_guide::proto::Hint>(hint);
  }

  const optimization_guide::proto::Hint* GetLoadedHint() const {
    return loaded_hint_.get();
  }

 private:
  std::unique_ptr<optimization_guide::proto::Hint> loaded_hint_;

  DISALLOW_COPY_AND_ASSIGN(HintCacheTest);
};

TEST_F(HintCacheTest, TestMemoryCache) {
  optimization_guide::proto::Hint hint1;
  hint1.set_key("subdomain.domain.org");
  hint1.set_key_representation(optimization_guide::proto::HOST_SUFFIX);
  optimization_guide::proto::Hint hint2;
  hint2.set_key("host.domain.org");
  hint2.set_key_representation(optimization_guide::proto::HOST_SUFFIX);
  optimization_guide::proto::Hint hint3;
  hint3.set_key("otherhost.subdomain.domain.org");
  hint3.set_key_representation(optimization_guide::proto::HOST_SUFFIX);

  HintCache::Data data;
  data.AddHint(hint1);
  data.AddHint(hint2);
  data.AddHint(hint3);
  HintCache hint_cache(std::move(data));

  // Not matched
  EXPECT_FALSE(hint_cache.HasHint("domain.org"));
  EXPECT_FALSE(hint_cache.HasHint("othersubdomain.domain.org"));

  // Matched
  EXPECT_TRUE(hint_cache.HasHint("otherhost.subdomain.domain.org"));
  EXPECT_TRUE(hint_cache.HasHint("host.subdomain.domain.org"));
  EXPECT_TRUE(hint_cache.HasHint("subhost.host.subdomain.domain.org"));

  // Cached in memory
  EXPECT_TRUE(hint_cache.IsHintLoaded("host.domain.org"));
  EXPECT_TRUE(hint_cache.IsHintLoaded("otherhost.subdomain.domain.org"));
  EXPECT_TRUE(hint_cache.IsHintLoaded("host.subdomain.domain.org"));
  EXPECT_TRUE(hint_cache.IsHintLoaded("subhost.host.subdomain.domain.org"));

  // Matched key
  EXPECT_EQ(hint2.key(), hint_cache.GetHint("host.domain.org")->key());
  EXPECT_EQ(hint3.key(),
            hint_cache.GetHint("otherhost.subdomain.domain.org")->key());
  EXPECT_EQ(hint1.key(),
            hint_cache.GetHint("host.subdomain.domain.org")->key());
}

TEST_F(HintCacheTest, TestMemoryCacheLoadCallback) {
  optimization_guide::proto::Hint hint1;
  hint1.set_key("subdomain.domain.org");
  hint1.set_key_representation(optimization_guide::proto::HOST_SUFFIX);

  HintCache::Data data;
  data.AddHint(hint1);
  HintCache hint_cache(std::move(data));

  EXPECT_TRUE(hint_cache.IsHintLoaded("host.subdomain.domain.org"));
  hint_cache.LoadHint(
      "host.subdomain.domain.org",
      base::BindOnce(&HintCacheTest::LoadCallback, base::Unretained(this)));

  EXPECT_TRUE(GetLoadedHint());
  EXPECT_EQ(hint1.key(), GetLoadedHint()->key());
}

}  // namespace

}  // namespace previews