[NTP::Push] Add Content Suggestions GCM App Handler

components/ntp_snippets/breaking_news/subscription_manager_unittest.cc

 // Copyright 2017 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 "components/ntp_snippets/breaking_news/subscription_manager.h"
 
 #include "base/message_loop/message_loop.h"
 #include "components/ntp_snippets/pref_names.h"
 #include "components/prefs/testing_pref_service.h"
 #include "net/url_request/test_url_fetcher_factory.h"
@@ skipping 36 matching lines @@
   }
 
   void RespondWithError(int error_code) {
     net::TestURLFetcher* url_fetcher = GetRunningFetcher();
     url_fetcher->set_status(net::URLRequestStatus::FromError(error_code));
     url_fetcher->SetResponseString(std::string());
     // Call the URLFetcher delegate to continue the test.
     url_fetcher->delegate()->OnURLFetchComplete(url_fetcher);
   }
 
+  const std::string url{"http://valid-url.test"};
+
  private:
   base::MessageLoop message_loop_;
   scoped_refptr<net::TestURLRequestContextGetter> request_context_getter_;
   net::TestURLFetcherFactory url_fetcher_factory_;
   std::unique_ptr<TestingPrefServiceSimple> pref_service_;
 };
 
 TEST_F(SubscriptionManagerTest, SubscribeSuccessfully) {
   std::string token = "1234567890";
-  SubscriptionManager manager(GetRequestContext(), GetPrefService(),
-                              GURL("http://valid-url.test"));
+  SubscriptionManager manager(GetRequestContext(), GetPrefService(), GURL(url),
+                              GURL(url));
   manager.Subscribe(token);
   RespondWithData("");
+  EXPECT_TRUE(manager.IsSubscribed());
   EXPECT_EQ(GetPrefService()->GetString(
                 ntp_snippets::prefs::kContentSuggestionsSubscriptionDataToken),
             token);
 }
 
 TEST_F(SubscriptionManagerTest, SubscribeWithErrors) {
   std::string token = "1234567890";
-  SubscriptionManager manager(GetRequestContext(), GetPrefService(),
-                              GURL("http://valid-url.test"));
+  SubscriptionManager manager(GetRequestContext(), GetPrefService(), GURL(url),
+                              GURL(url));
   manager.Subscribe(token);
   RespondWithError(net::ERR_TIMED_OUT);
+  EXPECT_FALSE(manager.IsSubscribed());
   EXPECT_FALSE(GetPrefService()->HasPrefPath(
       ntp_snippets::prefs::kContentSuggestionsSubscriptionDataToken));
 }
 
+TEST_F(SubscriptionManagerTest, UnsubscribeSuccessfully) {
+  std::string token = "1234567890";
+  GetPrefService()->SetString(
+      ntp_snippets::prefs::kContentSuggestionsSubscriptionDataToken, token);
+  SubscriptionManager manager(GetRequestContext(), GetPrefService(), GURL(url),
+                              GURL(url));
+  manager.Unsubscribe(token);
+  RespondWithData("");
+  EXPECT_FALSE(manager.IsSubscribed());
+  EXPECT_FALSE(GetPrefService()->HasPrefPath(
+      ntp_snippets::prefs::kContentSuggestionsSubscriptionDataToken));
+}
+
+TEST_F(SubscriptionManagerTest, UnsubscribeWithErrors) {
+  std::string token = "1234567890";
+  GetPrefService()->SetString(
+      ntp_snippets::prefs::kContentSuggestionsSubscriptionDataToken, token);
+  SubscriptionManager manager(GetRequestContext(), GetPrefService(), GURL(url),
+                              GURL(url));
+  manager.Unsubscribe(token);
+  RespondWithError(net::ERR_TIMED_OUT);
+  EXPECT_TRUE(manager.IsSubscribed());
+  EXPECT_EQ(GetPrefService()->GetString(
+                ntp_snippets::prefs::kContentSuggestionsSubscriptionDataToken),
+            token);
+}
+
 }  // namespace ntp_snippets