[Sync] Add tests for migration triggered by notifications

chrome/test/live_sync/live_sync_test.cc

 // Copyright (c) 2011 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 "chrome/test/live_sync/live_sync_test.h"
 
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/command_line.h"
 #include "base/message_loop.h"
 #include "base/path_service.h"
 #include "base/stringprintf.h"
 #include "base/string_util.h"
 #include "base/synchronization/waitable_event.h"
 #include "base/task.h"
 #include "base/test/test_timeouts.h"
 #include "base/threading/platform_thread.h"
 #include "base/utf_string_conversions.h"
 #include "base/values.h"
 #include "chrome/browser/password_manager/encryptor.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/profiles/profile_manager.h"
+#include "chrome/browser/sync/notifier/p2p_notifier.h"
 #include "chrome/browser/sync/profile_sync_service_harness.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_list.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/base/testing_browser_process.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "chrome/test/live_sync/sync_datatype_helper.h"
 #include "content/browser/browser_thread.h"
 #include "content/browser/tab_contents/tab_contents.h"
@@ skipping 72 matching lines @@
   net::URLRequestContextGetter* url_request_context_getter_;
   net::ProxyConfig proxy_config_;
 };
 
 LiveSyncTest::LiveSyncTest(TestType test_type)
     : sync_server_(net::TestServer::TYPE_SYNC, FilePath()),
       test_type_(test_type),
       server_type_(SERVER_TYPE_UNDECIDED),
       num_clients_(-1),
       use_verifier_(true),
