Sync: Refactoring of DEVICE_INFO syncable type - Part 1

chrome/browser/sync/sessions/session_data_type_controller_unittest.cc

Index: chrome/browser/sync/sessions/session_data_type_controller_unittest.cc
diff --git a/chrome/browser/sync/sessions/session_data_type_controller_unittest.cc b/chrome/browser/sync/sessions/session_data_type_controller_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..5e64bb892f61e1176fe7799b72b848dc3fbacd0f
--- /dev/null
+++ b/chrome/browser/sync/sessions/session_data_type_controller_unittest.cc
@@ -0,0 +1,205 @@
+// 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.
+
+#include "base/bind.h"
+#include "base/callback.h"
+#include "base/memory/weak_ptr.h"
+#include "base/run_loop.h"
+#include "chrome/browser/chrome_notification_types.h"
+#include "chrome/browser/sync/glue/local_device_info_provider_mock.h"
+#include "chrome/browser/sync/glue/synced_window_delegate.h"
+#include "chrome/browser/sync/profile_sync_components_factory_mock.h"
+#include "chrome/browser/sync/sessions/session_data_type_controller.h"
+#include "chrome/browser/sync/sessions/synced_window_delegates_getter.h"
+#include "chrome/test/base/testing_profile.h"
+#include "content/public/browser/notification_service.h"
+#include "content/public/test/test_browser_thread_bundle.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace browser_sync {
+
+namespace {
+
+class MockSyncedWindowDelegate : public SyncedWindowDelegate {
+ public:
+  explicit MockSyncedWindowDelegate(Profile* profile)
+    : is_restore_in_progress_(false),
+      profile_(profile) {}
+  virtual ~MockSyncedWindowDelegate() {}
+
+  virtual bool HasWindow() const OVERRIDE { return false; }
+  virtual SessionID::id_type GetSessionId() const OVERRIDE { return 0; }
+  virtual int GetTabCount() const OVERRIDE { return 0; }
+  virtual int GetActiveIndex() const OVERRIDE { return 0; }
+  virtual bool IsApp() const OVERRIDE { return false; }
+  virtual bool IsTypeTabbed() const OVERRIDE { return false; }
+  virtual bool IsTypePopup() const OVERRIDE { return false; }
+  virtual bool IsTabPinned(const SyncedTabDelegate* tab) const OVERRIDE {
+    return false;
+  }
+  virtual SyncedTabDelegate* GetTabAt(int index) const OVERRIDE { return NULL; }
+  virtual SessionID::id_type GetTabIdAt(int index) const OVERRIDE { return 0; }
+
+  virtual bool IsSessionRestoreInProgress() const OVERRIDE {
+    return is_restore_in_progress_;
+  }
+
+  void SetSessionRestoreInProgress(bool is_restore_in_progress) {
+    is_restore_in_progress_ = is_restore_in_progress;
+
+    if (!is_restore_in_progress_) {
+      content::NotificationService::current()->Notify(
+          chrome::NOTIFICATION_SESSION_RESTORE_COMPLETE,
+          content::Source<Profile>(profile_),
+          content::NotificationService::NoDetails());
+    }
+  }
+
+ private:
+  bool is_restore_in_progress_;
+  Profile* profile_;
+};
+
+class MockSyncedWindowDelegatesGetter : public SyncedWindowDelegatesGetter {
+ public:
+  virtual const std::set<SyncedWindowDelegate*>
+  GetSyncedWindowDelegates() OVERRIDE {
+    return delegates_;
+  }
+
+  void Add(SyncedWindowDelegate* delegate) {
+    delegates_.insert(delegate);
+  }
+
+ private:
+  std::set<SyncedWindowDelegate*> delegates_;
+};
+
+class SessionDataTypeControllerTest
+    : public testing::Test {
+ public:
+  SessionDataTypeControllerTest()
+      : load_finished_(false),
+        thread_bundle_(content::TestBrowserThreadBundle::DEFAULT),
+        weak_ptr_factory_(this) {}
+  virtual ~SessionDataTypeControllerTest() {}
+
+  virtual void SetUp() OVERRIDE {
+    synced_window_delegate_.reset(new MockSyncedWindowDelegate(&profile_));
+    synced_window_getter_.reset(new MockSyncedWindowDelegatesGetter());
+    synced_window_getter_->Add(synced_window_delegate_.get());
+
+    local_device_.reset(new LocalDeviceInfoProviderMock(
+        "cache_guid",
+        "Wayne Gretzky's Hacking Box",
+        "Chromium 10k",
+        "Chrome 10k",
+        sync_pb::SyncEnums_DeviceType_TYPE_LINUX));
+
+    controller_ = new SessionDataTypeController(
+        &profile_sync_factory_,
+        &profile_,
+        synced_window_getter_.get(),
+        local_device_.get(),
+        DataTypeController::DisableTypeCallback());
+  }
+
+  virtual void TearDown() OVERRIDE {
+    controller_ = NULL;
+    local_device_.reset();
+    synced_window_getter_.reset();
+    synced_window_delegate_.reset();
+    load_finished_ = false;
+  }
+
+  void Start() {
+    controller_->LoadModels(
+      base::Bind(&SessionDataTypeControllerTest::OnLoadFinished,
+                 weak_ptr_factory_.GetWeakPtr()));
+  }
+
+  void OnLoadFinished(syncer::ModelType type, syncer::SyncError error) {
+    EXPECT_FALSE(error.IsSet());
+    EXPECT_EQ(type, syncer::SESSIONS);
+    load_finished_ = true;
+  }
+
+  void ExpectModelLoaded() {
+    EXPECT_TRUE(load_finished_);
+    EXPECT_EQ(DataTypeController::MODEL_LOADED, controller_->state());
+  }
+
+  void ExpectModelStarting() {
+    EXPECT_FALSE(load_finished_);
+    EXPECT_EQ(DataTypeController::MODEL_STARTING, controller_->state());
+  }
+
+ protected:
+  scoped_refptr<SessionDataTypeController> controller_;
+  scoped_ptr<MockSyncedWindowDelegatesGetter> synced_window_getter_;
+  scoped_ptr<LocalDeviceInfoProviderMock> local_device_;
+  scoped_ptr<MockSyncedWindowDelegate> synced_window_delegate_;
+  bool load_finished_;
+
+ private:
+  content::TestBrowserThreadBundle thread_bundle_;
+  ProfileSyncComponentsFactoryMock profile_sync_factory_;
+  TestingProfile profile_;
+  base::WeakPtrFactory<SessionDataTypeControllerTest> weak_ptr_factory_;
+};
+
+TEST_F(SessionDataTypeControllerTest, StartModels) {
+  Start();
+  ExpectModelLoaded();
+}
+
+TEST_F(SessionDataTypeControllerTest, StartModelsDelayedByLocalDevice) {
+  local_device_->SetInitialized(false);
+  Start();
+  ExpectModelStarting();
+
+  local_device_->SetInitialized(true);
+  ExpectModelLoaded();
+}
+
+TEST_F(SessionDataTypeControllerTest, StartModelsDelayedByRestoreInProgress) {
+  synced_window_delegate_->SetSessionRestoreInProgress(true);
+  Start();
+  ExpectModelStarting();
+
+  synced_window_delegate_->SetSessionRestoreInProgress(false);
+  ExpectModelLoaded();
+}
+
+TEST_F(SessionDataTypeControllerTest,
+    StartModelsDelayedByLocalDeviceThenRestoreInProgress) {
+  local_device_->SetInitialized(false);
+  synced_window_delegate_->SetSessionRestoreInProgress(true);
+  Start();
+  ExpectModelStarting();
+
+  local_device_->SetInitialized(true);
+  ExpectModelStarting();
+
+  synced_window_delegate_->SetSessionRestoreInProgress(false);
+  ExpectModelLoaded();
+}
+
+TEST_F(SessionDataTypeControllerTest,
+    StartModelsDelayedByRestoreInProgressThenLocalDevice) {
+  local_device_->SetInitialized(false);
+  synced_window_delegate_->SetSessionRestoreInProgress(true);
+  Start();
+  ExpectModelStarting();
+
+  synced_window_delegate_->SetSessionRestoreInProgress(false);
+  ExpectModelStarting();
+
+  local_device_->SetInitialized(true);
+  ExpectModelLoaded();
+}
+
+} // namespace
+
+} // namespace browser_sync