headless: Add support for net logging

headless/lib/headless_browser_browsertest.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 <memory>
 
 #include "base/command_line.h"
 #include "base/files/file_enumerator.h"
 #include "base/files/file_util.h"
 #include "base/memory/ptr_util.h"
@@ skipping 771 matching lines @@
   content::PermissionManager* permission_manager =
       web_contents->browser_context()->GetPermissionManager();
   EXPECT_NE(nullptr, permission_manager);
 
   // Check that the permission manager returns ASK for a given permission type.
   EXPECT_EQ(blink::mojom::PermissionStatus::ASK,
             permission_manager->GetPermissionStatus(
                 content::PermissionType::NOTIFICATIONS, url, url));
 }
 
+class HeadlessBrowserTestWithNetLog : public HeadlessBrowserTest {
+ public:
+  HeadlessBrowserTestWithNetLog() {}
+
+  void SetUp() override {
+    base::ThreadRestrictions::SetIOAllowed(true);
+    EXPECT_TRUE(base::CreateTemporaryFile(&net_log_));
+    base::CommandLine::ForCurrentProcess()->AppendSwitchASCII("--log-net-log",
+                                                              net_log_.value());
+    HeadlessBrowserTest::SetUp();
+  }
+
+  void TearDown() override {
+    HeadlessBrowserTest::TearDown();
+    base::DeleteFile(net_log_, false);
+  }
+
+ protected:
+  base::FilePath net_log_;
+};
+
+IN_PROC_BROWSER_TEST_F(HeadlessBrowserTestWithNetLog, WriteNetLog) {
+  EXPECT_TRUE(embedded_test_server()->Start());
+
+  HeadlessBrowserContext* browser_context =
+      browser()->CreateBrowserContextBuilder().Build();
+
+  HeadlessWebContents* web_contents =
+      browser_context->CreateWebContentsBuilder()
+          .SetInitialURL(embedded_test_server()->GetURL("/hello.html"))
+          .Build();
+  EXPECT_TRUE(WaitForLoad(web_contents));
+  browser()->Shutdown();
+
+  base::ThreadRestrictions::SetIOAllowed(true);
+  std::string net_log_data;
+  EXPECT_TRUE(base::ReadFileToString(net_log_, &net_log_data));
+  EXPECT_GE(net_log_data.find("hello.html"), 0u);
+}
+
 }  // namespace headless