Integrate UI with authentication flow.

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 "base/bind.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/threading/sequenced_task_runner_handle.h"
@@ skipping 71 matching lines @@
 std::unique_ptr<BlimpContents> BlimpClientContextImpl::CreateBlimpContents() {
   std::unique_ptr<BlimpContents> blimp_contents =
       blimp_contents_manager_->CreateBlimpContents();
   delegate_->AttachBlimpContentsHelpers(blimp_contents.get());
   return blimp_contents;
 }
 
 void BlimpClientContextImpl::Connect() {
   // Lazy initialization of IdentitySource.
   if (!identity_source_) {
-    identity_source_ = base::MakeUnique<IdentitySource>(
-        delegate_,
-        base::Bind(&BlimpClientContextImpl::ConnectToAssignmentSource,
-                   base::Unretained(this)));
+    CreateIdentitySource();
   }
 
   // Start Blimp authentication flow. The OAuth2 token will be used in
   // assignment source.
   identity_source_->Connect();
 }
 
 void BlimpClientContextImpl::ConnectToAssignmentSource(
     const std::string& client_auth_token) {
   if (!assignment_source_) {
@@ skipping 40 matching lines @@
                  base::Unretained(net_components_.get()), assignment));
 }
 
 void BlimpClientContextImpl::RegisterFeatures() {
   // Register features' message senders and receivers.
   tab_control_feature_->set_outgoing_message_processor(
       thread_pipe_manager_->RegisterFeature(BlimpMessage::kTabControl,
                                             tab_control_feature_.get()));
 }
 
+void BlimpClientContextImpl::CreateIdentitySource() {
+  identity_source_ = base::MakeUnique<IdentitySource>(
+      delegate_, base::Bind(&BlimpClientContextImpl::ConnectToAssignmentSource,
+                            base::Unretained(this)));
+}
+
 }  // namespace client
 }  // namespace blimp