Migrating tests to use EmbeddedTestServer (misc)

chrome/renderer/safe_browsing/phishing_dom_feature_extractor_browsertest.cc

 // Copyright (c) 2012 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.
 //
 // Note that although this is not a "browser" test, it runs as part of
 // browser_tests.  This is because WebKit does not work properly if it is
 // shutdown and re-initialized.  Since browser_tests runs each test in a
 // new process, this avoids the problem.
 
 #include "chrome/renderer/safe_browsing/phishing_dom_feature_extractor.h"
@@ skipping 66 matching lines @@
     // Don't want to try to create a GPU process.
     command_line->AppendSwitch(switches::kDisableGpu);
 #endif
   }
 
   void SetUpOnMainThread() override {
     render_view_routing_id_ =
         GetWebContents()->GetRenderViewHost()->GetRoutingID();
     extractor_.reset(new PhishingDOMFeatureExtractor(&clock_));
 
-    ASSERT_TRUE(StartTestServer());
+    embedded_test_server()->RegisterRequestHandler(
+        base::Bind(&PhishingDOMFeatureExtractorTest::HandleRequest,
+                   base::Unretained(this)));
+    ASSERT_TRUE(embedded_test_server()->Start());
     host_resolver()->AddRule("*", "127.0.0.1");
   }
 
   // Runs the DOMFeatureExtractor on the RenderView, waiting for the
   // completion callback.  Returns the success boolean from the callback.
   bool ExtractFeatures(FeatureMap* features) {
     success_ = false;
     PostTaskToInProcessRendererAndWait(
         base::Bind(&PhishingDOMFeatureExtractorTest::ExtractFeaturesInternal,
         base::Unretained(this),
@@ skipping 27 matching lines @@
 
   // Does the actual work of removing the iframe "frame1" from the document.
   void RemoveIframe() {
     blink::WebFrame* main_frame = GetWebFrame();
     ASSERT_TRUE(main_frame);
     main_frame->executeScript(
         blink::WebString(
             "document.body.removeChild(document.getElementById('frame1'));"));
   }
 
-  bool StartTestServer() {
-    CHECK(!embedded_test_server_);
-    embedded_test_server_.reset(new net::test_server::EmbeddedTestServer());
-    embedded_test_server_->RegisterRequestHandler(
-        base::Bind(&PhishingDOMFeatureExtractorTest::HandleRequest,
-                   base::Unretained(this)));
-    return embedded_test_server_->InitializeAndWaitUntilReady();
-  }
-
   scoped_ptr<net::test_server::HttpResponse> HandleRequest(
       const net::test_server::HttpRequest& request) {
     std::map<std::string, std::string>::const_iterator host_it =
         request.headers.find("Host");
     if (host_it == request.headers.end())
       return scoped_ptr<net::test_server::HttpResponse>();
 
     std::string url =
         std::string("http://") + host_it->second + request.relative_url;
     std::map<std::string, std::string>::const_iterator it =
         responses_.find(url);
     if (it == responses_.end())
       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_OK);
     http_response->set_content_type("text/html");
     http_response->set_content(it->second);
     return http_response.Pass();
   }
 
   GURL GetURL(const std::string& host, const std::string& path) {
     GURL::Replacements replace;
     replace.SetHostStr(host);
     replace.SetPathStr(path);
-    return embedded_test_server_->base_url().ReplaceComponents(replace);
+    return embedded_test_server()->base_url().ReplaceComponents(replace);
   }
 
   // Returns the URL that was loaded.
   GURL LoadHtml(const std::string& host, const std::string& content) {
     GURL url(GetURL(host, ""));
     responses_[url.spec()] = content;
     ui_test_utils::NavigateToURL(browser(), url);
     return url;
   }
 
   int32 render_view_routing_id_;
 
   // Map of url -> response body for network requests from the renderer.
   // Any urls not in this map are served a 404 error.
   std::map<std::string, std::string> responses_;
 
-  scoped_ptr<net::test_server::EmbeddedTestServer> embedded_test_server_;
   MockFeatureExtractorClock clock_;
   scoped_ptr<PhishingDOMFeatureExtractor> extractor_;
   bool success_;  // holds the success value from ExtractFeatures
   base::WeakPtrFactory<PhishingDOMFeatureExtractorTest> weak_factory_;
 };
 
 IN_PROC_BROWSER_TEST_F(PhishingDOMFeatureExtractorTest, FormFeatures) {
   // This test doesn't exercise the extraction timing.
   EXPECT_CALL(clock_, Now()).WillRepeatedly(Return(base::TimeTicks::Now()));
 
   FeatureMap expected_features;
   expected_features.AddBooleanFeature(features::kPageHasForms);
   expected_features.AddRealFeature(features::kPageActionOtherDomainFreq, 0.25);
   expected_features.AddBooleanFeature(features::kPageHasTextInputs);
   expected_features.AddBooleanFeature(features::kPageHasCheckInputs);
   expected_features.AddBooleanFeature(features::kPageActionURL +
       std::string("http://cgi.host.com/submit"));
   expected_features.AddBooleanFeature(features::kPageActionURL +
       std::string("http://other.com/"));
-  expected_features.AddBooleanFeature(features::kPageActionURL +
-      std::string("http://host.com:") +
-      base::UintToString(embedded_test_server_->port()) +
-      std::string("/query"));
+
+  GURL url = embedded_test_server()->GetURL("/query");
+  GURL::Replacements replace_host;
+  replace_host.SetHostStr("host.com");
+  expected_features.AddBooleanFeature(
+      features::kPageActionURL + url.ReplaceComponents(replace_host).spec());
 
   FeatureMap features;
   LoadHtml(
       "host.com",
       "<html><head><body>"
       "<form action=\"query\"><input type=text><input type=checkbox></form>"
       "<form action=\"http://cgi.host.com/submit\"></form>"
       "<form action=\"http://other.com/\"></form>"
       "<form action=\"query\"></form>"
       "<form></form></body></html>");
@@ skipping 53 matching lines @@
       "</body></html");
   ASSERT_TRUE(ExtractFeatures(&features));
   ExpectFeatureMapsAreEqual(features, expected_features);
 
   expected_features.Clear();
   expected_features.AddRealFeature(features::kPageExternalLinksFreq, 0.25);
   expected_features.AddRealFeature(features::kPageSecureLinksFreq, 0.5);
   expected_features.AddBooleanFeature(features::kPageLinkDomain +
                                       std::string("chromium.org"));
 
-  net::SpawnedTestServer https_server(
-      net::SpawnedTestServer::TYPE_HTTPS,
-      net::SpawnedTestServer::kLocalhost,
-      base::FilePath(FILE_PATH_LITERAL("chrome/test/data")));
+  net::EmbeddedTestServer https_server(net::EmbeddedTestServer::TYPE_HTTPS);
+  https_server.ServeFilesFromSourceDirectory("chrome/test/data");
   ASSERT_TRUE(https_server.Start());
 
   // The PhishingDOMFeatureExtractor depends on URLs being domains and not IPs,
   // so use a domain.
-  std::string url_str = "https://host.com:";
-  url_str += base::UintToString(https_server.host_port_pair().port());
-  url_str += "/files/safe_browsing/secure_link_features.html";
-  ui_test_utils::NavigateToURL(browser(), GURL(url_str));
+  GURL url = https_server.GetURL("/safe_browsing/secure_link_features.html");
+  GURL::Replacements replace_host;
+  replace_host.SetHostStr("host.com");
+  ui_test_utils::NavigateToURL(browser(), url.ReplaceComponents(replace_host));
 
   // Click through the certificate error interstitial.
   content::InterstitialPage* interstitial_page =
       GetWebContents()->GetInterstitialPage();
   interstitial_page->Proceed();
   content::WaitForLoadStop(GetWebContents());
 
   features.Clear();
   ASSERT_TRUE(ExtractFeatures(&features));
   ExpectFeatureMapsAreEqual(features, expected_features);
@@ skipping 19 matching lines @@
       "<html><head><script></script><script></script></head></html>");
   ASSERT_TRUE(ExtractFeatures(&features));
   ExpectFeatureMapsAreEqual(features, expected_features);
 
   expected_features.Clear();
   expected_features.AddBooleanFeature(features::kPageNumScriptTagsGTOne);
   expected_features.AddBooleanFeature(features::kPageNumScriptTagsGTSix);
   expected_features.AddRealFeature(features::kPageImgOtherDomainFreq, 0.5);
 
   features.Clear();
