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));

[davidben, 2013/11/11 19:06:26]

Now that time zones are no longer relevant for keying time, you probably want to
be using UTC instead of the local time.

[sclittle, 2013/11/19 04:03:41]

Done.

+  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, DeleteAllPrecachedBefore) {
+  const base::Time kOldTime = GetTimeFromString("11 Oct 2013, 11:59:58");
+  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://old.com"), kOldTime);
+  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_->DeleteAllPrecachedBefore(kEndTime);
+
+  std::map<GURL, base::Time> expected_map;
+  expected_map[GURL("http://end.com")] = kEndTime;
+  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