Add Precache.CacheStatus.NonPrefetch.FromPrecache.

components/precache/core/precache_url_table_unittest.cc

 // Copyright 2013 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/precache/core/precache_url_table.h"
 
 #include <map>
 #include <memory>
 #include <set>
 
 #include "base/compiler_specific.h"
 #include "base/time/time.h"
 #include "sql/connection.h"
 #include "sql/statement.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace precache {
 
+void PrintTo(const PrecacheURLInfo& url_info, ::std::ostream* os) {
+  *os << "{" << url_info.was_precached << ", " << url_info.is_precached << ", "
+      << url_info.was_used << "}";
+}
+
 namespace {
 
 class PrecacheURLTableTest : public testing::Test {
  public:
   PrecacheURLTableTest() {}
   ~PrecacheURLTableTest() override {}
 
  protected:
   void SetUp() override {
     precache_url_table_.reset(new PrecacheURLTable());
@@ skipping 43 matching lines @@
   precache_url_table_->SetURLAsNotPrecached(GURL("http://deleted.com"));
 
   std::map<GURL, base::Time> expected_map;
   expected_map[GURL("http://stays.com")] = kStaysTime;
 
   std::map<GURL, base::Time> actual_map;
   precache_url_table_->GetAllDataForTesting(&actual_map);
   EXPECT_EQ(expected_map, actual_map);
 }
 
-TEST_F(PrecacheURLTableTest, IsURLPrecached) {
-  EXPECT_FALSE(precache_url_table_->IsURLPrecached(GURL("http://url.com")));
+TEST_F(PrecacheURLTableTest, GetURLInfo) {
+  const GURL url("http://url.com");
 
-  precache_url_table_->AddURL(GURL("http://url.com"), 1, true,
-                              base::Time::FromInternalValue(100));
+  EXPECT_EQ((PrecacheURLInfo{false, false, false}),
+            precache_url_table_->GetURLInfo(url));
 
-  EXPECT_TRUE(precache_url_table_->IsURLPrecached(GURL("http://url.com")));
+  precache_url_table_->AddURL(url, 1, true, base::Time::FromInternalValue(100));
 
-  precache_url_table_->SetURLAsNotPrecached(GURL("http://url.com"));
+  EXPECT_EQ((PrecacheURLInfo{true, true, false}),
+            precache_url_table_->GetURLInfo(url));
 
-  EXPECT_FALSE(precache_url_table_->IsURLPrecached(GURL("http://url.com")));
+  precache_url_table_->SetPrecachedURLAsUsed(url);
+
+  EXPECT_EQ((PrecacheURLInfo{true, false, true}),
+            precache_url_table_->GetURLInfo(url));
+
+  precache_url_table_->AddURL(url, 1, true, base::Time::FromInternalValue(100));
+
+  EXPECT_EQ((PrecacheURLInfo{true, true, false}),
+            precache_url_table_->GetURLInfo(url));
+
+  precache_url_table_->SetURLAsNotPrecached(url);
+
+  EXPECT_EQ((PrecacheURLInfo{true, false, false}),
+            precache_url_table_->GetURLInfo(url));
 }
 
 TEST_F(PrecacheURLTableTest, DeleteAllPrecachedBefore) {
   const base::Time kOldTime = base::Time::FromInternalValue(10);
   const base::Time kBeforeTime = base::Time::FromInternalValue(20);
   const base::Time kEndTime = base::Time::FromInternalValue(30);
   const base::Time kAfterTime = base::Time::FromInternalValue(40);
 
   precache_url_table_->AddURL(GURL("http://old.com"), 1, true, kOldTime);
   precache_url_table_->AddURL(GURL("http://before.com"), 1, true, kBeforeTime);
@@ skipping 47 matching lines @@
     EXPECT_EQ(0, statement.ColumnInt(2));
     EXPECT_EQ(1, statement.ColumnInt(3));
   }
   EXPECT_THAT(std::set<std::string>(begin(old_urls), end(old_urls)),
               ::testing::ContainerEq(actual_urls));
 }
 
 }  // namespace
 
 }  // namespace precache