headless: Fix two issues that are making the bot red

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_util.h"
 #include "base/memory/ptr_util.h"
 #include "base/strings/stringprintf.h"
+#include "base/threading/thread_restrictions.h"
 #include "content/public/test/browser_test.h"
 #include "headless/public/domains/network.h"
 #include "headless/public/domains/page.h"
 #include "headless/public/headless_browser.h"
 #include "headless/public/headless_devtools_client.h"
 #include "headless/public/headless_devtools_target.h"
 #include "headless/public/headless_web_contents.h"
 #include "headless/test/headless_browser_test.h"
 #include "headless/test/test_protocol_handler.h"
 #include "headless/test/test_url_request_job.h"
@@ skipping 575 matching lines @@
 }
 
 // TODO(skyostil): This test currently relies on being able to run a shell
 // script.
 #if defined(OS_POSIX)
 #define MAYBE_RendererCommandPrefixTest RendererCommandPrefixTest
 #else
 #define MAYBE_RendererCommandPrefixTest DISABLED_RendererCommandPrefixTest
 #endif  // defined(OS_POSIX)
 IN_PROC_BROWSER_TEST_F(HeadlessBrowserTest, MAYBE_RendererCommandPrefixTest) {
+  base::ThreadRestrictions::SetIOAllowed(true);
   base::FilePath launcher_stamp;
   base::CreateTemporaryFile(&launcher_stamp);
 
   base::FilePath launcher_script;
   FILE* launcher_file = base::CreateAndOpenTemporaryFile(&launcher_script);
   fprintf(launcher_file, "#!/bin/sh\n");
   fprintf(launcher_file, "echo $@ > %s\n", launcher_stamp.value().c_str());
   fprintf(launcher_file, "exec $@\n");
   fclose(launcher_file);
   base::SetPosixFilePermissions(launcher_script,
@@ skipping 18 matching lines @@
   // Make sure the launcher was invoked when starting the renderer.
   std::string stamp;
   EXPECT_TRUE(base::ReadFileToString(launcher_stamp, &stamp));
   EXPECT_GE(stamp.find("--type=renderer"), 0u);
 
   base::DeleteFile(launcher_script, false);
   base::DeleteFile(launcher_stamp, false);
 }
 
 }  // namespace headless