arc: Use incognito profile for OptIn and cookie fetcher

chrome/browser/chromeos/arc/arc_auth_service_unittest.cc

 // Copyright 2016 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 <string>
 #include <vector>
 
 #include "base/bind.h"
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
@@ skipping 53 matching lines @@
     EXPECT_EQ(ArcBridgeService::State::STOPPED,
               ArcBridgeService::Get()->state());
   }
 
   void TearDown() override {}
 
  protected:
   Profile* profile() { return profile_.get(); }
   FakeArcBridgeService* bridge_service() { return bridge_service_.get(); }
   ArcAuthService* auth_service() { return auth_service_.get(); }
-  net::FakeURLFetcherFactory& url_fetcher_factory() {
-    return url_fetcher_factory_;
+
+  void PrepareURLResponse(net::HttpStatusCode code, bool enable_auth_code) {
+    const GURL gaia_gurl = ArcAuthFetcher::CreateURL();
+    url_fetcher_factory_.SetFakeResponse(gaia_gurl, std::string(), code,
+                                         net::URLRequestStatus::SUCCESS);
+    if (enable_auth_code) {
+      std::string cookie = "oauth_code=";
+      cookie += kTestAuthCode;
+      cookie += "; Path=/o/oauth2/programmatic_auth; Secure; HttpOnly";
+      rt_cookie_ = cookie;
+    }
   }
-  void set_cookie(const std::string& cookie) { rt_cookie_ = cookie; }
 
  private:
   scoped_ptr<net::FakeURLFetcher> FakeURLFetcherCreator(
       const GURL& url,
       net::URLFetcherDelegate* delegate,
       const std::string& response_data,
       net::HttpStatusCode response_code,
       net::URLRequestStatus::Status status) {
     scoped_ptr<net::FakeURLFetcher> fetcher(new net::FakeURLFetcher(
         url, delegate, response_data, response_code, status));
@@ skipping 20 matching lines @@
 
 TEST_F(ArcAuthServiceTest, PrefChangeTriggersService) {
   ASSERT_EQ(ArcAuthService::State::DISABLE, auth_service()->state());
 
   PrefService* pref = profile()->GetPrefs();
   DCHECK_EQ(false, pref->GetBoolean(prefs::kArcEnabled));
 
   auth_service()->OnPrimaryUserProfilePrepared(profile());
   ASSERT_EQ(ArcAuthService::State::DISABLE, auth_service()->state());
 
-  // Need to initialize URLFetcher for test framework.
-  const GURL gaia_gurl = ArcAuthFetcher::CreateURL();
-  url_fetcher_factory().SetFakeResponse(gaia_gurl, std::string(), net::HTTP_OK,
-                                        net::URLRequestStatus::SUCCESS);
-
+  PrepareURLResponse(net::HTTP_OK, false);
   pref->SetBoolean(prefs::kArcEnabled, true);
   ASSERT_EQ(ArcAuthService::State::FETCHING_CODE, auth_service()->state());
 
   pref->SetBoolean(prefs::kArcEnabled, false);
   ASSERT_EQ(ArcAuthService::State::DISABLE, auth_service()->state());
 
   // Correctly stop service.
   auth_service()->Shutdown();
 }
 
 TEST_F(ArcAuthServiceTest, BaseWorkflow) {
   ASSERT_EQ(ArcBridgeService::State::STOPPED, bridge_service()->state());
   ASSERT_EQ(ArcAuthService::State::DISABLE, auth_service()->state());
   ASSERT_EQ(std::string(), auth_service()->GetAndResetAuthCode());
 
-  const GURL gaia_gurl = ArcAuthFetcher::CreateURL();
-  url_fetcher_factory().SetFakeResponse(gaia_gurl, std::string(), net::HTTP_OK,
-                                        net::URLRequestStatus::SUCCESS);
-  std::string cookie = "oauth_code=";
-  cookie += kTestAuthCode;
-  cookie += "; Path=/o/oauth2/programmatic_auth; Secure; HttpOnly";
-  set_cookie(cookie);
+  PrepareURLResponse(net::HTTP_OK, true);
   auth_service()->OnPrimaryUserProfilePrepared(profile());
 
   // By default ARC is not enabled.
   ASSERT_EQ(ArcAuthService::State::DISABLE, auth_service()->state());
 
   profile()->GetPrefs()->SetBoolean(prefs::kArcEnabled, true);
 
   // Setting profile and pref initiates a code fetching process.
   ASSERT_EQ(ArcAuthService::State::FETCHING_CODE, auth_service()->state());
 
@@ skipping 17 matching lines @@
   // Setting profile initiates a code fetching process.
   ASSERT_EQ(ArcAuthService::State::FETCHING_CODE, auth_service()->state());
 
   content::BrowserThread::GetBlockingPool()->FlushForTesting();
   base::RunLoop().RunUntilIdle();
 
   // UI is disabled in unit tests and this code is unchanged.
   ASSERT_EQ(ArcAuthService::State::FETCHING_CODE, auth_service()->state());
 
   // Send error response.
-  url_fetcher_factory().SetFakeResponse(gaia_gurl, std::string(),
-                                        net::HTTP_BAD_REQUEST,
-                                        net::URLRequestStatus::SUCCESS);
+  PrepareURLResponse(net::HTTP_BAD_REQUEST, false);
   auth_service()->Shutdown();
   ASSERT_EQ(ArcAuthService::State::DISABLE, auth_service()->state());
   auth_service()->OnPrimaryUserProfilePrepared(profile());
 
   ASSERT_EQ(ArcAuthService::State::FETCHING_CODE, auth_service()->state());
   content::BrowserThread::GetBlockingPool()->FlushForTesting();
   base::RunLoop().RunUntilIdle();
 
   ASSERT_EQ(ArcAuthService::State::NO_CODE, auth_service()->state());
 
   // Correctly stop service.
   auth_service()->Shutdown();
 }
 
+TEST_F(ArcAuthServiceTest, CancelFetchingDisablesArc) {
+  PrepareURLResponse(net::HTTP_OK, false);
+  PrefService* pref = profile()->GetPrefs();
+
+  auth_service()->OnPrimaryUserProfilePrepared(profile());
+  pref->SetBoolean(prefs::kArcEnabled, true);
+  ASSERT_EQ(ArcAuthService::State::FETCHING_CODE, auth_service()->state());
+
+  auth_service()->CancelAuthCode();
+  ASSERT_EQ(ArcAuthService::State::DISABLE, auth_service()->state());
+  ASSERT_EQ(false, pref->GetBoolean(prefs::kArcEnabled));
+
+  // Correctly stop service.
+  auth_service()->Shutdown();
+}
+
+TEST_F(ArcAuthServiceTest, CloseUIKeepsArcEnabled) {
+  PrepareURLResponse(net::HTTP_OK, true);
+  PrefService* pref = profile()->GetPrefs();
+
+  auth_service()->OnPrimaryUserProfilePrepared(profile());
+  pref->SetBoolean(prefs::kArcEnabled, true);
+
+  content::BrowserThread::GetBlockingPool()->FlushForTesting();
+  base::RunLoop().RunUntilIdle();
+
+  ASSERT_EQ(ArcAuthService::State::ENABLE, auth_service()->state());
+
+  auth_service()->CancelAuthCode();
+  ASSERT_EQ(ArcAuthService::State::ENABLE, auth_service()->state());
+  ASSERT_EQ(true, pref->GetBoolean(prefs::kArcEnabled));
+
+  // Correctly stop service.
+  auth_service()->Shutdown();
+}
+
 }  // namespace arc