| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <stdio.h> | 5 #include "chrome/browser/history/in_memory_url_index.h" |
| 6 | 6 |
| 7 #include <fstream> | |
| 8 #include <string> | |
| 9 #include <vector> | |
| 10 | |
| 11 #include "app/sql/statement.h" | |
| 12 #include "base/file_util.h" | |
| 13 #include "base/path_service.h" | |
| 14 #include "base/scoped_ptr.h" | 7 #include "base/scoped_ptr.h" |
| 15 #include "base/time.h" | |
| 16 #include "base/utf_string_conversions.h" | |
| 17 #include "chrome/browser/history/in_memory_url_index.h" | |
| 18 #include "chrome/browser/history/url_database.h" | |
| 19 #include "chrome/common/chrome_paths.h" | |
| 20 #include "testing/gtest/include/gtest/gtest.h" | 8 #include "testing/gtest/include/gtest/gtest.h" |
| 21 | 9 |
| 22 // The test version of the history url database table ('url') is contained in | |
| 23 // a database file created from a text file('url_history_provider_test.db.txt'). | |
| 24 // The only difference between this table and a live 'urls' table from a | |
| 25 // profile is that the last_visit_time column in the test table contains a | |
| 26 // number specifying the number of days relative to 'today' to which the | |
| 27 // absolute time should be set during the test setup stage. | |
| 28 // | |
| 29 // The format of the test database text file is of a SQLite .dump file. | |
| 30 // Note that only lines whose first character is an upper-case letter are | |
| 31 // processed when creating the test database. | |
| 32 | |
| 33 using base::Time; | |
| 34 using base::TimeDelta; | |
| 35 | |
| 36 namespace history { | 10 namespace history { |
| 37 | 11 |
| 38 class InMemoryURLIndexTest : public testing::Test, | 12 class InMemoryURLIndexTest : public testing::Test { |
| 39 public URLDatabase { | |
| 40 public: | |
| 41 InMemoryURLIndexTest() {} | |
| 42 | |
| 43 protected: | 13 protected: |
| 44 // Test setup. | |
| 45 void SetUp() { | |
| 46 // Create and populate a working copy of the URL history database. | |
| 47 FilePath history_proto_path; | |
| 48 PathService::Get(chrome::DIR_TEST_DATA, &history_proto_path); | |
| 49 history_proto_path = history_proto_path.Append( | |
| 50 FILE_PATH_LITERAL("History")); | |
| 51 history_proto_path = history_proto_path.Append( | |
| 52 FILE_PATH_LITERAL("url_history_provider_test.db.txt")); | |
| 53 EXPECT_TRUE(file_util::PathExists(history_proto_path)); | |
| 54 EXPECT_TRUE(db_.OpenInMemory()); | |
| 55 | |
| 56 std::ifstream proto_file(history_proto_path.value().c_str()); | |
| 57 static const size_t kCommandBufferMaxSize = 2048; | |
| 58 char sql_cmd_line[kCommandBufferMaxSize]; | |
| 59 while (!proto_file.eof()) { | |
| 60 proto_file.getline(sql_cmd_line, kCommandBufferMaxSize); | |
| 61 if (!proto_file.eof()) { | |
| 62 // We only process lines which begin with a upper-case letter. | |
| 63 // TODO(mrossetti): Can iswupper() be used here? | |
| 64 if (sql_cmd_line[0] >= 'A' && sql_cmd_line[0] <= 'Z') { | |
| 65 std::string sql_cmd(sql_cmd_line); | |
| 66 sql::Statement sql_stmt(db_.GetUniqueStatement(sql_cmd_line)); | |
| 67 EXPECT_TRUE(sql_stmt.Run()); | |
| 68 } | |
| 69 } | |
| 70 } | |
| 71 proto_file.close(); | |
| 72 | |
| 73 // Update the last_visit_time table column | |
| 74 // such that it represents a time relative to 'now'. | |
| 75 sql::Statement statement(db_.GetUniqueStatement( | |
| 76 "SELECT" HISTORY_URL_ROW_FIELDS "FROM urls;")); | |
| 77 EXPECT_TRUE(statement); | |
| 78 Time time_right_now = Time::NowFromSystemTime(); | |
| 79 TimeDelta day_delta = TimeDelta::FromDays(1); | |
| 80 while (statement.Step()) { | |
| 81 URLRow row; | |
| 82 FillURLRow(statement, &row); | |
| 83 Time last_visit = time_right_now; | |
| 84 for (int64 i = row.last_visit().ToInternalValue(); i > 0; --i) | |
| 85 last_visit -= day_delta; | |
| 86 row.set_last_visit(last_visit); | |
| 87 UpdateURLRow(row.id(), row); | |
| 88 } | |
| 89 } | |
| 90 | |
| 91 void TearDown() { | |
| 92 db_.Close(); | |
| 93 } | |
| 94 | |
| 95 virtual sql::Connection& GetDB() { return db_; } | |
| 96 | |
| 97 scoped_ptr<InMemoryURLIndex> url_index_; | 14 scoped_ptr<InMemoryURLIndex> url_index_; |
| 98 sql::Connection db_; | |
| 99 }; | 15 }; |
| 100 | 16 |
| 101 TEST_F(InMemoryURLIndexTest, Construction) { | 17 TEST_F(InMemoryURLIndexTest, Construction) { |
| 102 url_index_.reset(new InMemoryURLIndex); | 18 url_index_.reset(new InMemoryURLIndex); |
| 103 EXPECT_TRUE(url_index_.get()); | 19 EXPECT_TRUE(url_index_.get()); |
| 104 } | 20 } |
| 105 | 21 |
| 106 TEST_F(InMemoryURLIndexTest, Initialization) { | |
| 107 url_index_.reset(new InMemoryURLIndex); | |
| 108 url_index_->Init(this, UTF8ToUTF16("en,ja,hi,zh")); | |
| 109 // There should have been 25 of the 29 urls accepted during filtering. | |
| 110 EXPECT_EQ(25U, url_index_->history_item_count_); | |
| 111 // The resulting indexes should account for: | |
| 112 // 37 characters | |
| 113 // 88 words | |
| 114 EXPECT_EQ(37U, url_index_->char_word_map_.size()); | |
| 115 EXPECT_EQ(88U, url_index_->word_map_.size()); | |
| 116 } | |
| 117 | |
| 118 } // namespace history | 22 } // namespace history |
| OLD | NEW |