OLD | NEW |
1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2009 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 entry. | 3 // found in the LICENSE entry. |
4 | 4 |
5 #include <string> | 5 #include <string> |
6 | 6 |
7 #include "base/file_util.h" | 7 #include "base/file_util.h" |
8 #include "base/test/test_file_util.h" | 8 #include "base/test/test_file_util.h" |
9 #include "chrome/browser/sync/syncable/directory_manager.h" | 9 #include "chrome/browser/sync/syncable/directory_manager.h" |
10 #include "chrome/browser/sync/util/character_set_converters.h" | 10 #include "chrome/browser/sync/util/character_set_converters.h" |
11 #include "chrome/browser/sync/util/user_settings.h" | 11 #include "chrome/browser/sync/util/user_settings.h" |
12 #include "chrome/browser/sync/util/query_helpers.h" | 12 #include "chrome/browser/sync/util/query_helpers.h" |
13 #include "testing/gtest/include/gtest/gtest.h" | 13 #include "testing/gtest/include/gtest/gtest.h" |
14 | 14 |
| 15 using browser_sync::FilePathToUTF8; |
15 using browser_sync::UserSettings; | 16 using browser_sync::UserSettings; |
16 | 17 |
17 static const PathChar* kV10UserSettingsDB = PSTR("Version10Settings.sqlite3"); | 18 static const FilePath::CharType kV10UserSettingsDB[] = |
18 static const PathChar* kOldStyleSyncDataDB = PSTR("OldStyleSyncData.sqlite3"); | 19 FILE_PATH_LITERAL("Version10Settings.sqlite3"); |
| 20 static const FilePath::CharType kOldStyleSyncDataDB[] = |
| 21 FILE_PATH_LITERAL("OldStyleSyncData.sqlite3"); |
| 22 static const FilePath::CharType kSyncDataDB[] = |
| 23 FILE_PATH_LITERAL("SyncData.sqlite3"); |
19 | 24 |
20 class UserSettingsTest : public testing::Test { | 25 class UserSettingsTest : public testing::Test { |
21 public: | 26 public: |
22 UserSettingsTest() : sync_data_("Some sync data") { } | 27 UserSettingsTest() : sync_data_("Some sync data") { } |
23 void SetUpVersion10Databases() { | 28 void SetUpVersion10Databases() { |
24 CleanUpVersion10Databases(); | 29 CleanUpVersion10Databases(); |
25 sqlite3* primer_handle = NULL; | 30 sqlite3* primer_handle = NULL; |
26 ASSERT_TRUE(SQLITE_OK == SqliteOpen(kV10UserSettingsDB, | 31 ASSERT_TRUE(SQLITE_OK == SqliteOpen(FilePath(kV10UserSettingsDB), |
27 &primer_handle)); | 32 &primer_handle)); |
28 FilePath old_sync_data(kOldStyleSyncDataDB); | 33 FilePath old_sync_data(kOldStyleSyncDataDB); |
29 | 34 |
30 ASSERT_TRUE(sync_data_.length() == | 35 ASSERT_TRUE(sync_data_.length() == |
31 static_cast<size_t>(file_util::WriteFile( | 36 static_cast<size_t>(file_util::WriteFile( |
32 old_sync_data, sync_data_.data(), sync_data_.length()))); | 37 old_sync_data, sync_data_.data(), sync_data_.length()))); |
33 | 38 |
34 // Create settings table. | 39 // Create settings table. |
35 ExecOrDie(primer_handle, "CREATE TABLE settings" | 40 ExecOrDie(primer_handle, "CREATE TABLE settings" |
36 " (email, key, value, " | 41 " (email, key, value, " |
37 " PRIMARY KEY(email, key) ON CONFLICT REPLACE)"); | 42 " PRIMARY KEY(email, key) ON CONFLICT REPLACE)"); |
38 | 43 |
39 // Create and populate version table. | 44 // Create and populate version table. |
40 ExecOrDie(primer_handle, "CREATE TABLE db_version ( version )"); | 45 ExecOrDie(primer_handle, "CREATE TABLE db_version ( version )"); |
41 ExecOrDie(primer_handle, "INSERT INTO db_version values ( ? )", 10); | 46 ExecOrDie(primer_handle, "INSERT INTO db_version values ( ? )", 10); |
42 // Create shares table. | 47 // Create shares table. |
43 ExecOrDie(primer_handle, "CREATE TABLE shares" | 48 ExecOrDie(primer_handle, "CREATE TABLE shares" |
44 " (email, share_name, file_name," | 49 " (email, share_name, file_name," |
45 " PRIMARY KEY(email, share_name) ON CONFLICT REPLACE)"); | 50 " PRIMARY KEY(email, share_name) ON CONFLICT REPLACE)"); |
46 // Populate a share. | 51 // Populate a share. |
47 ExecOrDie(primer_handle, "INSERT INTO shares values ( ?, ?, ?)", | 52 ExecOrDie(primer_handle, "INSERT INTO shares values ( ?, ?, ?)", |
48 "foo@foo.com", "foo@foo.com", | 53 "foo@foo.com", "foo@foo.com", FilePathToUTF8(old_sync_data)); |
49 browser_sync::PathStringToUTF8Quick(kOldStyleSyncDataDB)); | |
50 sqlite3_close(primer_handle); | 54 sqlite3_close(primer_handle); |
51 } | 55 } |
52 | 56 |
53 void CleanUpVersion10Databases() { | 57 void CleanUpVersion10Databases() { |
54 ASSERT_TRUE(file_util::DieFileDie(FilePath(kV10UserSettingsDB), false)); | 58 ASSERT_TRUE(file_util::DieFileDie(FilePath(kV10UserSettingsDB), false)); |
55 ASSERT_TRUE(file_util::DieFileDie(FilePath(kOldStyleSyncDataDB), false)); | 59 ASSERT_TRUE(file_util::DieFileDie(FilePath(kOldStyleSyncDataDB), false)); |
56 ASSERT_TRUE(file_util::DieFileDie(FilePath(PSTR("SyncData.sqlite3")), | 60 ASSERT_TRUE(file_util::DieFileDie(FilePath(kSyncDataDB), false)); |
57 false)); | |
58 } | 61 } |
59 | 62 |
60 const std::string& sync_data() const { return sync_data_; } | 63 const std::string& sync_data() const { return sync_data_; } |
61 | 64 |
62 private: | 65 private: |
63 std::string sync_data_; | 66 std::string sync_data_; |
64 }; | 67 }; |
65 | 68 |
66 TEST_F(UserSettingsTest, MigrateFromV10ToV11) { | 69 TEST_F(UserSettingsTest, MigrateFromV10ToV11) { |
67 SetUpVersion10Databases(); | 70 SetUpVersion10Databases(); |
68 { | 71 { |
69 // Create a UserSettings, which should trigger migration code. We do this | 72 // Create a UserSettings, which should trigger migration code. We do this |
70 // inside a scoped block so it closes itself and we can poke around to see | 73 // inside a scoped block so it closes itself and we can poke around to see |
71 // what happened later. | 74 // what happened later. |
72 UserSettings settings; | 75 UserSettings settings; |
73 settings.Init(kV10UserSettingsDB); | 76 settings.Init(FilePath(kV10UserSettingsDB)); |
74 } | 77 } |
75 | 78 |
76 // Now poke around using sqlite to see if UserSettings migrated properly. | 79 // Now poke around using sqlite to see if UserSettings migrated properly. |
77 sqlite3* handle = NULL; | 80 sqlite3* handle = NULL; |
78 ASSERT_TRUE(SQLITE_OK == SqliteOpen(kV10UserSettingsDB, &handle)); | 81 ASSERT_TRUE(SQLITE_OK == SqliteOpen(FilePath(kV10UserSettingsDB), &handle)); |
79 ScopedStatement version_query(PrepareQuery(handle, | 82 ScopedStatement version_query(PrepareQuery(handle, |
80 "SELECT version FROM db_version")); | 83 "SELECT version FROM db_version")); |
81 ASSERT_TRUE(SQLITE_ROW == sqlite3_step(version_query.get())); | 84 ASSERT_TRUE(SQLITE_ROW == sqlite3_step(version_query.get())); |
82 | 85 |
83 const int version = sqlite3_column_int(version_query.get(), 0); | 86 const int version = sqlite3_column_int(version_query.get(), 0); |
84 EXPECT_TRUE(11 == version); | 87 EXPECT_EQ(11, version); |
85 EXPECT_FALSE(file_util::PathExists(FilePath(kOldStyleSyncDataDB))); | 88 EXPECT_FALSE(file_util::PathExists(FilePath(kOldStyleSyncDataDB))); |
86 | 89 |
87 PathString path(syncable::DirectoryManager::GetSyncDataDatabaseFilename()); | 90 const FilePath& path = |
| 91 syncable::DirectoryManager::GetSyncDataDatabaseFilename(); |
88 | 92 |
89 std::string contents; | 93 std::string contents; |
90 ASSERT_TRUE(file_util::ReadFileToString(FilePath(path), &contents)); | 94 ASSERT_TRUE(file_util::ReadFileToString(FilePath(path), &contents)); |
91 EXPECT_TRUE(sync_data() == contents); | 95 EXPECT_TRUE(sync_data() == contents); |
92 } | 96 } |
OLD | NEW |