[Blimp] Refactor Blimp test engine with embedded test server and URL rewriting

blimp/engine/testing/session/test_blimp_engine_session.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/engine/testing/session/test_blimp_engine_session.h"
+
+#include <string>
+
+#include "content/public/browser/browser_url_handler.h"
+
+namespace blimp {
+namespace engine {
+
+TestBlimpEngineSession::TestBlimpEngineSession(
+    std::unique_ptr<BlimpBrowserContext> browser_context,
+    net::NetLog* net_log,
+    BlimpEngineConfig* engine_config,
+    SettingsManager* settings_manager)
+    : BlimpEngineSession(std::move(browser_context),
+                         net_log,
+                         engine_config,
+                         settings_manager){}
+
+void TestBlimpEngineSession::ProcessMessage(
+    std::unique_ptr<BlimpMessage> message,
+    const net::CompletionCallback& callback) {
+  if (message->has_navigation() &&
+      message->navigation().type() == NavigationMessage::LOAD_URL) {
+    GURL url = GURL(message->navigation().load_url().url());
+    bool reverse_on_redirect = false;
+    content::BrowserURLHandler::GetInstance()->RewriteURLIfNecessary(
+        &url, browser_context(), &reverse_on_redirect);
+    message->mutable_navigation()->mutable_load_url()->set_url(url.spec());
+  }
+  BlimpEngineSession::ProcessMessage(std::move(message), callback);
+}
+
+}  // namespace engine
+}  // namespace blimp