-  net::SpawnedTestServer https_server(
-      net::SpawnedTestServer::TYPE_HTTPS,
-      net::SpawnedTestServer::kLocalhost,
-      base::FilePath(FILE_PATH_LITERAL("chrome/test/data")));
+  net::EmbeddedTestServer https_server(net::EmbeddedTestServer::TYPE_HTTPS);
+  https_server.ServeFilesFromSourceDirectory("chrome/test/data");
   ASSERT_TRUE(https_server.Start());
 
   // The PhishingDOMFeatureExtractor depends on URLs being domains and not IPs,
   // so use a domain.
-  std::string url_str = "https://host.com:";
-  url_str += base::UintToString(https_server.host_port_pair().port());
-  url_str += "/files/safe_browsing/secure_script_and_image.html";
-  ui_test_utils::NavigateToURL(browser(), GURL(url_str));
+  GURL url = embedded_test_server()->GetURL(
+      "/safe_browsing/secure_script_and_image.html");
+  GURL::Replacements replace_host;
+  replace_host.SetHostStr("host.com");
+  ui_test_utils::NavigateToURL(browser(), url.ReplaceComponents(replace_host));
 
   // Click through the certificate error interstitial.
   content::InterstitialPage* interstitial_page =
       GetWebContents()->GetInterstitialPage();
   interstitial_page->Proceed();
   content::WaitForLoadStop(GetWebContents());
 
   ASSERT_TRUE(ExtractFeatures(&features));
   ExpectFeatureMapsAreEqual(features, expected_features);
 }
 
 IN_PROC_BROWSER_TEST_F(PhishingDOMFeatureExtractorTest, SubFrames) {
   // This test doesn't exercise the extraction timing.
   EXPECT_CALL(clock_, Now()).WillRepeatedly(Return(base::TimeTicks::Now()));
 
   // Test that features are aggregated across all frames.
 
-  std::string port = base::UintToString(embedded_test_server_->port());
   responses_[GetURL("host2.com", "").spec()] =
       "<html><head><script></script><body>"
       "<form action=\"http://host4.com/\"><input type=checkbox></form>"
       "<form action=\"http://host2.com/submit\"></form>"
       "<a href=\"http://www.host2.com/home\">link</a>"
       "<iframe src=\"nested.html\"></iframe>"
       "<body></html>";
 
   responses_[GetURL("host2.com", "nested.html").spec()] =
       "<html><body><input type=password>"
@@ skipping 20 matching lines @@
                                       std::string("host4.com"));
   expected_features.AddRealFeature(features::kPageSecureLinksFreq, 0.25);
   expected_features.AddBooleanFeature(features::kPageNumScriptTagsGTOne);
   expected_features.AddRealFeature(features::kPageImgOtherDomainFreq, 1.0);
   expected_features.AddBooleanFeature(features::kPageActionURL +
       std::string("http://host2.com/submit"));
   expected_features.AddBooleanFeature(features::kPageActionURL +
       std::string("http://host4.com/"));
 
   FeatureMap features;
