| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 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 "components/omnibox/browser/shortcuts_database.h" | 5 #include "components/omnibox/browser/shortcuts_database.h" |
| 6 | 6 |
| 7 #include <stddef.h> | 7 #include <stddef.h> |
| 8 | 8 |
| 9 #include "base/files/scoped_temp_dir.h" | 9 #include "base/files/scoped_temp_dir.h" |
| 10 #include "base/format_macros.h" | 10 #include "base/format_macros.h" |
| (...skipping 78 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 89 void TearDown() override; | 89 void TearDown() override; |
| 90 | 90 |
| 91 void ClearDB(); | 91 void ClearDB(); |
| 92 size_t CountRecords() const; | 92 size_t CountRecords() const; |
| 93 | 93 |
| 94 ShortcutsDatabase::Shortcut ShortcutFromTestInfo( | 94 ShortcutsDatabase::Shortcut ShortcutFromTestInfo( |
| 95 const ShortcutsDatabaseTestInfo& info); | 95 const ShortcutsDatabaseTestInfo& info); |
| 96 | 96 |
| 97 void AddAll(); | 97 void AddAll(); |
| 98 | 98 |
| 99 base::ScopedTempDir temp_dir_; |
| 99 scoped_refptr<ShortcutsDatabase> db_; | 100 scoped_refptr<ShortcutsDatabase> db_; |
| 100 }; | 101 }; |
| 101 | 102 |
| 102 void ShortcutsDatabaseTest::SetUp() { | 103 void ShortcutsDatabaseTest::SetUp() { |
| 103 base::ScopedTempDir temp_dir; | 104 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
| 104 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 105 base::FilePath db_path(temp_dir_.path().Append(kShortcutsDatabaseName)); |
| 105 base::FilePath db_path(temp_dir.path().Append(kShortcutsDatabaseName)); | |
| 106 db_ = new ShortcutsDatabase(db_path); | 106 db_ = new ShortcutsDatabase(db_path); |
| 107 ASSERT_TRUE(db_->Init()); | 107 ASSERT_TRUE(db_->Init()); |
| 108 ClearDB(); | 108 ClearDB(); |
| 109 } | 109 } |
| 110 | 110 |
| 111 void ShortcutsDatabaseTest::TearDown() { | 111 void ShortcutsDatabaseTest::TearDown() { |
| 112 db_ = NULL; | 112 db_ = NULL; |
| 113 } | 113 } |
| 114 | 114 |
| 115 void ShortcutsDatabaseTest::ClearDB() { | 115 void ShortcutsDatabaseTest::ClearDB() { |
| (...skipping 247 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 363 // recovered. | 363 // recovered. |
| 364 { | 364 { |
| 365 sql::Connection connection; | 365 sql::Connection connection; |
| 366 ASSERT_TRUE(connection.Open(db_path)); | 366 ASSERT_TRUE(connection.Open(db_path)); |
| 367 sql::Statement statement(connection.GetUniqueStatement(kCountSql)); | 367 sql::Statement statement(connection.GetUniqueStatement(kCountSql)); |
| 368 ASSERT_TRUE(statement.is_valid()); | 368 ASSERT_TRUE(statement.is_valid()); |
| 369 ASSERT_TRUE(statement.Step()); | 369 ASSERT_TRUE(statement.Step()); |
| 370 EXPECT_EQ(row_count, statement.ColumnInt(0)); | 370 EXPECT_EQ(row_count, statement.ColumnInt(0)); |
| 371 } | 371 } |
| 372 } | 372 } |
| OLD | NEW |