src/chrome/browser/sync/test/integration/single_client_backup_rollback_test.cc

   1 // Copyright 2014 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/command_line.h"
   6 #include "base/message_loop/message_loop.h"
   7 #include "base/prefs/pref_service.h"
   8 #include "chrome/browser/profiles/profile.h"
   9 #include "chrome/browser/sync/profile_sync_service.h"
  10 #include "chrome/browser/sync/test/integration/bookmarks_helper.h"
  11 #include "chrome/browser/sync/test/integration/preferences_helper.h"
  12 #include "chrome/browser/sync/test/integration/sync_integration_test_util.h"
  13 #include "chrome/browser/sync/test/integration/sync_test.h"
  14 #include "chrome/common/chrome_switches.h"
  15 #include "chrome/common/pref_names.h"
  16 #include "components/bookmarks/browser/bookmark_model.h"
  17 #include "sync/test/fake_server/fake_server_verifier.h"
  18
  19 using bookmarks_helper::AddFolder;
  20 using bookmarks_helper::AddURL;
  21 using bookmarks_helper::GetOtherNode;
  22 using bookmarks_helper::ModelMatchesVerifier;
  23 using bookmarks_helper::Move;
  24 using bookmarks_helper::Remove;
  25 using sync_integration_test_util::AwaitCommitActivityCompletion;
  26
  27 namespace {
  28 const char kUrl1[] = "http://www.google.com";
  29 const char kUrl2[] = "http://map.google.com";
  30 const char kUrl3[] = "http://plus.google.com";
  31 }  // anonymous namespace
  32
  33 class SingleClientBackupRollbackTest : public SyncTest {
  34  public:
  35   SingleClientBackupRollbackTest() : SyncTest(SINGLE_CLIENT) {}
  36   virtual ~SingleClientBackupRollbackTest() {}
  37
  38   void DisableBackup() {
  39     CommandLine::ForCurrentProcess()->AppendSwitch(
  40           switches::kSyncDisableBackup);
  41   }
  42
  43   void EnableRollback() {
  44     CommandLine::ForCurrentProcess()->AppendSwitch(
  45           switches::kSyncEnableRollback);
  46   }
  47
  48  private:
  49   DISALLOW_COPY_AND_ASSIGN(SingleClientBackupRollbackTest);
  50 };
  51
  52 class BackupModeChecker {
  53  public:
  54   explicit BackupModeChecker(ProfileSyncService* service,
  55                              base::TimeDelta timeout)
  56       : pss_(service),
  57         timeout_(timeout) {}
  58
  59   bool Wait() {
  60     expiration_ = base::TimeTicks::Now() + timeout_;
  61     base::MessageLoop::current()->PostDelayedTask(
  62         FROM_HERE,
  63         base::Bind(&BackupModeChecker::PeriodicCheck, base::Unretained(this)),
  64         base::TimeDelta::FromSeconds(1));
  65     base::MessageLoop::current()->Run();
  66     return IsBackupComplete();
  67   }
  68
  69  private:
  70   void PeriodicCheck() {
  71     if (IsBackupComplete() || base::TimeTicks::Now() > expiration_) {
  72       base::MessageLoop::current()->Quit();
  73     } else {
  74       base::MessageLoop::current()->PostDelayedTask(
  75           FROM_HERE,
  76           base::Bind(&BackupModeChecker::PeriodicCheck, base::Unretained(this)),
  77           base::TimeDelta::FromSeconds(1));
  78     }
  79   }
  80
  81   bool IsBackupComplete() {
  82     return pss_->backend_mode() == ProfileSyncService::BACKUP &&
  83         pss_->ShouldPushChanges();
  84   }
  85
  86   ProfileSyncService* pss_;
  87   base::TimeDelta timeout_;
  88   base::TimeTicks expiration_;
  89 };
  90
  91 #if defined(ENABLE_PRE_SYNC_BACKUP)
  92 #define MAYBE_TestBackupRollback TestBackupRollback
  93 #else
  94 #define MAYBE_TestBackupRollback DISABLED_TestBackupRollback
  95 #endif
  96 IN_PROC_BROWSER_TEST_F(SingleClientBackupRollbackTest,
  97                        MAYBE_TestBackupRollback) {
  98   EnableRollback();
  99   ASSERT_TRUE(SetupClients()) << "SetupClients() failed.";
 100
 101   // Starting state:
 102   // other_node
 103   //    -> top
 104   //      -> tier1_a
 105   //        -> http://mail.google.com  "tier1_a_url0"
 106   //      -> tier1_b
 107   //        -> http://www.nhl.com "tier1_b_url0"
 108   const BookmarkNode* top = AddFolder(0, GetOtherNode(0), 0, "top");
 109   const BookmarkNode* tier1_a = AddFolder(0, top, 0, "tier1_a");
 110   const BookmarkNode* tier1_b = AddFolder(0, top, 1, "tier1_b");
 111   ASSERT_TRUE(AddURL(0, tier1_a, 0, "tier1_a_url0",
 112                      GURL("http://mail.google.com")));
 113   ASSERT_TRUE(AddURL(0, tier1_b, 0, "tier1_b_url0",
 114                      GURL("http://www.nhl.com")));
 115
 116   BackupModeChecker checker(GetSyncService(0),
 117                             base::TimeDelta::FromSeconds(15));
 118   ASSERT_TRUE(checker.Wait());
 119
 120   // Setup sync, wait for its completion, and make sure changes were synced.
 121   ASSERT_TRUE(SetupSync()) << "SetupSync() failed.";
 122   ASSERT_TRUE(AwaitCommitActivityCompletion(GetSyncService((0))));
 123   ASSERT_TRUE(ModelMatchesVerifier(0));
 124
 125   // Made bookmark changes while sync is on.
 126   Move(0, tier1_a->GetChild(0), tier1_b, 1);
 127   Remove(0, tier1_b, 0);
 128   ASSERT_TRUE(AddFolder(0, tier1_b, 1, "tier2_c"));
 129   ASSERT_TRUE(AwaitCommitActivityCompletion(GetSyncService((0))));
 130   ASSERT_TRUE(ModelMatchesVerifier(0));
 131
 132   // Let server to return rollback command on next sync request.
 133   GetFakeServer()->TriggerError(sync_pb::SyncEnums::USER_ROLLBACK);
 134
 135   // Make another change to trigger downloading of rollback command.
 136   Remove(0, tier1_b, 0);
 137
 138   // Wait for sync to switch to backup mode after finishing rollback.
 139   ASSERT_TRUE(checker.Wait());
 140
 141   // Verify bookmarks are restored.
 142   ASSERT_EQ(1, tier1_a->child_count());
 143   const BookmarkNode* url1 = tier1_a->GetChild(0);
 144   ASSERT_EQ(GURL("http://mail.google.com"), url1->url());
 145
 146   ASSERT_EQ(1, tier1_b->child_count());
 147   const BookmarkNode* url2 = tier1_b->GetChild(0);
 148   ASSERT_EQ(GURL("http://www.nhl.com"), url2->url());
 149 }
 150
 151 #if defined(ENABLE_PRE_SYNC_BACKUP)
 152 #define MAYBE_TestPrefBackupRollback TestPrefBackupRollback
 153 #else
 154 #define MAYBE_TestPrefBackupRollback DISABLED_TestPrefBackupRollback
 155 #endif
 156 // Verify local preferences are not affected by preferences in backup DB under
 157 // backup mode.
 158 IN_PROC_BROWSER_TEST_F(SingleClientBackupRollbackTest,
 159                        MAYBE_TestPrefBackupRollback) {
 160   EnableRollback();
 161
 162   ASSERT_TRUE(SetupClients()) << "SetupClients() failed.";
 163
 164   preferences_helper::ChangeStringPref(0, prefs::kHomePage, kUrl1);
 165
 166   BackupModeChecker checker(GetSyncService(0),
 167                             base::TimeDelta::FromSeconds(15));
 168   ASSERT_TRUE(checker.Wait());
 169
 170   // Shut down backup, then change preference.
 171   GetSyncService(0)->StartStopBackupForTesting();
 172   preferences_helper::ChangeStringPref(0, prefs::kHomePage, kUrl2);
 173
 174   // Restart backup. Preference shouldn't change after backup starts.
 175   GetSyncService(0)->StartStopBackupForTesting();
 176   ASSERT_TRUE(checker.Wait());
 177   ASSERT_EQ(kUrl2,
 178             preferences_helper::GetPrefs(0)->GetString(prefs::kHomePage));
 179
 180   // Start sync and change preference.
 181   ASSERT_TRUE(SetupSync()) << "SetupSync() failed.";
 182   preferences_helper::ChangeStringPref(0, prefs::kHomePage, kUrl3);
 183   ASSERT_TRUE(AwaitCommitActivityCompletion(GetSyncService((0))));
 184   ASSERT_TRUE(ModelMatchesVerifier(0));
 185
 186   // Let server return rollback command on next sync request.
 187   GetFakeServer()->TriggerError(sync_pb::SyncEnums::USER_ROLLBACK);
 188
 189   // Make another change to trigger downloading of rollback command.
 190   preferences_helper::ChangeStringPref(0, prefs::kHomePage, "");
 191
 192   // Wait for sync to switch to backup mode after finishing rollback.
 193   ASSERT_TRUE(checker.Wait());
 194
 195   // Verify preference is restored.
 196   ASSERT_EQ(kUrl2,
 197             preferences_helper::GetPrefs(0)->GetString(prefs::kHomePage));
 198 }
 199
 200 #if defined(ENABLE_PRE_SYNC_BACKUP)
 201 #define MAYBE_RollbackNoBackup RollbackNoBackup
 202 #else
 203 #define MAYBE_RollbackNoBackup DISABLED_RollbackNoBackup
 204 #endif
 205 IN_PROC_BROWSER_TEST_F(SingleClientBackupRollbackTest,
 206                        MAYBE_RollbackNoBackup) {
 207   EnableRollback();
 208
 209   ASSERT_TRUE(SetupClients()) << "SetupClients() failed.";
 210
 211   // Setup sync, wait for its completion, and make sure changes were synced.
 212   ASSERT_TRUE(SetupSync()) << "SetupSync() failed.";
 213
 214   // Starting state:
 215   // other_node
 216   //    -> http://mail.google.com  "url0"
 217   //    -> http://www.nhl.com "url1"
 218   ASSERT_TRUE(AddURL(0, GetOtherNode(0), 0, "url0",
 219                      GURL("http://mail.google.com")));
 220   ASSERT_TRUE(AddURL(0, GetOtherNode(0), 1, "url1",
 221                      GURL("http://www.nhl.com")));
 222
 223   ASSERT_TRUE(AwaitCommitActivityCompletion(GetSyncService((0))));
 224   ASSERT_TRUE(ModelMatchesVerifier(0));
 225
 226   // Let server to return rollback command on next sync request.
 227   GetFakeServer()->TriggerError(sync_pb::SyncEnums::USER_ROLLBACK);
 228
 229   // Make another change to trigger downloading of rollback command.
 230   Remove(0, GetOtherNode(0), 0);
 231
 232   // Wait for sync to switch to backup mode after finishing rollback.
 233   BackupModeChecker checker(GetSyncService(0),
 234                             base::TimeDelta::FromSeconds(15));
 235   ASSERT_TRUE(checker.Wait());
 236
 237   // Without backup DB, bookmarks added during sync shouldn't be removed.
 238   ASSERT_EQ(1, GetOtherNode(0)->child_count());
 239   ASSERT_EQ(GURL("http://www.nhl.com"),
 240             GetOtherNode(0)->GetChild(0)->url());
 241 }
 242
 243 #if defined(ENABLE_PRE_SYNC_BACKUP)
 244 #define MAYBE_DontChangeBookmarkOrdering DontChangeBookmarkOrdering
 245 #else
 246 #define MAYBE_DontChangeBookmarkOrdering DISABLED_DontChangeBookmarkOrdering
 247 #endif
 248 IN_PROC_BROWSER_TEST_F(SingleClientBackupRollbackTest,
 249                        MAYBE_DontChangeBookmarkOrdering) {
 250   ASSERT_TRUE(SetupClients()) << "SetupClients() failed.";
 251
 252   const BookmarkNode* sub_folder = AddFolder(0, GetOtherNode(0), 0, "test");
 253   ASSERT_TRUE(AddURL(0, sub_folder, 0, "", GURL(kUrl1)));
 254   ASSERT_TRUE(AddURL(0, sub_folder, 1, "", GURL(kUrl2)));
 255   ASSERT_TRUE(AddURL(0, sub_folder, 2, "", GURL(kUrl3)));
 256
 257   BackupModeChecker checker(GetSyncService(0),
 258                             base::TimeDelta::FromSeconds(15));
 259   ASSERT_TRUE(checker.Wait());
 260
 261   // Restart backup.
 262   GetSyncService(0)->StartStopBackupForTesting();
 263   GetSyncService(0)->StartStopBackupForTesting();
 264   ASSERT_TRUE(checker.Wait());
 265
 266   // Verify bookmarks are unchanged.
 267   ASSERT_EQ(3, sub_folder->child_count());
 268   ASSERT_EQ(GURL(kUrl1), sub_folder->GetChild(0)->url());
 269   ASSERT_EQ(GURL(kUrl2), sub_folder->GetChild(1)->url());
 270   ASSERT_EQ(GURL(kUrl3), sub_folder->GetChild(2)->url());
 271 }
 272