+  GURL url = embedded_test_server()->GetURL("/");
+  GURL::Replacements replace_host;
+  replace_host.SetHostStr("host2.com");
+  GURL::Replacements replace_host_2;
+  replace_host_2.SetHostStr("host3.com");
+
   std::string html(
       "<html><body><input type=text><a href=\"info.html\">link</a>"
-      "<iframe src=\"http://host2.com:");
-  html += port;
-  html += std::string(
-      "/\"></iframe>"
-      "<iframe src=\"http://host3.com:");
-  html += port;
-  html += std::string("/\"></iframe></body></html>");
+      "<iframe src=\"");
+  html += url.ReplaceComponents(replace_host).spec();
+  html += std::string("\"></iframe><iframe src=\"");
+  html += url.ReplaceComponents(replace_host_2).spec();
+  html += std::string("\"></iframe></body></html>");
 
   LoadHtml("host.com", html);
   ASSERT_TRUE(ExtractFeatures(&features));
   ExpectFeatureMapsAreEqual(features, expected_features);
 }
 
 // Test flakes with LSAN enabled. See http://crbug.com/373155.
 #if defined(LEAK_SANITIZER)
 #define MAYBE_Continuation DISABLED_Continuation
 #else
@@ skipping 42 matching lines @@
       // Time check after resuming iteration for the third chunk.
       .WillOnce(Return(now + base::TimeDelta::FromMilliseconds(48)))
       // Time check after the last 10 elements.
       .WillOnce(Return(now + base::TimeDelta::FromMilliseconds(54)))
       // A final time check for the histograms.
       .WillOnce(Return(now + base::TimeDelta::FromMilliseconds(56)));
 
   FeatureMap expected_features;
   expected_features.AddBooleanFeature(features::kPageHasForms);
   expected_features.AddRealFeature(features::kPageActionOtherDomainFreq, 0.5);
-  expected_features.AddBooleanFeature(features::kPageActionURL +
-      std::string("http://host.com:") +
-      base::UintToString(embedded_test_server_->port()) +
-      std::string("/ondomain"));
+  GURL url = embedded_test_server()->GetURL("/ondomain");
+  GURL::Replacements replace_host;
+  replace_host.SetHostStr("host.com");
+  expected_features.AddBooleanFeature(
+      features::kPageActionURL + url.ReplaceComponents(replace_host).spec());
   expected_features.AddBooleanFeature(features::kPageActionURL +
       std::string("http://host2.com/"));
 
   FeatureMap features;
   LoadHtml("host.com", response);
   ASSERT_TRUE(ExtractFeatures(&features));
   ExpectFeatureMapsAreEqual(features, expected_features);
   // Make sure none of the mock expectations carry over to the next test.
   ::testing::Mock::VerifyAndClearExpectations(&clock_);
 
@@ skipping 57 matching lines @@
   LoadHtml(
       "host.com",
       "<html><head></head><body>"
       "<iframe src=\"frame.html\" id=\"frame1\"></iframe>"
       "<form></form></body></html>");
   ASSERT_TRUE(ExtractFeatures(&features));
   ExpectFeatureMapsAreEqual(features, expected_features);
 }
 
 }  // namespace safe_browsing