Add incognito mode to tests for extension webstore priviate APIs....

chrome/browser/extensions/extension_webstore_private_browsertest.cc

 // Copyright (c) 2010 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 "base/string_number_conversions.h"
 #include "base/utf_string_conversions.h"
+#include "chrome/browser/browser_process.h"
 #include "chrome/browser/browser_signin.h"
 #include "chrome/browser/extensions/extension_browsertest.h"
 #include "chrome/browser/extensions/extension_test_message_listener.h"
 #include "chrome/browser/extensions/extension_webstore_private_api.h"
 #include "chrome/browser/net/gaia/token_service.h"
+#include "chrome/browser/profile_manager.h"
 #include "chrome/browser/sync/profile_sync_service.h"
 #include "chrome/browser/tab_contents/tab_contents.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/net/gaia/gaia_constants.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/test/ui_test_utils.h"
 #include "googleurl/src/gurl.h"
 #include "net/base/mock_host_resolver.h"
 
 using chrome::kHttpScheme;
@@ skipping 53 matching lines @@
   virtual void RequestSignin(TabContents* tab_contents,
                              const string16& preferred_email,
                              const string16& message,
                              SigninDelegate* delegate) {
     if (should_succeed_) {
       // Simulate valid login.
       username_ = username_after_login_;
       delegate->OnLoginSuccess();
 
       // Fake a token available notification.
-      TokenService* token_service = tab_contents->profile()->GetTokenService();
+      Profile* profile = tab_contents->profile();
+      if (profile->IsOffTheRecord()) {
+        profile = g_browser_process->profile_manager()->GetDefaultProfile();
+      }
+      TokenService* token_service = profile->GetTokenService();
       token_service->IssueAuthTokenForTest(GaiaConstants::kGaiaService,
                                            "new_token");
     } else {
       delegate->OnLoginFailure(GoogleServiceAuthError(
         GoogleServiceAuthError::REQUEST_CANCELED));
     }
   }
 
  private:
   bool should_succeed_;
@@ skipping 19 matching lines @@
   // command line switch we added in SetupCommandLine).
   GURL GetUrl(const std::string& relative_path) {
     GURL base_url = test_server()->GetURL(
         "files/extensions/webstore_private/" + relative_path);
     GURL::Replacements replacements;
     std::string replacement_host = std::string(kTestUrlHostname);
     replacements.SetHostStr(replacement_host);
     return base_url.ReplaceComponents(replacements);
   }
 
-  void RunLoginTest(const std::string& relative_path,
-                    const std::string& initial_login,
-                    bool login_succeeds,
-                    const std::string& login_result) {
+  void RunLoginTestImpl(bool incognito,
+                        const std::string& relative_path,
+                        const std::string& initial_login,
+                        bool login_succeeds,
+                        const std::string& login_result) {
     // Clear the token service so previous tests don't affect things.
     TokenService* token_service = browser()->profile()->GetTokenService();
     token_service->ResetCredentialsInMemory();
     if (!initial_login.empty()) {
       // Initialize the token service with an existing token.
       token_service->IssueAuthTokenForTest(GaiaConstants::kGaiaService,
                                            "existing_token");
     }
 
     FakeProfileSyncService sync_service;
     FakeBrowserSignin signin(login_succeeds, initial_login, login_result);
     WebstorePrivateApi::SetTestingProfileSyncService(&sync_service);
     WebstorePrivateApi::SetTestingBrowserSignin(&signin);
 
     ExtensionTestMessageListener listener("success", false);
     GURL url = GetUrl(relative_path);
-    ui_test_utils::NavigateToURL(browser(), url);
+    if (incognito) {
+      ui_test_utils::OpenURLOffTheRecord(browser()->profile(), url);
+    } else {
+      ui_test_utils::NavigateToURL(browser(), url);
+    }
     EXPECT_TRUE(listener.WaitUntilSatisfied());
 
     WebstorePrivateApi::SetTestingBrowserSignin(NULL);
     WebstorePrivateApi::SetTestingProfileSyncService(NULL);
   }
 
+  void RunLoginTest(const std::string& relative_path,
+                    const std::string& initial_login,
+                    bool login_succeeds,
+                    const std::string& login_result) {
+    RunLoginTestImpl(true,
+                     relative_path,
+                     initial_login,
+                     login_succeeds,
+                     login_result);
+    RunLoginTestImpl(false,
+                     relative_path,
+                     initial_login,
+                     login_succeeds,
+                     login_result);
+  }
+
  protected:
   std::string test_url_base_;
 };
 
 IN_PROC_BROWSER_TEST_F(ExtensionWebstorePrivateBrowserTest, BrowserLogin) {
   host_resolver()->AddRule(kTestUrlHostname, "127.0.0.1");
   ASSERT_TRUE(test_server()->Start());
 
   RunLoginTest("browser_login/expect_nonempty.html",
                "foo@bar.com", false, "");
 
   RunLoginTest("browser_login/prompt_no_preferred.html", "", true, "");
 
   RunLoginTest("browser_login/prompt_preferred.html", "", true, "foo@bar.com");
 
   RunLoginTest("browser_login/prompt_already_logged_in_error.html",
                "foo@bar.com", false, "foo@bar.com");
 
   RunLoginTest("browser_login/prompt_login_fails.html",
                "", false, "foo@bar.com");
 }