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 "base/files/file_path.h" 6 #include "base/files/scoped_temp_dir.h" 7 #include "base/strings/utf_string_conversions.h" 8 #include "chrome/browser/history/android/android_cache_database.h" 9 #include "chrome/browser/history/android/android_time.h" 10 #include "chrome/browser/history/history_database.h" 11 #include "chrome/test/base/testing_profile.h" 12 #include "testing/gtest/include/gtest/gtest.h" 13 14 using base::Time; 15 using base::TimeDelta; 16 17 namespace history { 18 19 class AndroidCacheDatabaseTest : public testing::Test { 20 public: 21 AndroidCacheDatabaseTest() { 22 } 23 virtual ~AndroidCacheDatabaseTest() {} 24 25 protected: 26 virtual void SetUp() { 27 // Get a temporary directory for the test DB files. 28 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); 29 base::FilePath history_db_name_ = 30 temp_dir_.path().AppendASCII("history.db"); 31 android_cache_db_name_ = temp_dir_.path().AppendASCII( 32 "TestAndroidCache.db"); 33 ASSERT_EQ(sql::INIT_OK, history_db_.Init(history_db_name_)); 34 ASSERT_EQ(sql::INIT_OK, 35 history_db_.InitAndroidCacheDatabase(android_cache_db_name_)); 36 } 37 38 base::ScopedTempDir temp_dir_; 39 base::FilePath android_cache_db_name_; 40 HistoryDatabase history_db_; 41 }; 42 43 TEST(AndroidCacheDatabaseAttachTest, AttachDatabaseInTransactionNesting) { 44 base::ScopedTempDir temp_dir; 45 base::FilePath android_cache_db_name; 46 HistoryDatabase history_db; 47 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); 48 base::FilePath history_db_name = temp_dir.path().AppendASCII("history.db"); 49 android_cache_db_name = temp_dir.path().AppendASCII( 50 "TestAndroidCache.db"); 51 ASSERT_EQ(sql::INIT_OK, history_db.Init(history_db_name)); 52 // Create nested transactions. 53 history_db.BeginTransaction(); 54 history_db.BeginTransaction(); 55 history_db.BeginTransaction(); 56 int transaction_nesting = history_db.transaction_nesting(); 57 ASSERT_EQ(sql::INIT_OK, 58 history_db.InitAndroidCacheDatabase(android_cache_db_name)); 59 // The count of nested transaction is still same. 60 EXPECT_EQ(transaction_nesting, history_db.transaction_nesting()); 61 } 62 63 TEST_F(AndroidCacheDatabaseTest, InitAndroidCacheDatabase) { 64 // Try to run a sql against the table to verify them exist. 65 AndroidCacheDatabase* cache_db = 66 static_cast<AndroidCacheDatabase*>(&history_db_); 67 EXPECT_TRUE(cache_db->GetDB().Execute( 68 "DELETE FROM android_cache_db.bookmark_cache")); 69 EXPECT_TRUE(cache_db->GetDB().Execute( 70 "DELETE FROM android_cache_db.search_terms")); 71 } 72 73 TEST_F(AndroidCacheDatabaseTest, SearchTermsTable) { 74 // Test AddSearchTerm. 75 Time search_time1 = Time::Now() - TimeDelta::FromDays(1); 76 base::string16 search_term1(base::UTF8ToUTF16("search term 1")); 77 SearchTermID id1 = history_db_.AddSearchTerm(search_term1, search_time1); 78 ASSERT_TRUE(id1); 79 SearchTermRow row1; 80 ASSERT_EQ(id1, history_db_.GetSearchTerm(search_term1, &row1)); 81 EXPECT_EQ(search_term1, row1.term); 82 EXPECT_EQ(ToDatabaseTime(search_time1), 83 ToDatabaseTime(row1.last_visit_time)); 84 EXPECT_EQ(id1, row1.id); 85 86 // Test UpdateSearchTerm. 87 SearchTermRow update_row1; 88 update_row1.term = (base::UTF8ToUTF16("update search term1")); 89 update_row1.last_visit_time = Time::Now(); 90 ASSERT_TRUE(history_db_.UpdateSearchTerm(id1, update_row1)); 91 EXPECT_EQ(id1, history_db_.GetSearchTerm(update_row1.term, &row1)); 92 EXPECT_EQ(update_row1.term, row1.term); 93 EXPECT_EQ(ToDatabaseTime(update_row1.last_visit_time), 94 ToDatabaseTime(row1.last_visit_time)); 95 EXPECT_EQ(id1, row1.id); 96 97 Time search_time2 = Time::Now() - TimeDelta::FromHours(1); 98 base::string16 search_term2(base::UTF8ToUTF16("search term 2")); 99 SearchTermID id2 = history_db_.AddSearchTerm(search_term2, search_time2); 100 ASSERT_TRUE(id2); 101 ASSERT_TRUE(history_db_.SetKeywordSearchTermsForURL(1, 1, search_term2)); 102 ASSERT_TRUE(history_db_.DeleteUnusedSearchTerms()); 103 104 // The search_term1 was removed. 105 EXPECT_FALSE(history_db_.GetSearchTerm(update_row1.term, NULL)); 106 // The search_term2 should still in the table. 107 ASSERT_EQ(id2, history_db_.GetSearchTerm(search_term2, &row1)); 108 EXPECT_EQ(id2, row1.id); 109 EXPECT_EQ(ToDatabaseTime(search_time2), 110 ToDatabaseTime(row1.last_visit_time)); 111 EXPECT_EQ(search_term2, row1.term); 112 } 113 114 } // namespace history 115