proxy_resolver_factory_impl_unittest.cc 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285
  1. // Copyright 2015 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "services/proxy_resolver/proxy_resolver_factory_impl.h"
  5. #include <utility>
  6. #include "base/bind.h"
  7. #include "base/callback_helpers.h"
  8. #include "base/memory/raw_ptr.h"
  9. #include "base/run_loop.h"
  10. #include "base/strings/utf_string_conversions.h"
  11. #include "base/test/task_environment.h"
  12. #include "mojo/public/cpp/bindings/receiver.h"
  13. #include "mojo/public/cpp/bindings/remote.h"
  14. #include "net/base/completion_once_callback.h"
  15. #include "net/base/test_completion_callback.h"
  16. #include "net/proxy_resolution/mock_proxy_resolver.h"
  17. #include "net/proxy_resolution/proxy_resolve_dns_operation.h"
  18. #include "net/test/event_waiter.h"
  19. #include "net/test/gtest_util.h"
  20. #include "services/proxy_resolver/proxy_resolver_v8_tracing.h"
  21. #include "testing/gmock/include/gmock/gmock.h"
  22. #include "testing/gtest/include/gtest/gtest.h"
  23. #include "third_party/abseil-cpp/absl/types/optional.h"
  24. using net::test::IsError;
  25. using net::test::IsOk;
  26. namespace proxy_resolver {
  27. namespace {
  28. const char kScriptData[] = "FooBarBaz";
  29. const char16_t kScriptData16[] = u"FooBarBaz";
  30. class FakeProxyResolver : public ProxyResolverV8Tracing {
  31. public:
  32. explicit FakeProxyResolver(base::OnceClosure on_destruction)
  33. : on_destruction_(std::move(on_destruction)) {}
  34. ~FakeProxyResolver() override { std::move(on_destruction_).Run(); }
  35. private:
  36. // ProxyResolverV8Tracing overrides.
  37. void GetProxyForURL(const GURL& url,
  38. const net::NetworkIsolationKey& network_isolation_key,
  39. net::ProxyInfo* results,
  40. net::CompletionOnceCallback callback,
  41. std::unique_ptr<net::ProxyResolver::Request>* request,
  42. std::unique_ptr<Bindings> bindings) override {}
  43. base::OnceClosure on_destruction_;
  44. };
  45. enum Event {
  46. NONE,
  47. RESOLVER_CREATED,
  48. CONNECTION_ERROR,
  49. RESOLVER_DESTROYED,
  50. };
  51. class TestProxyResolverFactory : public ProxyResolverV8TracingFactory {
  52. public:
  53. struct PendingRequest {
  54. raw_ptr<std::unique_ptr<ProxyResolverV8Tracing>> resolver;
  55. net::CompletionOnceCallback callback;
  56. };
  57. explicit TestProxyResolverFactory(net::EventWaiter<Event>* waiter)
  58. : waiter_(waiter) {}
  59. ~TestProxyResolverFactory() override {}
  60. void CreateProxyResolverV8Tracing(
  61. const scoped_refptr<net::PacFileData>& pac_script,
  62. std::unique_ptr<ProxyResolverV8Tracing::Bindings> bindings,
  63. std::unique_ptr<ProxyResolverV8Tracing>* resolver,
  64. net::CompletionOnceCallback callback,
  65. std::unique_ptr<net::ProxyResolverFactory::Request>* request) override {
  66. requests_handled_++;
  67. waiter_->NotifyEvent(RESOLVER_CREATED);
  68. EXPECT_EQ(kScriptData16, pac_script->utf16());
  69. EXPECT_TRUE(resolver);
  70. pending_request_ = std::make_unique<PendingRequest>();
  71. pending_request_->resolver = resolver;
  72. pending_request_->callback = std::move(callback);
  73. ASSERT_TRUE(bindings);
  74. bindings->Alert(u"alert");
  75. bindings->OnError(10, u"error");
  76. EXPECT_TRUE(bindings->GetHostResolver());
  77. }
  78. size_t requests_handled() { return requests_handled_; }
  79. PendingRequest* pending_request() { return pending_request_.get(); }
  80. private:
  81. raw_ptr<net::EventWaiter<Event>> waiter_;
  82. size_t requests_handled_ = 0;
  83. std::unique_ptr<PendingRequest> pending_request_;
  84. };
  85. class TestProxyResolverFactoryImpl : public ProxyResolverFactoryImpl {
  86. public:
  87. TestProxyResolverFactoryImpl(
  88. mojo::PendingReceiver<mojom::ProxyResolverFactory> receiver,
  89. std::unique_ptr<ProxyResolverV8TracingFactory> factory)
  90. : ProxyResolverFactoryImpl(std::move(receiver), std::move(factory)) {}
  91. };
  92. } // namespace
  93. class ProxyResolverFactoryImplTest
  94. : public testing::Test,
  95. public mojom::ProxyResolverFactoryRequestClient {
  96. public:
  97. ProxyResolverFactoryImplTest() {
  98. std::unique_ptr<TestProxyResolverFactory> test_factory =
  99. std::make_unique<TestProxyResolverFactory>(&waiter_);
  100. mock_factory_ = test_factory.get();
  101. mock_factory_impl_ = std::make_unique<TestProxyResolverFactoryImpl>(
  102. factory_.BindNewPipeAndPassReceiver(), std::move(test_factory));
  103. factory_.set_idle_handler(
  104. base::TimeDelta(),
  105. base::BindRepeating(&ProxyResolverFactoryImplTest::OnFactoryIdle,
  106. base::Unretained(this)));
  107. }
  108. ~ProxyResolverFactoryImplTest() override = default;
  109. void OnDisconnect() { waiter_.NotifyEvent(CONNECTION_ERROR); }
  110. void OnFakeProxyInstanceDestroyed() {
  111. instances_destroyed_++;
  112. waiter_.NotifyEvent(RESOLVER_DESTROYED);
  113. }
  114. void ReportResult(int32_t error) override {
  115. std::move(create_callback_).Run(error);
  116. }
  117. void Alert(const std::string& message) override {}
  118. void OnError(int32_t line_number, const std::string& message) override {}
  119. void ResolveDns(
  120. const std::string& hostname,
  121. net::ProxyResolveDnsOperation operation,
  122. const net::NetworkIsolationKey& network_isolation_key,
  123. mojo::PendingRemote<mojom::HostResolverRequestClient> client) override {}
  124. void set_idle_callback(base::OnceClosure callback) {
  125. idle_callback_ = std::move(callback);
  126. }
  127. protected:
  128. void OnFactoryIdle() {
  129. if (idle_callback_)
  130. std::move(idle_callback_).Run();
  131. }
  132. base::test::TaskEnvironment task_environment_;
  133. std::unique_ptr<TestProxyResolverFactoryImpl> mock_factory_impl_;
  134. raw_ptr<TestProxyResolverFactory> mock_factory_;
  135. mojo::Remote<mojom::ProxyResolverFactory> factory_;
  136. int instances_destroyed_ = 0;
  137. net::CompletionOnceCallback create_callback_;
  138. base::OnceClosure idle_callback_;
  139. net::EventWaiter<Event> waiter_;
  140. };
  141. TEST_F(ProxyResolverFactoryImplTest, DisconnectProxyResolverClient) {
  142. mojo::Remote<mojom::ProxyResolver> proxy_resolver;
  143. mojo::PendingRemote<mojom::ProxyResolverFactoryRequestClient> client;
  144. mojo::Receiver<ProxyResolverFactoryRequestClient> client_receiver(
  145. this, client.InitWithNewPipeAndPassReceiver());
  146. factory_->CreateResolver(kScriptData,
  147. proxy_resolver.BindNewPipeAndPassReceiver(),
  148. std::move(client));
  149. proxy_resolver.set_disconnect_handler(base::BindOnce(
  150. &ProxyResolverFactoryImplTest::OnDisconnect, base::Unretained(this)));
  151. waiter_.WaitForEvent(RESOLVER_CREATED);
  152. EXPECT_EQ(0, instances_destroyed_);
  153. ASSERT_EQ(1u, mock_factory_->requests_handled());
  154. net::TestCompletionCallback create_callback;
  155. create_callback_ = create_callback.callback();
  156. ASSERT_TRUE(mock_factory_->pending_request());
  157. *mock_factory_->pending_request()->resolver =
  158. std::make_unique<FakeProxyResolver>(base::BindOnce(
  159. &ProxyResolverFactoryImplTest::OnFakeProxyInstanceDestroyed,
  160. base::Unretained(this)));
  161. std::move(mock_factory_->pending_request()->callback).Run(net::OK);
  162. EXPECT_THAT(create_callback.WaitForResult(), IsOk());
  163. base::RunLoop wait_for_idle_loop;
  164. set_idle_callback(wait_for_idle_loop.QuitClosure());
  165. proxy_resolver.reset();
  166. waiter_.WaitForEvent(RESOLVER_DESTROYED);
  167. EXPECT_EQ(1, instances_destroyed_);
  168. wait_for_idle_loop.Run();
  169. }
  170. // Same as above, but disconnect the factory right after the CreateResolver
  171. // call, which should not prevent the request from succeeding.
  172. TEST_F(ProxyResolverFactoryImplTest, DisconnectProxyResolverFactory) {
  173. mojo::Remote<mojom::ProxyResolver> proxy_resolver;
  174. mojo::PendingRemote<mojom::ProxyResolverFactoryRequestClient> client;
  175. mojo::Receiver<ProxyResolverFactoryRequestClient> client_receiver(
  176. this, client.InitWithNewPipeAndPassReceiver());
  177. factory_->CreateResolver(kScriptData,
  178. proxy_resolver.BindNewPipeAndPassReceiver(),
  179. std::move(client));
  180. proxy_resolver.set_disconnect_handler(base::BindOnce(
  181. &ProxyResolverFactoryImplTest::OnDisconnect, base::Unretained(this)));
  182. waiter_.WaitForEvent(RESOLVER_CREATED);
  183. EXPECT_EQ(0, instances_destroyed_);
  184. ASSERT_EQ(1u, mock_factory_->requests_handled());
  185. net::TestCompletionCallback create_callback;
  186. create_callback_ = create_callback.callback();
  187. ASSERT_TRUE(mock_factory_->pending_request());
  188. *mock_factory_->pending_request()->resolver =
  189. std::make_unique<FakeProxyResolver>(base::BindOnce(
  190. &ProxyResolverFactoryImplTest::OnFakeProxyInstanceDestroyed,
  191. base::Unretained(this)));
  192. std::move(mock_factory_->pending_request()->callback).Run(net::OK);
  193. EXPECT_THAT(create_callback.WaitForResult(), IsOk());
  194. base::RunLoop wait_for_idle_loop;
  195. set_idle_callback(wait_for_idle_loop.QuitClosure());
  196. proxy_resolver.reset();
  197. waiter_.WaitForEvent(RESOLVER_DESTROYED);
  198. EXPECT_EQ(1, instances_destroyed_);
  199. wait_for_idle_loop.Run();
  200. }
  201. TEST_F(ProxyResolverFactoryImplTest, Error) {
  202. mojo::Remote<mojom::ProxyResolver> proxy_resolver;
  203. mojo::PendingRemote<mojom::ProxyResolverFactoryRequestClient> client;
  204. mojo::Receiver<ProxyResolverFactoryRequestClient> client_receiver(
  205. this, client.InitWithNewPipeAndPassReceiver());
  206. factory_->CreateResolver(kScriptData,
  207. proxy_resolver.BindNewPipeAndPassReceiver(),
  208. std::move(client));
  209. proxy_resolver.set_disconnect_handler(base::BindOnce(
  210. &ProxyResolverFactoryImplTest::OnDisconnect, base::Unretained(this)));
  211. waiter_.WaitForEvent(RESOLVER_CREATED);
  212. EXPECT_EQ(0, instances_destroyed_);
  213. ASSERT_EQ(1u, mock_factory_->requests_handled());
  214. net::TestCompletionCallback create_callback;
  215. create_callback_ = create_callback.callback();
  216. ASSERT_TRUE(mock_factory_->pending_request());
  217. std::move(mock_factory_->pending_request()->callback)
  218. .Run(net::ERR_PAC_SCRIPT_FAILED);
  219. EXPECT_THAT(create_callback.WaitForResult(),
  220. IsError(net::ERR_PAC_SCRIPT_FAILED));
  221. }
  222. TEST_F(ProxyResolverFactoryImplTest, DisconnectClientDuringResolverCreation) {
  223. mojo::Remote<mojom::ProxyResolver> proxy_resolver;
  224. mojo::PendingRemote<mojom::ProxyResolverFactoryRequestClient> client;
  225. mojo::Receiver<ProxyResolverFactoryRequestClient> client_receiver(
  226. this, client.InitWithNewPipeAndPassReceiver());
  227. factory_->CreateResolver(kScriptData,
  228. proxy_resolver.BindNewPipeAndPassReceiver(),
  229. std::move(client));
  230. proxy_resolver.set_disconnect_handler(base::BindOnce(
  231. &ProxyResolverFactoryImplTest::OnDisconnect, base::Unretained(this)));
  232. waiter_.WaitForEvent(RESOLVER_CREATED);
  233. EXPECT_EQ(0, instances_destroyed_);
  234. ASSERT_EQ(1u, mock_factory_->requests_handled());
  235. client_receiver.reset();
  236. waiter_.WaitForEvent(CONNECTION_ERROR);
  237. }
  238. } // namespace proxy_resolver