Precache tracking database

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 "base/basictypes.h"
+#include "base/compiler_specific.h"
+#include "base/time/time.h"
+#include "sql/connection.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace precache {
+
+namespace {
+
+base::Time GetTimeFromString(const char* time_string) {
+  base::Time time;
+  EXPECT_TRUE(base::Time::FromString(time_string, &time));
+  return time;
+}
+
+class PrecacheURLTableTest : public testing::Test {
+ public:
+  PrecacheURLTableTest() {}
+  virtual ~PrecacheURLTableTest() {}
+
+ protected:
+  virtual void SetUp() OVERRIDE {
+    precache_url_table_.reset(new PrecacheURLTable());
+    db_.reset(new sql::Connection());
+    ASSERT_TRUE(db_->OpenInMemory());
+    precache_url_table_->Init(db_.get());
+  }
+
+  scoped_ptr<PrecacheURLTable> precache_url_table_;
+  scoped_ptr<sql::Connection> db_;
+};
+
+TEST_F(PrecacheURLTableTest, AddURLWithNoExistingRow) {
+  const base::Time kTime = GetTimeFromString("11 Oct 2013, 12:00:00");
+  precache_url_table_->AddURL(GURL("http://url.com"), kTime);
+
+  std::map<GURL, base::Time> expected_map;
+  expected_map[GURL("http://url.com")] = kTime;
+
+  std::map<GURL, base::Time> actual_map;
+  precache_url_table_->GetAllDataForTesting(&actual_map);
+  EXPECT_EQ(expected_map, actual_map);
+}
+
+TEST_F(PrecacheURLTableTest, AddURLWithExistingRow) {
+  const base::Time kOldTime = GetTimeFromString("11 Oct 2013, 12:00:00");
+  const base::Time kNewTime = GetTimeFromString("11 Oct 2013, 12:00:00");
+  precache_url_table_->AddURL(GURL("http://url.com"), kOldTime);
+  precache_url_table_->AddURL(GURL("http://url.com"), kNewTime);
+
+  std::map<GURL, base::Time> expected_map;
+  expected_map[GURL("http://url.com")] = kNewTime;
+
+  std::map<GURL, base::Time> actual_map;
+  precache_url_table_->GetAllDataForTesting(&actual_map);
+  EXPECT_EQ(expected_map, actual_map);
+}
+
+TEST_F(PrecacheURLTableTest, DeleteURL) {
+  const base::Time kStaysTime = GetTimeFromString("11 Oct 2013, 12:00:00");
+  const base::Time kDeletedTime = GetTimeFromString("11 Oct 2013, 13:00:00");
+
+  precache_url_table_->AddURL(GURL("http://stays.com"), kStaysTime);
+  precache_url_table_->AddURL(GURL("http://deleted.com"), kDeletedTime);
+
+  precache_url_table_->DeleteURL(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, HasURL) {
+  EXPECT_FALSE(precache_url_table_->HasURL(GURL("http://url.com")));
+
+  precache_url_table_->AddURL(GURL("http://url.com"),
+                              GetTimeFromString("11 Oct 2013, 12:00:00"));
+
+  EXPECT_TRUE(precache_url_table_->HasURL(GURL("http://url.com")));
+
+  precache_url_table_->DeleteURL(GURL("http://url.com"));
+
+  EXPECT_FALSE(precache_url_table_->HasURL(GURL("http://url.com")));
+}
+
+TEST_F(PrecacheURLTableTest, DeleteAllPrecachedUntil) {
+  const base::Time kBeforeTime = GetTimeFromString("11 Oct 2013, 11:59:59");
+  const base::Time kEndTime = GetTimeFromString("11 Oct 2013, 12:00:00");
+  const base::Time kAfterTime = GetTimeFromString("11 Oct 2013, 12:00:01");
+
+  precache_url_table_->AddURL(GURL("http://before.com"), kBeforeTime);
+  precache_url_table_->AddURL(GURL("http://end.com"), kEndTime);
+  precache_url_table_->AddURL(GURL("http://after.com"), kAfterTime);
+
+  precache_url_table_->DeleteAllPrecachedUntil(kEndTime);
+
+  std::map<GURL, base::Time> expected_map;
+  expected_map[GURL("http://after.com")] = kAfterTime;
+
+  std::map<GURL, base::Time> actual_map;
+  precache_url_table_->GetAllDataForTesting(&actual_map);
+  EXPECT_EQ(expected_map, actual_map);
+}
+
+}  // namespace
+
+}  // namespace precache