sync: Refactor migration integration tests

chrome/browser/sync/test/integration/migration_waiter.h

Index: chrome/browser/sync/test/integration/migration_waiter.h
diff --git a/chrome/browser/sync/test/integration/migration_waiter.h b/chrome/browser/sync/test/integration/migration_waiter.h
new file mode 100644
index 0000000000000000000000000000000000000000..ebbc689843a8f208ca1601a0ac63b6f9e8e99836
--- /dev/null
+++ b/chrome/browser/sync/test/integration/migration_waiter.h
@@ -0,0 +1,47 @@
+// Copyright 2014 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#ifndef CHROME_BROWSER_SYNC_TEST_INTEGRATION_MIGRATION_WAITER_H_
+#define CHROME_BROWSER_SYNC_TEST_INTEGRATION_MIGRATION_WAITER_H_
+
+#include "base/macros.h"
+#include "chrome/browser/sync/test/integration/status_change_checker.h"
+#include "sync/internal_api/public/base/model_type.h"
+
+class MigrationWatcher;
+
+// Helper class that checks if the sync backend has successfully completed
+// migration for a set of data types.
+//
+// Collaborates with the MigrationWatcher, defined above.
+class MigrationWaiter : public StatusChangeChecker {
+ public:
+  // Initialize a waiter that will wait until |watcher|'s migrated types
+  // match the provided |exptected_types|.
+  MigrationWaiter(MigrationWatcher* watcher,

[Nicolas Zea, 2014/08/20 22:24:10]

nit: swap param order?

[rlarocque, 2014/08/21 00:41:20]

Done.

+                  syncer::ModelTypeSet expected_types);
+
+  virtual ~MigrationWaiter();
+
+  // Implementation of StatusChangeChecker.
+  virtual bool IsExitConditionSatisfied() OVERRIDE;
+  virtual std::string GetDebugMessage() const OVERRIDE;
+
+  // Function to trigger the waiting.
+  void Wait();
+
+  // Callback invoked by our associated waiter when migration state changes.
+  void OnMigrationStateChange();
+
+ private:
+  // The MigrationWatcher we're observering.
+  MigrationWatcher* watcher_;

[Nicolas Zea, 2014/08/20 22:24:10]

nit: make const pointer?

[rlarocque, 2014/08/21 00:41:19]

Done.

+
+  // The set of data types that are expected to eventually undergo migration.
+  syncer::ModelTypeSet expected_types_;

[Nicolas Zea, 2014/08/20 22:24:10]

const?

[rlarocque, 2014/08/21 00:41:20]

Done.

+
+  DISALLOW_COPY_AND_ASSIGN(MigrationWaiter);
+};
+
+#endif  // CHROME_BROWSER_SYNC_TEST_INTEGRATION_MIGRATION_WAITER_H_