| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 #ifndef SQL_TEST_TEST_HELPERS_H_ | 5 #ifndef SQL_TEST_TEST_HELPERS_H_ |
| 6 #define SQL_TEST_TEST_HELPERS_H_ | 6 #define SQL_TEST_TEST_HELPERS_H_ |
| 7 | 7 |
| 8 #include <string> | |
| 9 | |
| 10 #include "base/basictypes.h" | 8 #include "base/basictypes.h" |
| 11 #include "base/compiler_specific.h" | 9 #include "base/compiler_specific.h" |
| 12 | 10 |
| 13 // Collection of test-only convenience functions. | 11 // Collection of test-only convenience functions. |
| 14 | 12 |
| 15 namespace base { | 13 namespace base { |
| 16 class FilePath; | 14 class FilePath; |
| 17 } | 15 } |
| 18 | 16 |
| 19 namespace sql { | 17 namespace sql { |
| 20 class Connection; | 18 class Connection; |
| 21 } | 19 } |
| 22 | 20 |
| 23 namespace sql { | 21 namespace sql { |
| 24 namespace test { | 22 namespace test { |
| 25 | 23 |
| 26 // SQLite stores the database size in the header, and if the actual | |
| 27 // OS-derived size is smaller, the database is considered corrupt. | |
| 28 // [This case is actually a common form of corruption in the wild.] | |
| 29 // This helper sets the in-header size to one page larger than the | |
| 30 // actual file size. The resulting file will return SQLITE_CORRUPT | |
| 31 // for most operations unless PRAGMA writable_schema is turned ON. | |
| 32 // | |
| 33 // Returns false if any error occurs accessing the file. | |
| 34 bool CorruptSizeInHeader(const base::FilePath& db_path) WARN_UNUSED_RESULT; | |
| 35 | |
| 36 // Return the number of tables in sqlite_master. | 24 // Return the number of tables in sqlite_master. |
| 37 size_t CountSQLTables(sql::Connection* db) WARN_UNUSED_RESULT; | 25 size_t CountSQLTables(sql::Connection* db) WARN_UNUSED_RESULT; |
| 38 | 26 |
| 39 // Return the number of indices in sqlite_master. | 27 // Return the number of indices in sqlite_master. |
| 40 size_t CountSQLIndices(sql::Connection* db) WARN_UNUSED_RESULT; | 28 size_t CountSQLIndices(sql::Connection* db) WARN_UNUSED_RESULT; |
| 41 | 29 |
| 42 // Returns the number of columns in the named table. 0 indicates an | 30 // Returns the number of columns in the named table. 0 indicates an |
| 43 // error (probably no such table). | 31 // error (probably no such table). |
| 44 size_t CountTableColumns(sql::Connection* db, const char* table) | 32 size_t CountTableColumns(sql::Connection* db, const char* table) |
| 45 WARN_UNUSED_RESULT; | 33 WARN_UNUSED_RESULT; |
| 46 | 34 |
| 47 // Sets |*count| to the number of rows in |table|. Returns false in | 35 // Sets |*count| to the number of rows in |table|. Returns false in |
| 48 // case of error, such as the table not existing. | 36 // case of error, such as the table not existing. |
| 49 bool CountTableRows(sql::Connection* db, const char* table, size_t* count); | 37 bool CountTableRows(sql::Connection* db, const char* table, size_t* count); |
| 50 | 38 |
| 51 // Creates a SQLite database at |db_path| from the sqlite .dump output | 39 // Creates a SQLite database at |db_path| from the sqlite .dump output |
| 52 // at |sql_path|. Returns false if |db_path| already exists, or if | 40 // at |sql_path|. Returns false if |db_path| already exists, or if |
| 53 // sql_path does not exist or cannot be read, or if there is an error | 41 // sql_path does not exist or cannot be read, or if there is an error |
| 54 // executing the statements. | 42 // executing the statements. |
| 55 bool CreateDatabaseFromSQL(const base::FilePath& db_path, | 43 bool CreateDatabaseFromSQL(const base::FilePath& db_path, |
| 56 const base::FilePath& sql_path) WARN_UNUSED_RESULT; | 44 const base::FilePath& sql_path) WARN_UNUSED_RESULT; |
| 57 | 45 |
| 58 // Return the results of running "PRAGMA integrity_check" on |db|. | |
| 59 // TODO(shess): sql::Connection::IntegrityCheck() is basically the | |
| 60 // same, but not as convenient for testing. Maybe combine. | |
| 61 std::string IntegrityCheck(sql::Connection* db) WARN_UNUSED_RESULT; | |
| 62 | |
| 63 } // namespace test | 46 } // namespace test |
| 64 } // namespace sql | 47 } // namespace sql |
| 65 | 48 |
| 66 #endif // SQL_TEST_TEST_HELPERS_H_ | 49 #endif // SQL_TEST_TEST_HELPERS_H_ |
| OLD | NEW |