Allows client to access auth token from command line specified file.

blimp/test/browser_tests/blimp_browser_test.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/test/browser_tests/blimp_browser_test.h"
 
 #include <string>
 
 #include "base/base_switches.h"
 #include "base/command_line.h"
 #include "base/files/file_path.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/path_service.h"
 #include "base/run_loop.h"
 #include "blimp/client/app/blimp_client_switches.h"
 #include "blimp/client/session/assignment_source.h"
+#include "blimp/common/get_client_token.h"
+#include "blimp/common/switches.h"
 #include "blimp/engine/app/blimp_browser_main_parts.h"
 #include "blimp/engine/app/blimp_content_browser_client.h"
 #include "blimp/engine/app/blimp_engine_config.h"
 #include "blimp/engine/app/switches.h"
 #include "blimp/engine/app/test_content_main_delegate.h"
 #include "blimp/engine/session/blimp_engine_session.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/common/url_constants.h"
 
@@ skipping 25 matching lines @@
 void BlimpBrowserTest::OnGetEnginePort(uint16_t port) {
   DCHECK_GT(port, 0);
 
   // A test client is started after BlimpBrowserTest::SetUpOnMainThread().
   engine_port_ = port;
   QuitRunLoop();
 }
 
 client::Assignment BlimpBrowserTest::GetAssignment() {
   client::Assignment assignment;
-  assignment.client_token = client::kDummyClientToken;
+  assignment.client_token = GetClientToken(
+      *base::CommandLine::ForCurrentProcess());
   assignment.engine_endpoint =
       net::IPEndPoint(net::IPAddress::IPv4Localhost(), engine_port_);
   assignment.transport_protocol = client::Assignment::TransportProtocol::TCP;
   return assignment;
 }
 
 void BlimpBrowserTest::SetUpOnMainThread() {
   GetEngineSession()->GetEnginePortForTesting(
       base::Bind(&BlimpBrowserTest::OnGetEnginePort, base::Unretained(this)));
   RunUntilQuit();
 }
 
 void BlimpBrowserTest::TearDownOnMainThread() {
   content::BrowserThread::GetMessageLoopProxyForThread(
       content::BrowserThread::UI)
       ->PostTask(FROM_HERE, base::MessageLoop::QuitWhenIdleClosure());
 }
 
 void BlimpBrowserTest::SetUpCommandLine(base::CommandLine* command_line) {
   // Engine switches.
   blimp::engine::SetCommandLineDefaults(command_line);
 
   // Use dynamically allocated port for browser test.
   command_line->AppendSwitchASCII(blimp::engine::kEnginePort, "0");
   base::FilePath src_root;
   PathService::Get(base::DIR_SOURCE_ROOT, &src_root);
-  command_line->AppendSwitchASCII(
-      blimp::engine::kClientTokenPath,
+  command_line->AppendSwitchASCII(kClientTokenPath,
       src_root.Append(kClientTokenFilePath).value());
 }
 
 void BlimpBrowserTest::RunTestOnMainThreadLoop() {
   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
 
   // Pump startup related events.
   base::MessageLoop::current()->RunUntilIdle();
 
   SetUpOnMainThread();
@@ skipping 14 matching lines @@
   run_loop_ = base::WrapUnique(new base::RunLoop());
   run_loop_->Run();
   run_loop_ = nullptr;
 }
 
 void BlimpBrowserTest::QuitRunLoop() {
   run_loop_->Quit();
 }
 
 }  // namespace blimp