-      test_server_handle_(base::kNullProcessHandle),
-      enable_notifications_(true) {
+      test_server_handle_(base::kNullProcessHandle) {
   InProcessBrowserTest::set_show_window(true);
   sync_datatype_helper::AssociateWithTest(this);
   switch (test_type_) {
     case SINGLE_CLIENT: {
       num_clients_ = 1;
       break;
     }
     case TWO_CLIENT: {
       num_clients_ = 2;
       break;
@@ skipping 152 matching lines @@
   // Create the required number of sync profiles, browsers and clients.
   for (int i = 0; i < num_clients_; ++i) {
     profiles_.push_back(MakeProfile(
         base::StringPrintf(FILE_PATH_LITERAL("Profile%d"), i)));
     EXPECT_FALSE(GetProfile(i) == NULL) << "GetProfile(" << i << ") failed.";
 
     browsers_.push_back(Browser::Create(GetProfile(i)));
     EXPECT_FALSE(GetBrowser(i) == NULL) << "GetBrowser(" << i << ") failed.";
 
     clients_.push_back(
-        new ProfileSyncServiceHarness(GetProfile(i), username_, password_,
-                                      enable_notifications_));
+        new ProfileSyncServiceHarness(GetProfile(i), username_, password_));
     EXPECT_FALSE(GetClient(i) == NULL) << "GetClient(" << i << ") failed.";
 
     ui_test_utils::WaitForBookmarkModelToLoad(
         GetProfile(i)->GetBookmarkModel());
   }
 
   // Create the verifier profile.
   verifier_.reset(MakeProfile(FILE_PATH_LITERAL("Verifier")));
   ui_test_utils::WaitForBookmarkModelToLoad(verifier()->GetBookmarkModel());
   return (verifier_.get() != NULL);
@@ skipping 98 matching lines @@
 
 bool LiveSyncTest::SetUpLocalPythonTestServer() {
   EXPECT_TRUE(sync_server_.Start())
       << "Could not launch local python test server.";
 
   CommandLine* cl = CommandLine::ForCurrentProcess();
   std::string sync_service_url = sync_server_.GetURL("chromiumsync").spec();
   cl->AppendSwitchASCII(switches::kSyncServiceURL, sync_service_url);
   VLOG(1) << "Started local python test server at " << sync_service_url;
 
-  if (enable_notifications_) {
-    int xmpp_port = 0;
-    if (!sync_server_.server_data().GetInteger("xmpp_port", &xmpp_port)) {
-      LOG(ERROR) << "Could not find valid xmpp_port value";
-      return false;
-    }
-    if ((xmpp_port <= 0) || (xmpp_port > kuint16max)) {
-      LOG(ERROR) << "Invalid xmpp port: " << xmpp_port;
-      return false;
-    }
+  int xmpp_port = 0;
+  if (!sync_server_.server_data().GetInteger("xmpp_port", &xmpp_port)) {
+    LOG(ERROR) << "Could not find valid xmpp_port value";
+    return false;
+  }
+  if ((xmpp_port <= 0) || (xmpp_port > kuint16max)) {
+    LOG(ERROR) << "Invalid xmpp port: " << xmpp_port;
+    return false;
+  }
 
-    net::HostPortPair xmpp_host_port_pair(sync_server_.host_port_pair());
-    xmpp_host_port_pair.set_port(xmpp_port);
-    xmpp_port_.reset(new net::ScopedPortException(xmpp_port));
+  net::HostPortPair xmpp_host_port_pair(sync_server_.host_port_pair());
+  xmpp_host_port_pair.set_port(xmpp_port);
+  xmpp_port_.reset(new net::ScopedPortException(xmpp_port));
 
-    if (!cl->HasSwitch(switches::kSyncNotificationHost)) {
-      cl->AppendSwitchASCII(switches::kSyncNotificationHost,
-                            xmpp_host_port_pair.ToString());
-      // The local XMPP server only supports insecure connections.
-      cl->AppendSwitch(switches::kSyncAllowInsecureXmppConnection);
-    }
+  if (!cl->HasSwitch(switches::kSyncNotificationHost)) {
+    cl->AppendSwitchASCII(switches::kSyncNotificationHost,
+                          xmpp_host_port_pair.ToString());
+    // The local XMPP server only supports insecure connections.
+    cl->AppendSwitch(switches::kSyncAllowInsecureXmppConnection);
   }
 
   return true;
 }
 
 bool LiveSyncTest::SetUpLocalTestServer() {
   CommandLine* cl = CommandLine::ForCurrentProcess();
   CommandLine::StringType server_cmdline_string = cl->GetSwitchValueNative(
       switches::kSyncServerCommandLine);
   CommandLine::StringVector server_cmdline_vector;
@@ skipping 78 matching lines @@
 }
 
 bool LiveSyncTest::EnableEncryption(int index, syncable::ModelType type) {
   return GetClient(index)->EnableEncryptionForType(type);
 }
 
 bool LiveSyncTest::IsEncrypted(int index, syncable::ModelType type) {
   return GetClient(index)->IsTypeEncrypted(type);
 }
 
-void LiveSyncTest::DisableNotifications() {
-  // TODO(akalin): It would be better to assert that the test server
-  // hasn't been started yet.  That would require adding an
-  // IsStarted() method to TestServer.
-  ASSERT_TRUE(profiles_.empty());
-  ASSERT_TRUE(clients_.empty());
-  enable_notifications_ = false;
-}
-
 bool LiveSyncTest::AwaitQuiescence() {
   return ProfileSyncServiceHarness::AwaitQuiescence(clients());
 }
 
-bool LiveSyncTest::ServerSupportsErrorTriggering() {
-  EXPECT_TRUE(server_type_ != SERVER_TYPE_UNDECIDED);
+bool LiveSyncTest::ServerSupportsNotificationControl() {
+  EXPECT_NE(SERVER_TYPE_UNDECIDED, server_type_);
 
   // Supported only if we're using the python testserver.
   return server_type_ == LOCAL_PYTHON_SERVER;
+}
+
+void LiveSyncTest::DisableNotifications() {
+  ASSERT_TRUE(ServerSupportsNotificationControl());
+  std::string path = "chromiumsync/disablenotifications";
+  ui_test_utils::NavigateToURL(browser(), sync_server_.GetURL(path));
+  ASSERT_EQ("Notifications disabled",
+            UTF16ToASCII(browser()->GetSelectedTabContents()->GetTitle()));
+}
+
+void LiveSyncTest::TriggerNotification(
+    const syncable::ModelTypeSet& changed_types) {
+  ASSERT_TRUE(ServerSupportsNotificationControl());
+  const std::string& data =
+      sync_notifier::P2PNotificationData("from_server",
+                                         sync_notifier::NOTIFY_ALL,
+                                         changed_types).ToString();
+  const std::string& path =
+      std::string("chromiumsync/sendnotification?channel=") +
+      sync_notifier::kSyncP2PNotificationChannel + "&data=" + data;
+  ui_test_utils::NavigateToURL(browser(), sync_server_.GetURL(path));
+  ASSERT_EQ("Notification sent",
+            UTF16ToASCII(browser()->GetSelectedTabContents()->GetTitle()));
+}
+
+bool LiveSyncTest::ServerSupportsErrorTriggering() {
+  EXPECT_NE(SERVER_TYPE_UNDECIDED, server_type_);
+
+  // Supported only if we're using the python testserver.
+  return server_type_ == LOCAL_PYTHON_SERVER;
 }
 
 void LiveSyncTest::TriggerMigrationDoneError(
     const syncable::ModelTypeSet& model_types) {
   ASSERT_TRUE(ServerSupportsErrorTriggering());
   std::string path = "chromiumsync/migrate";
   char joiner = '?';
   for (syncable::ModelTypeSet::const_iterator it = model_types.begin();
        it != model_types.end(); ++it) {
     path.append(base::StringPrintf("%ctype=%d", joiner,
@@ skipping 33 matching lines @@
                                   const net::ProxyConfig& proxy_config) {
   base::WaitableEvent done(false, false);
   BrowserThread::PostTask(
       BrowserThread::IO,
       FROM_HERE,
       new SetProxyConfigTask(&done,
                              context_getter,
                              proxy_config));
   done.Wait();
 }