1// Copyright (c) 2012 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
5#ifndef NET_PROXY_MOCK_PROXY_RESOLVER_H_
6#define NET_PROXY_MOCK_PROXY_RESOLVER_H_
7
8#include <vector>
9
10#include "base/memory/scoped_ptr.h"
11#include "net/base/net_errors.h"
12#include "net/proxy/proxy_resolver.h"
13#include "url/gurl.h"
14
15namespace base {
16class MessageLoop;
17}
18
19namespace net {
20
21// Asynchronous mock proxy resolver. All requests complete asynchronously,
22// user must call Request::CompleteNow() on a pending request to signal it.
23class MockAsyncProxyResolverBase : public ProxyResolver {
24 public:
25  class Request : public base::RefCounted<Request> {
26   public:
27    Request(MockAsyncProxyResolverBase* resolver,
28            const GURL& url,
29            ProxyInfo* results,
30            const net::CompletionCallback& callback);
31
32    const GURL& url() const { return url_; }
33    ProxyInfo* results() const { return results_; }
34    const net::CompletionCallback& callback() const { return callback_; }
35
36    void CompleteNow(int rv);
37
38   private:
39    friend class base::RefCounted<Request>;
40
41    virtual ~Request();
42
43    MockAsyncProxyResolverBase* resolver_;
44    const GURL url_;
45    ProxyInfo* results_;
46    net::CompletionCallback callback_;
47    base::MessageLoop* origin_loop_;
48  };
49
50  class SetPacScriptRequest {
51   public:
52    SetPacScriptRequest(
53        MockAsyncProxyResolverBase* resolver,
54        const scoped_refptr<ProxyResolverScriptData>& script_data,
55        const net::CompletionCallback& callback);
56    ~SetPacScriptRequest();
57
58    const ProxyResolverScriptData* script_data() const {
59      return script_data_.get();
60    }
61
62    void CompleteNow(int rv);
63
64   private:
65    MockAsyncProxyResolverBase* resolver_;
66    const scoped_refptr<ProxyResolverScriptData> script_data_;
67    net::CompletionCallback callback_;
68    base::MessageLoop* origin_loop_;
69  };
70
71  typedef std::vector<scoped_refptr<Request> > RequestsList;
72
73  virtual ~MockAsyncProxyResolverBase();
74
75  // ProxyResolver implementation.
76  virtual int GetProxyForURL(const GURL& url,
77                             ProxyInfo* results,
78                             const net::CompletionCallback& callback,
79                             RequestHandle* request_handle,
80                             const BoundNetLog& /*net_log*/) OVERRIDE;
81  virtual void CancelRequest(RequestHandle request_handle) OVERRIDE;
82  virtual LoadState GetLoadState(RequestHandle request_handle) const OVERRIDE;
83  virtual int SetPacScript(
84      const scoped_refptr<ProxyResolverScriptData>& script_data,
85      const net::CompletionCallback& callback) OVERRIDE;
86  virtual void CancelSetPacScript() OVERRIDE;
87
88  const RequestsList& pending_requests() const {
89    return pending_requests_;
90  }
91
92  const RequestsList& cancelled_requests() const {
93    return cancelled_requests_;
94  }
95
96  SetPacScriptRequest* pending_set_pac_script_request() const;
97
98  bool has_pending_set_pac_script_request() const {
99    return pending_set_pac_script_request_.get() != NULL;
100  }
101
102  void RemovePendingRequest(Request* request);
103
104  void RemovePendingSetPacScriptRequest(SetPacScriptRequest* request);
105
106 protected:
107  explicit MockAsyncProxyResolverBase(bool expects_pac_bytes);
108
109 private:
110  RequestsList pending_requests_;
111  RequestsList cancelled_requests_;
112  scoped_ptr<SetPacScriptRequest> pending_set_pac_script_request_;
113};
114
115class MockAsyncProxyResolver : public MockAsyncProxyResolverBase {
116 public:
117  MockAsyncProxyResolver()
118      : MockAsyncProxyResolverBase(false /*expects_pac_bytes*/) {}
119};
120
121class MockAsyncProxyResolverExpectsBytes : public MockAsyncProxyResolverBase {
122 public:
123  MockAsyncProxyResolverExpectsBytes()
124      : MockAsyncProxyResolverBase(true /*expects_pac_bytes*/) {}
125};
126
127}  // namespace net
128
129#endif  // NET_PROXY_MOCK_PROXY_RESOLVER_H_
130