Use TaskScheduler in directory_lister.cc.

net/url_request/url_request_file_dir_job_unittest.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 "net/url_request/url_request_file_dir_job.h"
 
 #include <memory>
 #include <string>
 
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
+#include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
-#include "base/threading/thread_task_runner_handle.h"
+#include "base/test/scoped_task_scheduler.h"
 #include "net/base/filename_util.h"
 #include "net/base/io_buffer.h"
 #include "net/base/net_errors.h"
 #include "net/test/gtest_util.h"
 #include "net/url_request/url_request.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using net::test::IsError;
 using net::test::IsOk;
 
 namespace net {
 
 namespace {
 
 const int kBufferSize = 4096;
 
 class TestJobFactory : public URLRequestJobFactory {
  public:
   explicit TestJobFactory(const base::FilePath& path) : path_(path) {}
   ~TestJobFactory() override {}
 
   URLRequestJob* MaybeCreateJobWithProtocolHandler(
       const std::string& scheme,
       URLRequest* request,
       NetworkDelegate* network_delegate) const override {
-    URLRequestJob* job = new URLRequestFileDirJob(
-        request, network_delegate, path_, base::ThreadTaskRunnerHandle::Get());
+    URLRequestJob* job =
+        new URLRequestFileDirJob(request, network_delegate, path_);
 
     return job;
   }
 
   URLRequestJob* MaybeInterceptRedirect(URLRequest* request,
                                         NetworkDelegate* network_delegate,
                                         const GURL& location) const override {
     return nullptr;
   }
 
@@ skipping 34 matching lines @@
   bool got_response_started() const { return got_response_started_; }
 
  private:
   bool got_response_started_ = false;
 
   DISALLOW_COPY_AND_ASSIGN(TestDirectoryURLRequestDelegate);
 };
 
 class URLRequestFileDirTest : public testing::Test {
  public:
-  URLRequestFileDirTest() : buffer_(new IOBuffer(kBufferSize)) {}
+  URLRequestFileDirTest()
+      : scoped_task_scheduler_(base::MessageLoop::current()),
+        buffer_(new IOBuffer(kBufferSize)) {}
 
  protected:
+  base::test::ScopedTaskScheduler scoped_task_scheduler_;
   TestURLRequestContext context_;
   TestDirectoryURLRequestDelegate delegate_;
   scoped_refptr<IOBuffer> buffer_;
 };
 
 TEST_F(URLRequestFileDirTest, ListCompletionOnNoPending) {
   base::ScopedTempDir directory;
   // It is necessary to pass an existing directory to UrlRequest object,
   // but it will be deleted for testing purpose after request is started.
   ASSERT_TRUE(directory.CreateUniqueTempDir());
@@ skipping 132 matching lines @@
   ASSERT_GT(delegate.bytes_received(), 0);
   ASSERT_LE(delegate.bytes_received(), kBufferSize);
   EXPECT_TRUE(delegate.data_received().find(
                   directory.GetPath().BaseName().MaybeAsASCII()) !=
               std::string::npos);
 }
 
 }  // namespace
 
 }  // namespace net