Retry: sync: Gracefully handle early shutdown

sync/internal_api/syncapi_server_connection_manager_unittest.cc

 // Copyright (c) 2012 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 "sync/internal_api/syncapi_server_connection_manager.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/compiler_specific.h"
 #include "base/synchronization/waitable_event.h"
 #include "base/test/test_timeouts.h"
 #include "base/threading/thread.h"
 #include "base/time/time.h"
 #include "net/base/net_errors.h"
+#include "sync/internal_api/public/base/cancelation_signal.h"
 #include "sync/internal_api/public/http_post_provider_factory.h"
 #include "sync/internal_api/public/http_post_provider_interface.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace syncer {
 namespace {
 
 using base::TimeDelta;
 
 class BlockingHttpPost : public HttpPostProviderInterface {
@@ skipping 25 matching lines @@
   virtual void Abort() OVERRIDE {
     wait_for_abort_.Signal();
   }
  private:
   base::WaitableEvent wait_for_abort_;
 };
 
 class BlockingHttpPostFactory : public HttpPostProviderFactory {
  public:
   virtual ~BlockingHttpPostFactory() {}
+  virtual void Init(const std::string& user_agent) OVERRIDE {}
   virtual HttpPostProviderInterface* Create() OVERRIDE {
     return new BlockingHttpPost();
   }
   virtual void Destroy(HttpPostProviderInterface* http) OVERRIDE {
     delete static_cast<BlockingHttpPost*>(http);
   }
-  virtual void Shutdown() OVERRIDE {}
 };
 
 }  // namespace
 
-TEST(SyncAPIServerConnectionManagerTest, EarlyAbortPost) {
+// Ask the ServerConnectionManager to stop before it is created.
+TEST(SyncAPIServerConnectionManagerTest, VeryEarlyAbortPost) {
+  CancelationSignal signal;
+  signal.Signal();
   SyncAPIServerConnectionManager server(
-      "server", 0, true, false, new BlockingHttpPostFactory());
+      "server", 0, true, false, new BlockingHttpPostFactory(), &signal);
 
   ServerConnectionManager::PostBufferParams params;
   ScopedServerStatusWatcher watcher(&server, &params.response);
 
-  server.TerminateAllIO();
   bool result = server.PostBufferToPath(
       &params, "/testpath", "testauth", &watcher);
 
+  EXPECT_FALSE(result);
+  EXPECT_EQ(HttpResponse::CONNECTION_UNAVAILABLE,
+            params.response.server_status);
+}
+
+// Ask the ServerConnectionManager to stop before its first request is made.
+TEST(SyncAPIServerConnectionManagerTest, EarlyAbortPost) {
+  CancelationSignal signal;
+  SyncAPIServerConnectionManager server(
+      "server", 0, true, false, new BlockingHttpPostFactory(), &signal);
+
+  ServerConnectionManager::PostBufferParams params;
+  ScopedServerStatusWatcher watcher(&server, &params.response);
+
+  signal.Signal();
+  bool result = server.PostBufferToPath(
+      &params, "/testpath", "testauth", &watcher);
+
   EXPECT_FALSE(result);
   EXPECT_EQ(HttpResponse::CONNECTION_UNAVAILABLE,
             params.response.server_status);
 }
 
+// Ask the ServerConnectionManager to stop during a request.
 TEST(SyncAPIServerConnectionManagerTest, AbortPost) {
+  CancelationSignal signal;
   SyncAPIServerConnectionManager server(
-      "server", 0, true, false, new BlockingHttpPostFactory());
+      "server", 0, true, false, new BlockingHttpPostFactory(), &signal);
 
   ServerConnectionManager::PostBufferParams params;
   ScopedServerStatusWatcher watcher(&server, &params.response);
 
   base::Thread abort_thread("Test_AbortThread");
   ASSERT_TRUE(abort_thread.Start());
   abort_thread.message_loop()->PostDelayedTask(
       FROM_HERE,
-      base::Bind(&ServerConnectionManager::TerminateAllIO,
-                 base::Unretained(&server)),
+      base::Bind(&CancelationSignal::Signal,
+                 base::Unretained(&signal)),
       TestTimeouts::tiny_timeout());
 
   bool result = server.PostBufferToPath(
       &params, "/testpath", "testauth", &watcher);
 
   EXPECT_FALSE(result);
   EXPECT_EQ(HttpResponse::CONNECTION_UNAVAILABLE,
             params.response.server_status);
   abort_thread.Stop();
 }
 
 }  // namespace syncer