Index: sql/recovery_unittest.cc |
diff --git a/sql/recovery_unittest.cc b/sql/recovery_unittest.cc |
index 02ede5b0463c24f4fae18f9ebfe7992bca1c26c9..44b9b49a232f963b649475824b51a6d30a25f4f5 100644 |
--- a/sql/recovery_unittest.cc |
+++ b/sql/recovery_unittest.cc |
@@ -16,6 +16,7 @@ |
#include "base/files/scoped_temp_dir.h" |
#include "base/path_service.h" |
#include "base/strings/string_number_conversions.h" |
+#include "base/strings/stringprintf.h" |
#include "base/test/histogram_tester.h" |
#include "sql/connection.h" |
#include "sql/meta_table.h" |
@@ -839,4 +840,77 @@ TEST_F(SQLRecoveryTest, AttachFailure) { |
tester.ExpectBucketCount(kErrorHistogramName, SQLITE_NOTADB, 1); |
} |
-} // namespace |
+// Helper for SQLRecoveryTest.PageSize. Creates a fresh db based on db_prefix, |
+// with the given initial page size, and verifies it against the expected size. |
+// Then changes to the final page size and recovers, verifying that the |
+// recovered database ends up with the expected final page size. |
+void TestPageSize(const base::FilePath& db_prefix, |
+ int initial_page_size, |
+ const std::string& expected_initial_page_size, |
+ int final_page_size, |
+ const std::string& expected_final_page_size) { |
+ SCOPED_TRACE( |
+ base::StringPrintf("page sizes {initial=%d/%s, final=%d/%s}", |
+ initial_page_size, expected_initial_page_size.c_str(), |
+ final_page_size, expected_final_page_size.c_str())); |
+ |
+ const char kCreateSql[] = "CREATE TABLE x (t TEXT)"; |
+ const char kInsertSql1[] = "INSERT INTO x VALUES ('This is a test')"; |
+ const char kInsertSql2[] = "INSERT INTO x VALUES ('That was a test')"; |
+ const char kSelectSql[] = "SELECT * FROM x ORDER BY t"; |
+ |
+ const base::FilePath db_path = db_prefix.InsertBeforeExtensionASCII( |
+ base::StringPrintf("%d", initial_page_size)); |
+ sql::Connection::Delete(db_path); |
+ sql::Connection db; |
+ db.set_page_size(initial_page_size); |
+ ASSERT_TRUE(db.Open(db_path)); |
+ ASSERT_TRUE(db.Execute(kCreateSql)); |
+ ASSERT_TRUE(db.Execute(kInsertSql1)); |
+ ASSERT_TRUE(db.Execute(kInsertSql2)); |
+ ASSERT_EQ(expected_initial_page_size, |
+ ExecuteWithResult(&db, "PRAGMA page_size")); |
+ |
+ // Recovery will use the page size set in the connection object, which may not |
+ // match the file's page size. |
+ db.set_page_size(final_page_size); |
+ sql::Recovery::RecoverDatabase(&db, db_path); |
+ |
+ // Recovery poisoned the handle, must re-open. |
+ db.Close(); |
+ |
+ // Make sure the page size is read from the file. |
+ db.set_page_size(0); |
+ ASSERT_TRUE(db.Open(db_path)); |
+ ASSERT_EQ(expected_final_page_size, |
+ ExecuteWithResult(&db, "PRAGMA page_size")); |
+ EXPECT_EQ("That was a test\nThis is a test", |
+ ExecuteWithResults(&db, kSelectSql, "|", "\n")); |
+} |
+ |
+// Verify that sql::Recovery maintains the page size, and the virtual table |
+// works with page sizes other than SQLite's default. Also verify the case |
+// where the default page size has changed. |
+TEST_F(SQLRecoveryTest, PageSize) { |
+ const std::string default_page_size = |
+ ExecuteWithResult(&db(), "PRAGMA page_size"); |
+ |
+ // The database should have the default page size after recovery. |
michaeln
2016/10/27 00:02:29
SCOPED_TRACE looks like a good tool for this
alte
Scott Hess - ex-Googler
2016/10/27 19:14:58
Indeed, and that removes the need for convoluted s
|
+ TestPageSize(db_path(), 0, default_page_size, 0, default_page_size); |
+ |
+ // Sync user 32k pages. |
+ TestPageSize(db_path(), 32768, "32768", 32768, "32768"); |
+ |
+ // Many clients use 4k pages. This is the SQLite default after 3.12.0. |
+ TestPageSize(db_path(), 4096, "4096", 4096, "4096"); |
+ |
+ // 1k is the default page size before 3.12.0. |
+ TestPageSize(db_path(), 1024, "1024", 1024, "1024"); |
+ |
+ // Databases with no page size specified should recover with the new default |
+ // page size. 2k has never been the default page size. |
+ ASSERT_NE("2048", default_page_size); |
+ TestPageSize(db_path(), 2048, "2048", 0, default_page_size); |
+} |
+ |
+} // namespac |