[GCM] Stop/restart GCM when the state is forced off/on

chrome/browser/services/gcm/gcm_profile_service_unittest.cc

 // Copyright (c) 2013 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 <algorithm>
 #include <map>
 
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/message_loop/message_loop.h"
@@ skipping 631 matching lines @@
   EXPECT_EQ(GCMClientMock::CHECKED_OUT, consumer()->GetGCMClient()->status());
 
   // Sign-in with a different username.
   consumer()->SignIn(kTestingUsername2);
 
   // GCMClient should be loaded again.
   EXPECT_TRUE(consumer()->IsGCMClientReady());
   EXPECT_EQ(GCMClientMock::LOADED, consumer()->GetGCMClient()->status());
 }
 
+TEST_F(GCMProfileServiceTest, StopAndRestartGCM) {
+  // Positive channel signal is provided in SetUp.
+  consumer()->CreateGCMProfileServiceInstance();
+  consumer()->SignIn(kTestingUsername);
+
+  // GCMClient should be loaded.
+  EXPECT_TRUE(consumer()->IsGCMClientReady());
+  EXPECT_EQ(GCMClientMock::LOADED, consumer()->GetGCMClient()->status());
+
+  // Stops the GCM.
+  consumer()->GetGCMProfileService()->Stop();
+  PumpIOLoop();
+
+  // GCMClient should be stopped.
+  EXPECT_FALSE(consumer()->IsGCMClientReady());
+  EXPECT_EQ(GCMClientMock::STOPPED, consumer()->GetGCMClient()->status());
+
+  // Restarts the GCM.
+  consumer()->GetGCMProfileService()->Start();
+  PumpIOLoop();
+
+  // GCMClient should be loaded.
+  EXPECT_TRUE(consumer()->IsGCMClientReady());
+  EXPECT_EQ(GCMClientMock::LOADED, consumer()->GetGCMClient()->status());
+
+  // Stops the GCM.
+  consumer()->GetGCMProfileService()->Stop();
+  PumpIOLoop();
+
+  // GCMClient should be stopped.
+  EXPECT_FALSE(consumer()->IsGCMClientReady());
+  EXPECT_EQ(GCMClientMock::STOPPED, consumer()->GetGCMClient()->status());
+
+  // Signs out.
+  consumer()->SignOut();
+
+  // GCMClient should be checked out.
+  EXPECT_FALSE(consumer()->IsGCMClientReady());
+  EXPECT_EQ(GCMClientMock::CHECKED_OUT, consumer()->GetGCMClient()->status());
+}
+
 TEST_F(GCMProfileServiceTest, RegisterWhenNotSignedIn) {
   consumer()->CreateGCMProfileServiceInstance();
 
   std::vector<std::string> sender_ids;
   sender_ids.push_back("sender1");
   consumer()->Register(kTestingAppId, sender_ids);
 
   EXPECT_TRUE(consumer()->registration_id().empty());
   EXPECT_EQ(GCMClient::NOT_SIGNED_IN, consumer()->registration_result());
 }
@@ skipping 717 matching lines @@
   consumer()->gcm_event_router()->clear_results();
   WaitUntilCompleted();
 
   EXPECT_EQ(FakeGCMEventRouter::MESSAGE_EVENT,
             consumer()->gcm_event_router()->received_event());
   EXPECT_TRUE(
       in_message5.data == consumer()->gcm_event_router()->message().data);
 }
 
 }  // namespace gcm