OLD | NEW |
1 // Copyright 2017 The Chromium Authors. All rights reserved. | 1 // Copyright 2017 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/files/file_util.h" | 5 #include "base/files/file_util.h" |
6 #include "base/message_loop/message_loop.h" | |
7 #include "base/path_service.h" | 6 #include "base/path_service.h" |
8 #include "base/run_loop.h" | 7 #include "base/run_loop.h" |
| 8 #include "base/test/scoped_task_environment.h" |
9 #include "base/threading/thread_task_runner_handle.h" | 9 #include "base/threading/thread_task_runner_handle.h" |
10 #include "content/browser/loader/test_url_loader_client.h" | 10 #include "content/browser/loader/test_url_loader_client.h" |
11 #include "content/network/network_context.h" | 11 #include "content/network/network_context.h" |
12 #include "content/network/url_loader_impl.h" | 12 #include "content/network/url_loader_impl.h" |
13 #include "content/public/common/content_paths.h" | 13 #include "content/public/common/content_paths.h" |
14 #include "mojo/public/c/system/data_pipe.h" | 14 #include "mojo/public/c/system/data_pipe.h" |
15 #include "mojo/public/cpp/system/wait.h" | 15 #include "mojo/public/cpp/system/wait.h" |
16 #include "net/test/embedded_test_server/embedded_test_server.h" | 16 #include "net/test/embedded_test_server/embedded_test_server.h" |
17 #include "net/traffic_annotation/network_traffic_annotation_test_helper.h" | 17 #include "net/traffic_annotation/network_traffic_annotation_test_helper.h" |
18 #include "testing/gtest/include/gtest/gtest.h" | 18 #include "testing/gtest/include/gtest/gtest.h" |
(...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
59 MOJO_RESULT_OK); | 59 MOJO_RESULT_OK); |
60 CHECK_EQ(num_bytes, static_cast<uint32_t>(size)); | 60 CHECK_EQ(num_bytes, static_cast<uint32_t>(size)); |
61 | 61 |
62 return std::string(buffer.data(), buffer.size()); | 62 return std::string(buffer.data(), buffer.size()); |
63 } | 63 } |
64 | 64 |
65 } // namespace | 65 } // namespace |
66 | 66 |
67 class URLLoaderImplTest : public testing::Test { | 67 class URLLoaderImplTest : public testing::Test { |
68 public: | 68 public: |
69 URLLoaderImplTest() : context_(NetworkContext::CreateForTesting()) {} | 69 URLLoaderImplTest() |
| 70 : scoped_task_environment_( |
| 71 base::test::ScopedTaskEnvironment::MainThreadType::IO), |
| 72 context_(NetworkContext::CreateForTesting()) {} |
70 ~URLLoaderImplTest() override {} | 73 ~URLLoaderImplTest() override {} |
71 | 74 |
72 void SetUp() override { | 75 void SetUp() override { |
73 test_server_.AddDefaultHandlers( | 76 test_server_.AddDefaultHandlers( |
74 base::FilePath(FILE_PATH_LITERAL("content/test/data"))); | 77 base::FilePath(FILE_PATH_LITERAL("content/test/data"))); |
75 ASSERT_TRUE(test_server_.Start()); | 78 ASSERT_TRUE(test_server_.Start()); |
76 } | 79 } |
77 | 80 |
78 void Load(const GURL& url, | 81 void Load(const GURL& url, |
79 TestURLLoaderClient* client, | 82 TestURLLoaderClient* client, |
(...skipping 28 matching lines...) Expand all Loading... |
108 std::string data = | 111 std::string data = |
109 ReadData(client.response_body().value(), file_contents.size()); | 112 ReadData(client.response_body().value(), file_contents.size()); |
110 CHECK_EQ(data, file_contents); | 113 CHECK_EQ(data, file_contents); |
111 } | 114 } |
112 | 115 |
113 net::EmbeddedTestServer* test_server() { return &test_server_; } | 116 net::EmbeddedTestServer* test_server() { return &test_server_; } |
114 NetworkContext* context() { return context_.get(); } | 117 NetworkContext* context() { return context_.get(); } |
115 void DestroyContext() { context_.reset(); } | 118 void DestroyContext() { context_.reset(); } |
116 | 119 |
117 private: | 120 private: |
118 base::MessageLoopForIO message_loop_; | 121 base::test::ScopedTaskEnvironment scoped_task_environment_; |
119 net::EmbeddedTestServer test_server_; | 122 net::EmbeddedTestServer test_server_; |
120 std::unique_ptr<NetworkContext> context_; | 123 std::unique_ptr<NetworkContext> context_; |
121 }; | 124 }; |
122 | 125 |
123 TEST_F(URLLoaderImplTest, Basic) { | 126 TEST_F(URLLoaderImplTest, Basic) { |
124 LoadAndCompareFile("simple_page.html"); | 127 LoadAndCompareFile("simple_page.html"); |
125 } | 128 } |
126 | 129 |
127 TEST_F(URLLoaderImplTest, Empty) { | 130 TEST_F(URLLoaderImplTest, Empty) { |
128 LoadAndCompareFile("empty.html"); | 131 LoadAndCompareFile("empty.html"); |
(...skipping 53 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
182 // client receiving a connection error. | 185 // client receiving a connection error. |
183 DestroyContext(); | 186 DestroyContext(); |
184 EXPECT_FALSE(loader_impl); | 187 EXPECT_FALSE(loader_impl); |
185 | 188 |
186 client.RunUntilConnectionError(); | 189 client.RunUntilConnectionError(); |
187 EXPECT_FALSE(client.has_received_completion()); | 190 EXPECT_FALSE(client.has_received_completion()); |
188 EXPECT_EQ(0u, client.download_data_length()); | 191 EXPECT_EQ(0u, client.download_data_length()); |
189 } | 192 } |
190 | 193 |
191 } // namespace content | 194 } // namespace content |
OLD | NEW |