Refactoring, moving and renaming the NetworkActionPredictor.

chrome/browser/predictors/predictor_database_unittest.cc

Index: chrome/browser/predictors/predictor_database_unittest.cc
diff --git a/chrome/browser/predictors/predictor_database_unittest.cc b/chrome/browser/predictors/predictor_database_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..4fa0b1e075563ef26c1d715f223de27ddfb4c004
--- /dev/null
+++ b/chrome/browser/predictors/predictor_database_unittest.cc
@@ -0,0 +1,240 @@
+// Copyright (c) 2011 The Chromium Authors. All rights reserved.

[dominich, 2012/03/06 16:42:13]

2012

[Shishir, 2012/03/14 21:14:37]

Done.

+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include <vector>
+
+#include "base/message_loop.h"
+#include "base/time.h"
+#include "base/utf_string_conversions.h"
+#include "chrome/browser/predictors/predictor_database.h"
+#include "chrome/browser/predictors/predictor_database_factory.h"
+#include "chrome/test/base/testing_profile.h"
+#include "content/test/test_browser_thread.h"
+#include "sql/statement.h"
+
+#include "testing/gtest/include/gtest/gtest.h"
+
+using base::Time;
+using base::TimeDelta;
+using content::BrowserThread;
+using predictors::AutocompleteActionPredictorTable;
+
+namespace {
+
+struct AutocompleteActionPredictorTable::Row test_db[] = {

[dominich, 2012/03/06 16:42:13]

consider splitting this file into one per table type, or per predictor type.

[Shishir, 2012/03/14 21:14:37]

Done.

+  AutocompleteActionPredictorTable::Row(
+      "BD85DBA2-8C29-49F9-84AE-48E1E90880DF",
+      ASCIIToUTF16("goog"), GURL("http://www.google.com/"),
+      1, 0),
+  AutocompleteActionPredictorTable::Row(
+      "BD85DBA2-8C29-49F9-84AE-48E1E90880E0",
+      ASCIIToUTF16("slash"), GURL("http://slashdot.org/"),
+      3, 2),
+  AutocompleteActionPredictorTable::Row(
+      "BD85DBA2-8C29-49F9-84AE-48E1E90880E1",
+      ASCIIToUTF16("news"), GURL("http://slashdot.org/"),
+      0, 1),
+};
+
+}  // end namespace
+
+namespace predictors {
+
+class AutocompleteActionPredictorTableTest : public testing::Test {
+ public:
+  AutocompleteActionPredictorTableTest();
+  virtual ~AutocompleteActionPredictorTableTest();
+
+  virtual void SetUp();
+  virtual void TearDown();
+
+  size_t CountRecords() const;
+
+  void AddAll();
+
+  bool RowsAreEqual(const AutocompleteActionPredictorTable::Row& lhs,
+                    const AutocompleteActionPredictorTable::Row& rhs) const;
+
+  TestingProfile* profile() { return &profile_; }
+
+ protected:
+
+  // Test functions that can be run against this text fixture or
+  // AutocompleteActionPredictorTableReopenTest that inherits from this.
+  void TestAddRow();
+  void TestGetRow();
+  void TestUpdateRow();
+  void TestDeleteRow();
+  void TestDeleteRows();
+  void TestDeleteAllRows();
+
+ private:
+  TestingProfile profile_;
+  scoped_refptr<PredictorDatabase> db_;
+  MessageLoop loop_;
+  content::TestBrowserThread db_thread_;
+};
+
+class AutocompleteActionPredictorTableReopenTest
+    : public AutocompleteActionPredictorTableTest {
+ public:
+  virtual void SetUp() {
+    // By calling SetUp twice, we make sure that the table already exists for
+    // this fixture.
+    AutocompleteActionPredictorTableTest::SetUp();
+    AutocompleteActionPredictorTableTest::TearDown();
+    AutocompleteActionPredictorTableTest::SetUp();
+  }
+};
+
+AutocompleteActionPredictorTableTest::AutocompleteActionPredictorTableTest()
+    : loop_(MessageLoop::TYPE_DEFAULT),
+      db_thread_(BrowserThread::DB, &loop_) {
+}
+
+AutocompleteActionPredictorTableTest::~AutocompleteActionPredictorTableTest() {
+}
+
+void AutocompleteActionPredictorTableTest::SetUp() {
+  db_ = new PredictorDatabase(&profile_);
+  db_->Initialize();
+}
+
+void AutocompleteActionPredictorTableTest::TearDown() {
+  db_ = NULL;
+}
+
+size_t AutocompleteActionPredictorTableTest::CountRecords() const {
+  sql::Statement s(db_->db_.GetUniqueStatement(
+      "SELECT count(*) FROM network_action_predictor"));
+  EXPECT_TRUE(s.Step());
+  return static_cast<size_t>(s.ColumnInt(0));
+}
+
+void AutocompleteActionPredictorTableTest::AddAll() {
+  for (size_t i = 0; i < arraysize(test_db); ++i)
+    db_->autocomplete_table()->AddRow(test_db[i]);
+
+  EXPECT_EQ(arraysize(test_db), CountRecords());
+}
+
+bool AutocompleteActionPredictorTableTest::RowsAreEqual(
+    const AutocompleteActionPredictorTable::Row& lhs,
+    const AutocompleteActionPredictorTable::Row& rhs) const {
+  return (lhs.id == rhs.id &&
+          lhs.user_text == rhs.user_text &&
+          lhs.url == rhs.url &&
+          lhs.number_of_hits == rhs.number_of_hits &&
+          lhs.number_of_misses == rhs.number_of_misses);
+}
+
+void AutocompleteActionPredictorTableTest::TestAddRow() {
+  EXPECT_EQ(0U, CountRecords());
+  db_->autocomplete_table()->AddRow(test_db[0]);
+  EXPECT_EQ(1U, CountRecords());
+  db_->autocomplete_table()->AddRow(test_db[1]);
+  EXPECT_EQ(2U, CountRecords());
+  db_->autocomplete_table()->AddRow(test_db[2]);
+  EXPECT_EQ(3U, CountRecords());
+}
+
+void AutocompleteActionPredictorTableTest::TestGetRow() {
+  db_->autocomplete_table()->AddRow(test_db[0]);
+  AutocompleteActionPredictorTable::Row row;
+  db_->autocomplete_table()->GetRow(test_db[0].id, &row);
+  EXPECT_TRUE(RowsAreEqual(test_db[0], row))
+      << "Expected: Row with id " << test_db[0].id << "\n"
+      << "Got:      Row with id " << row.id;
+}
+
+void AutocompleteActionPredictorTableTest::TestUpdateRow() {
+  AddAll();
+  AutocompleteActionPredictorTable::Row row = test_db[1];
+  row.number_of_hits = row.number_of_hits + 1;
+  db_->autocomplete_table()->UpdateRow(row);
+
+  AutocompleteActionPredictorTable::Row updated_row;
+  db_->autocomplete_table()->GetRow(test_db[1].id, &updated_row);
+
+  EXPECT_TRUE(RowsAreEqual(row, updated_row))
+      << "Expected: Row with id " << row.id << "\n"
+      << "Got:      Row with id " << updated_row.id;
+}
+
+void AutocompleteActionPredictorTableTest::TestDeleteRow() {
+  AddAll();
+  db_->autocomplete_table()->DeleteRow(test_db[2].id);
+  EXPECT_EQ(arraysize(test_db) - 1, CountRecords());
+}
+
+void AutocompleteActionPredictorTableTest::TestDeleteRows() {
+  AddAll();
+  std::vector<AutocompleteActionPredictorTable::Row::Id> id_list;
+  id_list.push_back(test_db[0].id);
+  id_list.push_back(test_db[2].id);
+  db_->autocomplete_table()->DeleteRows(id_list);
+  EXPECT_EQ(arraysize(test_db) - 2, CountRecords());
+
+  AutocompleteActionPredictorTable::Row row;
+  db_->autocomplete_table()->GetRow(test_db[1].id, &row);
+  EXPECT_TRUE(RowsAreEqual(test_db[1], row));
+}
+
+void AutocompleteActionPredictorTableTest::TestDeleteAllRows() {
+  AddAll();
+  db_->autocomplete_table()->DeleteAllRows();
+  EXPECT_EQ(0U, CountRecords());
+}
+
+// AutocompleteActionPredictorTableTest tests
+TEST_F(AutocompleteActionPredictorTableTest, AddRow) {
+  TestAddRow();
+}
+
+TEST_F(AutocompleteActionPredictorTableTest, GetRow) {
+  TestGetRow();
+}
+
+TEST_F(AutocompleteActionPredictorTableTest, UpdateRow) {
+  TestUpdateRow();
+}
+
+TEST_F(AutocompleteActionPredictorTableTest, DeleteRow) {
+  TestDeleteRow();
+}
+
+TEST_F(AutocompleteActionPredictorTableTest, DeleteRows) {
+  TestDeleteRows();
+}
+
+TEST_F(AutocompleteActionPredictorTableTest, DeleteAllRows) {
+  TestDeleteAllRows();
+}
+
+// AutocompleteActionPredictorTableReopenTest tests
+TEST_F(AutocompleteActionPredictorTableReopenTest, AddRow) {
+  TestAddRow();
+}
+
+TEST_F(AutocompleteActionPredictorTableReopenTest, GetRow) {
+  TestGetRow();
+}
+
+TEST_F(AutocompleteActionPredictorTableReopenTest, UpdateRow) {
+  TestUpdateRow();
+}
+
+TEST_F(AutocompleteActionPredictorTableReopenTest, DeleteRow) {
+  TestDeleteRow();
+}
+
+TEST_F(AutocompleteActionPredictorTableReopenTest, DeleteRows) {
+  TestDeleteRows();
+}
+
+TEST_F(AutocompleteActionPredictorTableReopenTest, DeleteAllRows) {
+  TestDeleteAllRows();
+}
+
+}  // namespace predictors