<webview>: Add additional WebRequest API test coverage.

extensions/browser/guest_view/web_view/web_view_apitest.cc

 // Copyright 2014 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 "extensions/browser/guest_view/web_view/web_view_apitest.h"
 
 #include "base/path_service.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
 #include "content/public/browser/render_process_host.h"
@@ skipping 16 matching lines @@
 #include "net/base/filename_util.h"
 #include "net/test/embedded_test_server/embedded_test_server.h"
 #include "net/test/embedded_test_server/http_request.h"
 #include "net/test/embedded_test_server/http_response.h"
 
 namespace {
 
 const char kEmptyResponsePath[] = "/close-socket";
 const char kRedirectResponsePath[] = "/server-redirect";
 const char kRedirectResponseFullPath[] = "/guest_redirect.html";
+const char kUserAgentRedirectResponsePath[] = "/detect-user-agent";
 const char kTestDataDirectory[] = "testDataDirectory";
 const char kTestServerPort[] = "testServer.port";
 const char kTestWebSocketPort[] = "testWebSocketPort";
 
 class EmptyHttpResponse : public net::test_server::HttpResponse {
  public:
   virtual std::string ToResponseString() const override {
     return std::string();
   }
 };
 
+// Handles |request| by serving a redirect response if the |User-Agent| is
+// foobar.
+static scoped_ptr<net::test_server::HttpResponse> UserAgentResponseHandler(
+    const std::string& path,
+    const GURL& redirect_target,
+    const net::test_server::HttpRequest& request) {
+  if (!StartsWithASCII(path, request.relative_url, true))
+    return scoped_ptr<net::test_server::HttpResponse>();
+
+  std::map<std::string, std::string>::const_iterator it =
+        request.headers.find("User-Agent");
+  EXPECT_TRUE(it != request.headers.end());
+  if (!StartsWithASCII("foobar", it->second, true))
+    return scoped_ptr<net::test_server::HttpResponse>();
+
+  scoped_ptr<net::test_server::BasicHttpResponse> http_response(
+      new net::test_server::BasicHttpResponse);
+  http_response->set_code(net::HTTP_MOVED_PERMANENTLY);
+  http_response->AddCustomHeader("Location", redirect_target.spec());
+  return http_response.PassAs<net::test_server::HttpResponse>();
+}
+
 // Handles |request| by serving a redirect response.
 scoped_ptr<net::test_server::HttpResponse> RedirectResponseHandler(
     const std::string& path,
     const GURL& redirect_target,
     const net::test_server::HttpRequest& request) {
   if (!StartsWithASCII(path, request.relative_url, true))
     return scoped_ptr<net::test_server::HttpResponse>();
 
   scoped_ptr<net::test_server::BasicHttpResponse> http_response(
       new net::test_server::BasicHttpResponse);
@@ skipping 80 matching lines @@
 
   test_config_.SetInteger(kTestServerPort, embedded_test_server()->port());
 
   embedded_test_server()->RegisterRequestHandler(
       base::Bind(&RedirectResponseHandler,
                  kRedirectResponsePath,
                  embedded_test_server()->GetURL(kRedirectResponseFullPath)));
 
   embedded_test_server()->RegisterRequestHandler(
       base::Bind(&EmptyResponseHandler, kEmptyResponsePath));
+
+  embedded_test_server()->RegisterRequestHandler(
+      base::Bind(
+          &UserAgentResponseHandler,
+          kUserAgentRedirectResponsePath,
+          embedded_test_server()->GetURL(kRedirectResponseFullPath)));
 }
 
 void WebViewAPITest::StopTestServer() {
   if (!embedded_test_server()->ShutdownAndWaitUntilComplete()) {
     LOG(ERROR) << "Failed to shutdown test server.";
   }
 }
 
 void WebViewAPITest::TearDownOnMainThread() {
   TestGetConfigFunction::set_test_config_state(nullptr);
@@ skipping 283 matching lines @@
 IN_PROC_BROWSER_TEST_F(WebViewAPITest, TestTerminateAfterExit) {
   RunTest("testTerminateAfterExit", "web_view/apitest");
 }
 
 IN_PROC_BROWSER_TEST_F(WebViewAPITest, TestWebRequestAPI) {
   StartTestServer();
   RunTest("testWebRequestAPI", "web_view/apitest");
   StopTestServer();
 }
 
+IN_PROC_BROWSER_TEST_F(WebViewAPITest, TestWebRequestAPIWithHeaders) {
+  StartTestServer();
+  RunTest("testWebRequestAPIWithHeaders", "web_view/apitest");
+  StopTestServer();
+}
+
 // Tests the existence of WebRequest API event objects on the request
 // object, on the webview element, and hanging directly off webview.
 IN_PROC_BROWSER_TEST_F(WebViewAPITest, TestWebRequestAPIExistence) {
   StartTestServer();
   RunTest("testWebRequestAPIExistence", "web_view/apitest");
   StopTestServer();
 }
 
 IN_PROC_BROWSER_TEST_F(WebViewAPITest, TestWebRequestAPIGoogleProperty) {
   StartTestServer();
   RunTest("testWebRequestAPIGoogleProperty", "web_view/apitest");
   StopTestServer();
 }
 
 }  // namespace extensions