Index: net/url_request/url_request_simple_job_unittest.cc |
diff --git a/net/url_request/url_request_simple_job_unittest.cc b/net/url_request/url_request_simple_job_unittest.cc |
index 088b157e525845a458de5b80382df87d4c64f11f..895616bff00a649e02057748d55ac250038edcd9 100644 |
--- a/net/url_request/url_request_simple_job_unittest.cc |
+++ b/net/url_request/url_request_simple_job_unittest.cc |
@@ -10,12 +10,11 @@ |
#include "base/bind_helpers.h" |
#include "base/macros.h" |
#include "base/memory/ptr_util.h" |
-#include "base/message_loop/message_loop.h" |
#include "base/run_loop.h" |
#include "base/sequenced_task_runner.h" |
#include "base/strings/string_piece.h" |
#include "base/strings/stringprintf.h" |
-#include "base/test/scoped_task_scheduler.h" |
+#include "base/task_scheduler/task_scheduler.h" |
#include "net/base/request_priority.h" |
#include "net/test/gtest_util.h" |
#include "net/traffic_annotation/network_traffic_annotation_test_helper.h" |
@@ -112,8 +111,7 @@ class SimpleJobProtocolHandler : |
class URLRequestSimpleJobTest : public ::testing::Test { |
public: |
- URLRequestSimpleJobTest() |
- : context_(true), scoped_task_scheduler_(base::MessageLoop::current()) { |
+ URLRequestSimpleJobTest() : context_(true) { |
job_factory_.SetProtocolHandler( |
"data", base::MakeUnique<SimpleJobProtocolHandler>()); |
context_.set_job_factory(&job_factory_); |
@@ -140,8 +138,6 @@ class URLRequestSimpleJobTest : public ::testing::Test { |
std::unique_ptr<URLRequest> request_; |
private: |
- base::test::ScopedTaskScheduler scoped_task_scheduler_; |
- |
DISALLOW_COPY_AND_ASSIGN(URLRequestSimpleJobTest); |
}; |
@@ -223,7 +219,8 @@ TEST_F(URLRequestSimpleJobTest, CancelAfterFirstReadStarted) { |
request_->Start(); |
cancel_delegate.WaitUntilHeadersReceived(); |
- // Run ScopedTaskScheduler tasks. |
+ // Run TaskScheduler tasks and their reply on the main thread. |
+ base::TaskScheduler::GetInstance()->FlushForTesting(); |
base::RunLoop().RunUntilIdle(); |
EXPECT_THAT(cancel_delegate.request_status(), IsError(ERR_ABORTED)); |