sync: don't display one-click if setup is in progress

chrome/browser/ui/sync/one_click_signin_helper_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 "chrome/browser/content_settings/cookie_settings.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/signin/signin_manager_factory.h"
 #include "chrome/browser/signin/signin_manager_fake.h"
+#include "chrome/browser/sync/profile_sync_service_factory.h"
+#include "chrome/browser/sync/test_profile_sync_service.h"
 #include "chrome/browser/ui/sync/one_click_signin_helper.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/testing_profile.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/test/test_browser_thread.h"
 #include "content/public/test/test_renderer_host.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
@@ skipping 26 matching lines @@
 OneClickSigninHelperTest::OneClickSigninHelperTest()
     : ui_thread_(content::BrowserThread::UI, &message_loop_) {
 }
 
 void OneClickSigninHelperTest::SetUp() {
   // Don't call base class so that default browser context and test WebContents
   // are not created now.  They will be created in CreateMockWebContents()
   // as needed.
 }
 
+class OneClickTestProfileSyncService : public TestProfileSyncService {
+ public:
+  virtual ~OneClickTestProfileSyncService() {}
+
+  // Helper routine to be used in conjunction with
+  // ProfileKeyedServiceFactory::SetTestingFactory().
+  static ProfileKeyedService* Build(Profile* profile) {
+    return new OneClickTestProfileSyncService(profile);
+  }
+
+  // Need to control this for certain tests.
+  virtual bool FirstSetupInProgress() const OVERRIDE {
+    return first_setup_in_progress_;
+  }
+
+  // Controls return value of FirstSetupInProgress. Because some bits
+  // of UI depend on that value, it's useful to control it separately
+  // from the internal work and components that are triggered (such as
+  // ReconfigureDataTypeManager) to facilitate unit tests.
+  void set_first_setup_in_progress(bool in_progress) {
+    first_setup_in_progress_ = in_progress;
+  }
+
+ private:
+  explicit OneClickTestProfileSyncService(Profile* profile)
+      : TestProfileSyncService(NULL,
+                               profile,
+                               NULL,
+                               ProfileSyncService::MANUAL_START,
+                               false,  // synchronous_backend_init
+                               base::Closure()),
+        first_setup_in_progress_(false) {}
+
+  bool first_setup_in_progress_;
+};
+
 content::WebContents* OneClickSigninHelperTest::CreateMockWebContents(
     bool use_incognito,
     const std::string& username) {
   TestingProfile* testing_profile = new TestingProfile();
   browser_context_.reset(testing_profile);
 
   testing_profile->set_incognito(use_incognito);
   SigninManager* signin_manager = static_cast<SigninManager*>(
       SigninManagerFactory::GetInstance()->SetTestingFactoryAndUse(
           testing_profile, FakeSigninManager::Build));
   if (!username.empty()) {
     signin_manager->StartSignIn(username, std::string(), std::string(),
                                 std::string());
   }
-
   return CreateTestWebContents();
 }
 
 void OneClickSigninHelperTest::EnableOneClick(bool enable) {
   PrefService* pref_service = Profile::FromBrowserContext(
       browser_context_.get())->GetPrefs();
   pref_service->SetBoolean(prefs::kReverseAutologinEnabled, enable);
 }
 
 void OneClickSigninHelperTest::AllowSigninCookies(bool enable) {
@@ skipping 16 matching lines @@
 
   EnableOneClick(true);
   EXPECT_TRUE(OneClickSigninHelper::CanOffer(web_contents, true));
   EXPECT_TRUE(OneClickSigninHelper::CanOffer(web_contents, false));
 
   EnableOneClick(false);
   EXPECT_FALSE(OneClickSigninHelper::CanOffer(web_contents, true));
   EXPECT_FALSE(OneClickSigninHelper::CanOffer(web_contents, false));
 }
 
+TEST_F(OneClickSigninHelperTest, CanOfferFirstSetup) {
+  content::WebContents* web_contents = CreateMockWebContents(false, "");
+
+  // Invoke OneClickTestProfileSyncService factory function and grab result.
+  OneClickTestProfileSyncService* sync =
+      static_cast<OneClickTestProfileSyncService*>(
+          ProfileSyncServiceFactory::GetInstance()->SetTestingFactoryAndUse(
+              static_cast<Profile*>(browser_context()),
+              OneClickTestProfileSyncService::Build));
+
+  sync->set_first_setup_in_progress(true);
+
+  EXPECT_FALSE(OneClickSigninHelper::CanOffer(web_contents, true));
+  EXPECT_TRUE(OneClickSigninHelper::CanOffer(web_contents, false));
+}
+
 TEST_F(OneClickSigninHelperTest, CanOfferProfileConnected) {
   content::WebContents* web_contents = CreateMockWebContents(false,
                                                              "foo@gmail.com");
 
   EXPECT_FALSE(OneClickSigninHelper::CanOffer(web_contents, true));
   EXPECT_TRUE(OneClickSigninHelper::CanOffer(web_contents, false));
 }
 
 TEST_F(OneClickSigninHelperTest, CanOfferIncognito) {
   content::WebContents* web_contents = CreateMockWebContents(true, "");
 
   EXPECT_FALSE(OneClickSigninHelper::CanOffer(web_contents, true));
   EXPECT_FALSE(OneClickSigninHelper::CanOffer(web_contents, false));
 }
 
 TEST_F(OneClickSigninHelperTest, CanOfferNoSigninCookies) {
   content::WebContents* web_contents = CreateMockWebContents(false, "");
   AllowSigninCookies(false);
 
   EXPECT_FALSE(OneClickSigninHelper::CanOffer(web_contents, true));
   EXPECT_FALSE(OneClickSigninHelper::CanOffer(web_contents, false));
 }