Add LoadState reporting to the mojo proxy resolver.

net/proxy/mojo_proxy_resolver_impl_unittest.cc

 // Copyright 2015 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/proxy/mojo_proxy_resolver_impl.h"
 
 #include <string>
 #include <vector>
 
 #include "base/run_loop.h"
@@ skipping 10 matching lines @@
 namespace net {
 namespace {
 
 class TestRequestClient : public interfaces::ProxyResolverRequestClient,
                           public mojo::ErrorHandler {
  public:
   explicit TestRequestClient(
       mojo::InterfaceRequest<interfaces::ProxyResolverRequestClient> request);
 
   void WaitForResult();
+  void WaitForLoadStateChange();
   void WaitForConnectionError();
 
   Error error() { return error_; }
   const mojo::Array<interfaces::ProxyServerPtr>& results() { return results_; }
+  LoadState load_state() { return load_state_; }
 
  private:
   // interfaces::ProxyResolverRequestClient override.
   void ReportResult(int32_t error,
                     mojo::Array<interfaces::ProxyServerPtr> results) override;
+  void LoadStateChanged(int32_t load_state) override;
 
   // mojo::ErrorHandler override.
   void OnConnectionError() override;
 
   bool done_ = false;
   bool encountered_connection_error_ = false;
   Error error_ = ERR_FAILED;
+  LoadState load_state_ = LOAD_STATE_IDLE;
   mojo::Array<interfaces::ProxyServerPtr> results_;
   base::Closure run_loop_quit_closure_;
   base::Closure connection_error_callback_;
 
   mojo::Binding<interfaces::ProxyResolverRequestClient> binding_;
 };
 
 TestRequestClient::TestRequestClient(
     mojo::InterfaceRequest<interfaces::ProxyResolverRequestClient> request)
     : binding_(this, request.Pass()) {
   binding_.set_error_handler(this);
 }
 
 void TestRequestClient::WaitForResult() {
   if (done_)
     return;
 
   base::RunLoop run_loop;
   run_loop_quit_closure_ = run_loop.QuitClosure();
   run_loop.Run();
   ASSERT_TRUE(done_);
 }
 
+void TestRequestClient::WaitForLoadStateChange() {

[Anand Mistry (off Chromium), 2015/03/10 03:36:09]

Switch to net::EventWaiter. You are, after all, waiting for a specific load
state.

[Sam McNally, 2015/03/10 06:47:35]

Done.

+  base::RunLoop run_loop;
+  run_loop_quit_closure_ = run_loop.QuitClosure();
+  run_loop.Run();
+}
+
 void TestRequestClient::WaitForConnectionError() {
   if (encountered_connection_error_)
     return;
 
   base::RunLoop run_loop;
   connection_error_callback_ = run_loop.QuitClosure();
   run_loop.Run();
   ASSERT_TRUE(encountered_connection_error_);
 }
 
 void TestRequestClient::ReportResult(
     int32_t error,
     mojo::Array<interfaces::ProxyServerPtr> results) {
   if (!run_loop_quit_closure_.is_null()) {
     run_loop_quit_closure_.Run();
   }
   ASSERT_FALSE(done_);
   error_ = static_cast<Error>(error);
   results_ = results.Pass();
   done_ = true;
 }
 
+void TestRequestClient::LoadStateChanged(int32_t load_state) {
+  load_state_ = static_cast<LoadState>(load_state);
+  if (!run_loop_quit_closure_.is_null())
+    run_loop_quit_closure_.Run();
+}
+
 void TestRequestClient::OnConnectionError() {
   if (!connection_error_callback_.is_null())
     connection_error_callback_.Run();
   encountered_connection_error_ = true;
 }
 
 class SetPacScriptClient {
  public:
   base::Callback<void(int32_t)> CreateCallback();
   Error error() { return error_; }
@@ skipping 102 matching lines @@
 }
 
 }  // namespace
 
 class MojoProxyResolverImplTest : public testing::Test {
  protected:
   void SetUp() override {
     scoped_ptr<CallbackMockProxyResolver> mock_resolver(
         new CallbackMockProxyResolver);
     mock_proxy_resolver_ = mock_resolver.get();
-    resolver_.reset(new MojoProxyResolverImpl(mock_resolver.Pass()));
+    resolver_impl_.reset(new MojoProxyResolverImpl(mock_resolver.Pass()));
+    resolver_ = resolver_impl_.get();
   }
 
   CallbackMockProxyResolver* mock_proxy_resolver_;
 
-  scoped_ptr<interfaces::ProxyResolver> resolver_;
+  scoped_ptr<MojoProxyResolverImpl> resolver_impl_;
+  interfaces::ProxyResolver* resolver_;
 };
 
 TEST_F(MojoProxyResolverImplTest, GetProxyForUrl) {
   interfaces::ProxyResolverRequestClientPtr client_ptr;
   TestRequestClient client(mojo::GetProxy(&client_ptr));
 
   resolver_->GetProxyForUrl("http://example.com", client_ptr.Pass());
   ASSERT_EQ(1u, mock_proxy_resolver_->pending_requests().size());
   scoped_refptr<MockAsyncProxyResolverBase::Request> request =
       mock_proxy_resolver_->pending_requests()[0];
   EXPECT_EQ(GURL("http://example.com"), request->url());
+
+  resolver_impl_->LoadStateChanged(request.get(),
+                                   LOAD_STATE_RESOLVING_HOST_IN_PROXY_SCRIPT);
+  client.WaitForLoadStateChange();
+  EXPECT_EQ(LOAD_STATE_RESOLVING_HOST_IN_PROXY_SCRIPT, client.load_state());
+
   request->results()->UsePacString(
       "PROXY proxy.example.com:1; "
       "SOCKS4 socks4.example.com:2; "
       "SOCKS5 socks5.example.com:3; "
       "HTTPS https.example.com:4; "
       "QUIC quic.example.com:65000; "
       "DIRECT");
   request->CompleteNow(OK);
   client.WaitForResult();
 
@@ skipping 161 matching lines @@
 
 TEST_F(MojoProxyResolverImplTest, DestroyService) {
   interfaces::ProxyResolverRequestClientPtr client_ptr;
   TestRequestClient client(mojo::GetProxy(&client_ptr));
 
   resolver_->GetProxyForUrl("http://example.com", client_ptr.Pass());
   resolver_->SetPacScript("pac script", base::Bind(&Fail));
   ASSERT_EQ(1u, mock_proxy_resolver_->pending_requests().size());
   scoped_refptr<MockAsyncProxyResolverBase::Request> request =
       mock_proxy_resolver_->pending_requests()[0];
-  resolver_.reset();
+  resolver_impl_.reset();
   client.WaitForConnectionError();
 }
 
 }  // namespace net