Add a full Blimp integration test.

blimp/client/core/blimp_client_context_impl.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 "blimp/client/core/blimp_client_context_impl.h"
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/command_line.h"
@@ skipping 114 matching lines @@
     delegate_->AttachBlimpContentsHelpers(blimp_contents.get());
   return blimp_contents;
 }
 
 void BlimpClientContextImpl::Connect() {
   // Start Blimp authentication flow. The OAuth2 token will be used in
   // assignment source.
   GetIdentitySource()->Connect();
 }
 
-void BlimpClientContextImpl::ConnectToAssignmentSource(
+void BlimpClientContextImpl::ConnectWithAssignment(
+    const Assignment& assignment) {
+  io_thread_task_runner_->PostTask(
+      FROM_HERE,
+      base::Bind(&ClientNetworkComponents::ConnectWithAssignment,
+                 base::Unretained(net_components_.get()), assignment));
+}
+
+void BlimpClientContextImpl::OnAuthTokenReceived(
     const std::string& client_auth_token) {
   if (!assignment_source_) {
     assignment_source_.reset(new AssignmentSource(
         GetAssignerURL(), io_thread_task_runner_, file_thread_task_runner_));
   }
 
   VLOG(1) << "Trying to get assignment.";
   assignment_source_->GetAssignment(
       client_auth_token,
-      base::Bind(&BlimpClientContextImpl::ConnectWithAssignment,
+      base::Bind(&BlimpClientContextImpl::OnAssignmentReceived,
                  weak_factory_.GetWeakPtr()));
 }
 
 void BlimpClientContextImpl::OnConnected() {
   UMA_HISTOGRAM_BOOLEAN("Blimp.Connected", true);
 }
 
 void BlimpClientContextImpl::OnDisconnected(int result) {
   UMA_HISTOGRAM_BOOLEAN("Blimp.Connected", false);
 }
 
 GURL BlimpClientContextImpl::GetAssignerURL() {
   return GURL(kDefaultAssignerUrl);
 }
 
 IdentitySource* BlimpClientContextImpl::GetIdentitySource() {
   if (!identity_source_) {
     CreateIdentitySource();
   }
   return identity_source_.get();
 }
 
-void BlimpClientContextImpl::ConnectWithAssignment(
+void BlimpClientContextImpl::OnAssignmentReceived(
     AssignmentRequestResult result,
     const Assignment& assignment) {
   VLOG(1) << "Assignment result: " << result;
 
   if (delegate_) {
     delegate_->OnAssignmentConnectionAttempted(result, assignment);
   }
 
   if (result != ASSIGNMENT_REQUEST_RESULT_OK) {
     LOG(ERROR) << "Assignment failed, reason: " << result;
     return;
   }
 
-  io_thread_task_runner_->PostTask(
-      FROM_HERE,
-      base::Bind(&ClientNetworkComponents::ConnectWithAssignment,
-                 base::Unretained(net_components_.get()), assignment));
+  ConnectWithAssignment(assignment);
 }
 
 void BlimpClientContextImpl::RegisterFeatures() {
   // Register features' message senders and receivers.
   thread_pipe_manager_->RegisterFeature(BlimpMessage::kBlobChannel,
                                         blob_channel_feature_.get());
   geolocation_feature_->set_outgoing_message_processor(
       thread_pipe_manager_->RegisterFeature(BlimpMessage::kGeolocation,
                                             geolocation_feature_.get()));
   ime_feature_->set_outgoing_message_processor(
@@ skipping 19 matching lines @@
 }
 
 void BlimpClientContextImpl::InitializeSettings() {
   if (base::CommandLine::ForCurrentProcess()->HasSwitch(
           switches::kDownloadWholeDocument))
     settings_feature_->SetRecordWholeDocument(true);
 }
 
 void BlimpClientContextImpl::CreateIdentitySource() {
   identity_source_ = base::MakeUnique<IdentitySource>(
-      delegate_, base::Bind(&BlimpClientContextImpl::ConnectToAssignmentSource,
+      delegate_, base::Bind(&BlimpClientContextImpl::OnAuthTokenReceived,
                             base::Unretained(this)));
 }
 
 void BlimpClientContextImpl::OnImageDecodeError() {
   // Currently we just drop the connection on image decoding error.
   io_thread_task_runner_->PostTask(
       FROM_HERE,
       base::Bind(
           &BrowserConnectionHandler::DropCurrentConnection,
           base::Unretained(net_components_->GetBrowserConnectionHandler())));
 }
 
 }  // namespace client
 }  // namespace blimp