Save the oauth client id used in App Notification setup

chrome/browser/extensions/app_notify_channel_setup.cc

 // Copyright (c) 2011 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/extensions/app_notify_channel_setup.h"
 
 #include "base/basictypes.h"
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/json/json_reader.h"
@@ skipping 13 matching lines @@
 #include "net/base/escape.h"
 #include "net/base/load_flags.h"
 #include "net/http/http_request_headers.h"
 #include "net/url_request/url_request_status.h"
 
 using base::StringPrintf;
 using content::BrowserThread;
 using content::URLFetcher;
 
 namespace {
+
 static const char kChannelSetupAuthError[] = "unauthorized";
 static const char kChannelSetupInternalError[] = "internal_error";
 static const char kChannelSetupCanceledByUser[] = "canceled_by_user";
 // TODO(munjal): Change these to use production URLs when we have them.
 static const char kAuthorizationHeaderFormat[] =
     "Authorization: GoogleLogin auth=%s";
 static const char kOAuth2IssueTokenURL[] =
     "https://www.googleapis.com/oauth2/v2/IssueToken";
 static const char kOAuth2IssueTokenBodyFormat[] =
     "force=true"
     "&response_type=token"
     "&scope=%s"
     "&client_id=%s"
     "&origin=%s";
 static const char kOAuth2IssueTokenScope[] =
     "https://www.googleapis.com/auth/chromewebstore.notification";
 static const char kCWSChannelServiceURL[] =
     "https://www.googleapis.com/chromewebstore/"
     "v1.1/channels/id";
 static const char* kRelevantGaiaServices[] = {
   GaiaConstants::kCWSService,
   GaiaConstants::kLSOService,
 };
 static const int kRelevantGaiaServicesCount = arraysize(kRelevantGaiaServices);
+
+static AppNotifyChannelSetup::InterceptorForTests* g_interceptor_for_tests =
+    NULL;
+
 }  // namespace.
 
+// static
+void AppNotifyChannelSetup::SetInterceptorForTests(
+    AppNotifyChannelSetup::InterceptorForTests* interceptor) {
+  // Only one interceptor at a time, please.
+  CHECK(g_interceptor_for_tests == NULL);
+  g_interceptor_for_tests = interceptor;
+}
+
 AppNotifyChannelSetup::AppNotifyChannelSetup(
     Profile* profile,
     const std::string& extension_id,
     const std::string& client_id,
     const GURL& requestor_url,
     int return_route_id,
     int callback_id,
     AppNotifyChannelUI* ui,
     base::WeakPtr<AppNotifyChannelSetup::Delegate> delegate)
     : profile_(profile),
       extension_id_(extension_id),
       client_id_(client_id),
       requestor_url_(requestor_url),
       return_route_id_(return_route_id),
       callback_id_(callback_id),
       delegate_(delegate),
       ui_(ui),
       state_(INITIAL),
       fetch_token_success_count_(0),
       fetch_token_fail_count_(0) {}
 
 AppNotifyChannelSetup::~AppNotifyChannelSetup() {}
 
 void AppNotifyChannelSetup::Start() {
+  if (g_interceptor_for_tests) {
+    std::string channel_id;
+    std::string error;
+    g_interceptor_for_tests->DoIntercept(this, &channel_id, &error);
+    delegate_->AppNotifyChannelSetupComplete(channel_id, error, this);
+    return;
+  }
   AddRef();  // Balanced in ReportResult.
   BeginLogin();
 }
 
 void AppNotifyChannelSetup::Observe(
     int type,
     const content::NotificationSource& source,
     const content::NotificationDetails& details) {
   if (type == chrome::NOTIFICATION_TOKEN_AVAILABLE) {
     TokenService::TokenAvailableDetails* tok_details =
@@ skipping 216 matching lines @@
   }
 }
 
 void AppNotifyChannelSetup::ReportResult(
     const std::string& channel_id,
     const std::string& error) {
   CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   CHECK(state_ == CHANNEL_ID_SETUP_DONE || state_ == ERROR_STATE);
 
   if (delegate_.get()) {
-    delegate_->AppNotifyChannelSetupComplete(
-        channel_id, error, return_route_id_, callback_id_);
+    delegate_->AppNotifyChannelSetupComplete(channel_id, error, this);
   }
   Release(); // Matches AddRef in Start.
 }
 
 // static
 GURL AppNotifyChannelSetup::GetCWSChannelServiceURL() {
   CommandLine* command_line = CommandLine::ForCurrentProcess();
   if (command_line->HasSwitch(switches::kAppNotifyChannelServerURL)) {
     std::string switch_value = command_line->GetSwitchValueASCII(
         switches::kAppNotifyChannelServerURL);
@@ skipping 51 matching lines @@
   DictionaryValue* dict = static_cast<DictionaryValue*>(value.get());
   if (!dict->Get("id", &channel_id_value))
     return false;
   if (channel_id_value->GetType() != base::Value::TYPE_STRING)
     return false;
 
   StringValue* channel_id = static_cast<StringValue*>(channel_id_value);
   channel_id->GetAsString(result);
   return true;
 }