blob: 854776163f40399eb5d0292f73265ec71de92023 [file] [log] [blame]
James Robinson646469d2014-10-03 15:33:28 -07001// 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#include "net/proxy/network_delegate_error_observer.h"
6
7#include "base/bind.h"
8#include "base/bind_helpers.h"
9#include "base/message_loop/message_loop_proxy.h"
10#include "base/threading/thread.h"
11#include "net/base/net_errors.h"
12#include "net/base/network_delegate.h"
13#include "testing/gtest/include/gtest/gtest.h"
14
15namespace net {
16
17namespace {
18
19class TestNetworkDelegate : public net::NetworkDelegate {
20 public:
21 TestNetworkDelegate() : got_pac_error_(false) {}
22 virtual ~TestNetworkDelegate() {}
23
24 bool got_pac_error() const { return got_pac_error_; }
25
26 private:
27 // net::NetworkDelegate implementation.
28 virtual int OnBeforeURLRequest(URLRequest* request,
29 const CompletionCallback& callback,
James Robinsone2ac7e82014-10-15 13:21:59 -070030 GURL* new_url) override {
James Robinson646469d2014-10-03 15:33:28 -070031 return OK;
32 }
33 virtual int OnBeforeSendHeaders(URLRequest* request,
34 const CompletionCallback& callback,
James Robinsone2ac7e82014-10-15 13:21:59 -070035 HttpRequestHeaders* headers) override {
James Robinson646469d2014-10-03 15:33:28 -070036 return OK;
37 }
38 virtual void OnSendHeaders(URLRequest* request,
James Robinsone2ac7e82014-10-15 13:21:59 -070039 const HttpRequestHeaders& headers) override {}
James Robinson646469d2014-10-03 15:33:28 -070040 virtual int OnHeadersReceived(
41 URLRequest* request,
42 const CompletionCallback& callback,
43 const HttpResponseHeaders* original_response_headers,
44 scoped_refptr<HttpResponseHeaders>* override_response_headers,
James Robinsone2ac7e82014-10-15 13:21:59 -070045 GURL* allowed_unsafe_redirect_url) override {
James Robinson646469d2014-10-03 15:33:28 -070046 return net::OK;
47 }
48 virtual void OnBeforeRedirect(URLRequest* request,
James Robinsone2ac7e82014-10-15 13:21:59 -070049 const GURL& new_location) override {}
50 virtual void OnResponseStarted(URLRequest* request) override {}
James Robinson646469d2014-10-03 15:33:28 -070051 virtual void OnRawBytesRead(const URLRequest& request,
James Robinsone2ac7e82014-10-15 13:21:59 -070052 int bytes_read) override {}
53 virtual void OnCompleted(URLRequest* request, bool started) override {}
54 virtual void OnURLRequestDestroyed(URLRequest* request) override {}
James Robinson646469d2014-10-03 15:33:28 -070055
56 virtual void OnPACScriptError(int line_number,
James Robinsone2ac7e82014-10-15 13:21:59 -070057 const base::string16& error) override {
James Robinson646469d2014-10-03 15:33:28 -070058 got_pac_error_ = true;
59 }
60 virtual AuthRequiredResponse OnAuthRequired(
61 URLRequest* request,
62 const AuthChallengeInfo& auth_info,
63 const AuthCallback& callback,
James Robinsone2ac7e82014-10-15 13:21:59 -070064 AuthCredentials* credentials) override {
James Robinson646469d2014-10-03 15:33:28 -070065 return AUTH_REQUIRED_RESPONSE_NO_ACTION;
66 }
67 virtual bool OnCanGetCookies(const URLRequest& request,
James Robinsone2ac7e82014-10-15 13:21:59 -070068 const CookieList& cookie_list) override {
James Robinson646469d2014-10-03 15:33:28 -070069 return true;
70 }
71 virtual bool OnCanSetCookie(const URLRequest& request,
72 const std::string& cookie_line,
James Robinsone2ac7e82014-10-15 13:21:59 -070073 CookieOptions* options) override {
James Robinson646469d2014-10-03 15:33:28 -070074 return true;
75 }
76 virtual bool OnCanAccessFile(const net::URLRequest& request,
James Robinsone2ac7e82014-10-15 13:21:59 -070077 const base::FilePath& path) const override {
James Robinson646469d2014-10-03 15:33:28 -070078 return true;
79 }
James Robinsone2ac7e82014-10-15 13:21:59 -070080 virtual bool OnCanThrottleRequest(const URLRequest& request) const override {
James Robinson646469d2014-10-03 15:33:28 -070081 return false;
82 }
83 virtual int OnBeforeSocketStreamConnect(
84 SocketStream* stream,
James Robinsone2ac7e82014-10-15 13:21:59 -070085 const CompletionCallback& callback) override {
James Robinson646469d2014-10-03 15:33:28 -070086 return OK;
87 }
88
89 bool got_pac_error_;
90};
91
92} // namespace
93
94// Check that the OnPACScriptError method can be called from an arbitrary
95// thread.
96TEST(NetworkDelegateErrorObserverTest, CallOnThread) {
97 base::Thread thread("test_thread");
98 thread.Start();
99 TestNetworkDelegate network_delegate;
100 NetworkDelegateErrorObserver observer(
101 &network_delegate, base::MessageLoopProxy::current().get());
102 thread.message_loop()
103 ->PostTask(FROM_HERE,
104 base::Bind(&NetworkDelegateErrorObserver::OnPACScriptError,
105 base::Unretained(&observer),
106 42,
107 base::string16()));
108 thread.Stop();
109 base::MessageLoop::current()->RunUntilIdle();
110 ASSERT_TRUE(network_delegate.got_pac_error());
111}
112
113// Check that passing a NULL network delegate works.
114TEST(NetworkDelegateErrorObserverTest, NoDelegate) {
115 base::Thread thread("test_thread");
116 thread.Start();
117 NetworkDelegateErrorObserver observer(
118 NULL, base::MessageLoopProxy::current().get());
119 thread.message_loop()
120 ->PostTask(FROM_HERE,
121 base::Bind(&NetworkDelegateErrorObserver::OnPACScriptError,
122 base::Unretained(&observer),
123 42,
124 base::string16()));
125 thread.Stop();
126 base::MessageLoop::current()->RunUntilIdle();
127 // Shouldn't have crashed until here...
128}
129
130} // namespace net