src/chrome/browser/predictors/autocomplete_action_predictor_table_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <vector>
   6
   7 #include "base/message_loop/message_loop.h"
   8 #include "base/strings/utf_string_conversions.h"
   9 #include "base/time/time.h"
  10 #include "chrome/browser/predictors/autocomplete_action_predictor_table.h"
  11 #include "chrome/browser/predictors/predictor_database.h"
  12 #include "chrome/browser/predictors/predictor_database_factory.h"
  13 #include "chrome/test/base/testing_profile.h"
  14 #include "content/public/test/test_browser_thread.h"
  15 #include "sql/statement.h"
  16
  17 #include "testing/gtest/include/gtest/gtest.h"
  18
  19 using base::Time;
  20 using base::TimeDelta;
  21 using content::BrowserThread;
  22 using predictors::AutocompleteActionPredictorTable;
  23
  24 namespace predictors {
  25
  26 class AutocompleteActionPredictorTableTest : public testing::Test {
  27  public:
  28   AutocompleteActionPredictorTableTest();
  29   virtual ~AutocompleteActionPredictorTableTest();
  30
  31   virtual void SetUp();
  32   virtual void TearDown();
  33
  34   size_t CountRecords() const;
  35
  36   void AddAll();
  37
  38   bool RowsAreEqual(const AutocompleteActionPredictorTable::Row& lhs,
  39                     const AutocompleteActionPredictorTable::Row& rhs) const;
  40
  41   TestingProfile* profile() { return &profile_; }
  42
  43  protected:
  44
  45   // Test functions that can be run against this text fixture or
  46   // AutocompleteActionPredictorTableReopenTest that inherits from this.
  47   void TestGetRow();
  48   void TestAddAndUpdateRows();
  49   void TestDeleteRows();
  50   void TestDeleteAllRows();
  51
  52   AutocompleteActionPredictorTable::Rows test_db_;
  53
  54  private:
  55   TestingProfile profile_;
  56   scoped_ptr<PredictorDatabase> db_;
  57   base::MessageLoop loop_;
  58   content::TestBrowserThread db_thread_;
  59 };
  60
  61 class AutocompleteActionPredictorTableReopenTest
  62     : public AutocompleteActionPredictorTableTest {
  63  public:
  64   virtual void SetUp() {
  65     // By calling SetUp twice, we make sure that the table already exists for
  66     // this fixture.
  67     AutocompleteActionPredictorTableTest::SetUp();
  68     AutocompleteActionPredictorTableTest::TearDown();
  69     AutocompleteActionPredictorTableTest::SetUp();
  70   }
  71 };
  72
  73 AutocompleteActionPredictorTableTest::AutocompleteActionPredictorTableTest()
  74     : loop_(base::MessageLoop::TYPE_DEFAULT),
  75       db_thread_(BrowserThread::DB, &loop_) {
  76 }
  77
  78 AutocompleteActionPredictorTableTest::~AutocompleteActionPredictorTableTest() {
  79 }
  80
  81 void AutocompleteActionPredictorTableTest::SetUp() {
  82   db_.reset(new PredictorDatabase(&profile_));
  83   loop_.RunUntilIdle();
  84
  85   test_db_.push_back(AutocompleteActionPredictorTable::Row(
  86       "BD85DBA2-8C29-49F9-84AE-48E1E90880DF",
  87       ASCIIToUTF16("goog"), GURL("http://www.google.com/"),
  88       1, 0));
  89   test_db_.push_back(AutocompleteActionPredictorTable::Row(
  90       "BD85DBA2-8C29-49F9-84AE-48E1E90880E0",
  91       ASCIIToUTF16("slash"), GURL("http://slashdot.org/"),
  92       3, 2));
  93   test_db_.push_back(AutocompleteActionPredictorTable::Row(
  94       "BD85DBA2-8C29-49F9-84AE-48E1E90880E1",
  95       ASCIIToUTF16("news"), GURL("http://slashdot.org/"),
  96       0, 1));
  97 }
  98
  99 void AutocompleteActionPredictorTableTest::TearDown() {
 100   db_.reset(NULL);
 101   loop_.RunUntilIdle();
 102   test_db_.clear();
 103 }
 104
 105 size_t AutocompleteActionPredictorTableTest::CountRecords() const {
 106   sql::Statement s(db_->GetDatabase()->GetUniqueStatement(
 107       "SELECT count(*) FROM network_action_predictor"));
 108   EXPECT_TRUE(s.Step());
 109   return static_cast<size_t>(s.ColumnInt(0));
 110 }
 111
 112 void AutocompleteActionPredictorTableTest::AddAll() {
 113   db_->autocomplete_table()->AddAndUpdateRows(
 114       test_db_, AutocompleteActionPredictorTable::Rows());
 115
 116   EXPECT_EQ(test_db_.size(), CountRecords());
 117 }
 118
 119 bool AutocompleteActionPredictorTableTest::RowsAreEqual(
 120     const AutocompleteActionPredictorTable::Row& lhs,
 121     const AutocompleteActionPredictorTable::Row& rhs) const {
 122   return (lhs.id == rhs.id &&
 123           lhs.user_text == rhs.user_text &&
 124           lhs.url == rhs.url &&
 125           lhs.number_of_hits == rhs.number_of_hits &&
 126           lhs.number_of_misses == rhs.number_of_misses);
 127 }
 128
 129 void AutocompleteActionPredictorTableTest::TestGetRow() {
 130   db_->autocomplete_table()->AddAndUpdateRows(
 131       AutocompleteActionPredictorTable::Rows(1, test_db_[0]),
 132       AutocompleteActionPredictorTable::Rows());
 133   AutocompleteActionPredictorTable::Row row;
 134   db_->autocomplete_table()->GetRow(test_db_[0].id, &row);
 135   EXPECT_TRUE(RowsAreEqual(test_db_[0], row))
 136       << "Expected: Row with id " << test_db_[0].id << "\n"
 137       << "Got:      Row with id " << row.id;
 138 }
 139
 140 void AutocompleteActionPredictorTableTest::TestAddAndUpdateRows() {
 141   EXPECT_EQ(0U, CountRecords());
 142
 143   AutocompleteActionPredictorTable::Rows rows_to_add;
 144   rows_to_add.push_back(test_db_[0]);
 145   rows_to_add.push_back(test_db_[1]);
 146   db_->autocomplete_table()->AddAndUpdateRows(
 147       rows_to_add,
 148       AutocompleteActionPredictorTable::Rows());
 149   EXPECT_EQ(2U, CountRecords());
 150
 151   AutocompleteActionPredictorTable::Row row1 = test_db_[1];
 152   row1.number_of_hits = row1.number_of_hits + 1;
 153   db_->autocomplete_table()->AddAndUpdateRows(
 154       AutocompleteActionPredictorTable::Rows(1, test_db_[2]),
 155       AutocompleteActionPredictorTable::Rows(1, row1));
 156   EXPECT_EQ(3U, CountRecords());
 157
 158   AutocompleteActionPredictorTable::Row updated_row1;
 159   db_->autocomplete_table()->GetRow(test_db_[1].id, &updated_row1);
 160   EXPECT_TRUE(RowsAreEqual(row1, updated_row1))
 161       << "Expected: Row with id " << row1.id << "\n"
 162       << "Got:      Row with id " << updated_row1.id;
 163
 164   AutocompleteActionPredictorTable::Row row0 = test_db_[0];
 165   row0.number_of_hits = row0.number_of_hits + 2;
 166   AutocompleteActionPredictorTable::Row row2 = test_db_[2];
 167   row2.number_of_hits = row2.number_of_hits + 2;
 168   AutocompleteActionPredictorTable::Rows rows_to_update;
 169   rows_to_update.push_back(row0);
 170   rows_to_update.push_back(row2);
 171   db_->autocomplete_table()->AddAndUpdateRows(
 172       AutocompleteActionPredictorTable::Rows(),
 173       rows_to_update);
 174   EXPECT_EQ(3U, CountRecords());
 175
 176   AutocompleteActionPredictorTable::Row updated_row0, updated_row2;
 177   db_->autocomplete_table()->GetRow(test_db_[0].id, &updated_row0);
 178   db_->autocomplete_table()->GetRow(test_db_[2].id, &updated_row2);
 179   EXPECT_TRUE(RowsAreEqual(row0, updated_row0))
 180       << "Expected: Row with id " << row0.id << "\n"
 181       << "Got:      Row with id " << updated_row0.id;
 182   EXPECT_TRUE(RowsAreEqual(row2, updated_row2))
 183       << "Expected: Row with id " << row2.id << "\n"
 184       << "Got:      Row with id " << updated_row2.id;
 185 }
 186
 187 void AutocompleteActionPredictorTableTest::TestDeleteRows() {
 188   AddAll();
 189   std::vector<AutocompleteActionPredictorTable::Row::Id> id_list;
 190   id_list.push_back(test_db_[0].id);
 191   id_list.push_back(test_db_[2].id);
 192   db_->autocomplete_table()->DeleteRows(id_list);
 193   EXPECT_EQ(test_db_.size() - 2, CountRecords());
 194
 195   AutocompleteActionPredictorTable::Row row;
 196   db_->autocomplete_table()->GetRow(test_db_[1].id, &row);
 197   EXPECT_TRUE(RowsAreEqual(test_db_[1], row));
 198 }
 199
 200 void AutocompleteActionPredictorTableTest::TestDeleteAllRows() {
 201   AddAll();
 202   db_->autocomplete_table()->DeleteAllRows();
 203   EXPECT_EQ(0U, CountRecords());
 204 }
 205
 206 // AutocompleteActionPredictorTableTest tests
 207 TEST_F(AutocompleteActionPredictorTableTest, GetRow) {
 208   TestGetRow();
 209 }
 210
 211 TEST_F(AutocompleteActionPredictorTableTest, AddAndUpdateRows) {
 212   TestAddAndUpdateRows();
 213 }
 214
 215 TEST_F(AutocompleteActionPredictorTableTest, DeleteRows) {
 216   TestDeleteRows();
 217 }
 218
 219 TEST_F(AutocompleteActionPredictorTableTest, DeleteAllRows) {
 220   TestDeleteAllRows();
 221 }
 222
 223 // AutocompleteActionPredictorTableReopenTest tests
 224 TEST_F(AutocompleteActionPredictorTableReopenTest, GetRow) {
 225   TestGetRow();
 226 }
 227
 228 TEST_F(AutocompleteActionPredictorTableReopenTest, AddAndUpdateRows) {
 229   TestAddAndUpdateRows();
 230 }
 231
 232 TEST_F(AutocompleteActionPredictorTableReopenTest, DeleteRows) {
 233   TestDeleteRows();
 234 }
 235
 236 TEST_F(AutocompleteActionPredictorTableReopenTest, DeleteAllRows) {
 237   TestDeleteAllRows();
 238 }
 239
 240 }  // namespace predictors