[Blimp] Adds blimp engine browser test framework and LoadUrl test.

blimp/test/browser_tests/blimp_browser_test.cc

Index: blimp/test/browser_tests/blimp_browser_test.cc
diff --git a/blimp/test/browser_tests/blimp_browser_test.cc b/blimp/test/browser_tests/blimp_browser_test.cc
new file mode 100644
index 0000000000000000000000000000000000000000..7c0ea5295e53c1ee74c7a28a6b9cad8d5017d0e3
--- /dev/null
+++ b/blimp/test/browser_tests/blimp_browser_test.cc
@@ -0,0 +1,121 @@
+// 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/run_loop.h"
+#include "blimp/client/app/blimp_client_switches.h"
+#include "blimp/client/session/assignment_source.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"
+
+namespace blimp {
+namespace {
+const char kTestDataFilePath[] = "blimp/engine/test/data";
+const char kClientTokenFilePath[] = "blimp/test/data/test_client_token";
+}  // namespace
+
+BlimpBrowserTest::BlimpBrowserTest() {
+  CreateTestServer(base::FilePath(FILE_PATH_LITERAL(kTestDataFilePath)));
+}
+
+BlimpBrowserTest::~BlimpBrowserTest() {}
+
+void BlimpBrowserTest::SetUp() {
+  base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
+  SetUpCommandLine(command_line);
+  BrowserTestBase::SetUp();
+}
+
+engine::BlimpEngineSession* BlimpBrowserTest::GetEngineSession() {
+  return engine::TestContentMainDelegate::GetInstance()
+      ->browser_client()
+      ->blimp_browser_main_parts()
+      ->GetBlimpEngineSession();
+}
+
+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.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");
+  command_line->AppendSwitchASCII(blimp::engine::kClientTokenPath,
+                                  kClientTokenFilePath);
+}
+
+void BlimpBrowserTest::RunTestOnMainThreadLoop() {
+  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
+
+  // Pump startup related events.
+  base::MessageLoop::current()->RunUntilIdle();
+
+  SetUpOnMainThread();
+  RunTestOnMainThread();
+  TearDownOnMainThread();
+
+  for (content::RenderProcessHost::iterator i(
+           content::RenderProcessHost::AllHostsIterator());
+       !i.IsAtEnd(); i.Advance()) {
+    i.GetCurrentValue()->FastShutdownIfPossible();
+  }
+}
+
+void BlimpBrowserTest::RunUntilQuit() {
+  base::MessageLoop::ScopedNestableTaskAllower nestable_allower(
+      base::MessageLoop::current());
+  EXPECT_FALSE(run_loop_);
+  run_loop_ = base::WrapUnique(new base::RunLoop());
+  run_loop_->Run();
+  run_loop_ = nullptr;
+}
+
+void BlimpBrowserTest::QuitRunLoop() {
+  run_loop_->Quit();
+}
+
+}  // namespace blimp