15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/proxy/network_delegate_error_observer.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind_helpers.h" 9868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/message_loop/message_loop_proxy.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/threading/thread.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_errors.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/network_delegate.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net { 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class TestNetworkDelegate : public net::NetworkDelegate { 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestNetworkDelegate() : got_pac_error_(false) {} 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~TestNetworkDelegate() {} 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool got_pac_error() const { return got_pac_error_; } 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // net::NetworkDelegate implementation. 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int OnBeforeURLRequest(URLRequest* request, 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback, 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GURL* new_url) OVERRIDE { 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return OK; 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int OnBeforeSendHeaders(URLRequest* request, 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback, 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HttpRequestHeaders* headers) OVERRIDE { 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return OK; 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnSendHeaders(URLRequest* request, 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const HttpRequestHeaders& headers) OVERRIDE {} 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int OnHeadersReceived( 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) URLRequest* request, 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback, 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const HttpResponseHeaders* original_response_headers, 44effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch scoped_refptr<HttpResponseHeaders>* override_response_headers, 45effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch GURL* allowed_unsafe_redirect_url) OVERRIDE { 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return net::OK; 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnBeforeRedirect(URLRequest* request, 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const GURL& new_location) OVERRIDE {} 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnResponseStarted(URLRequest* request) OVERRIDE {} 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnRawBytesRead(const URLRequest& request, 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int bytes_read) OVERRIDE {} 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnCompleted(URLRequest* request, bool started) OVERRIDE {} 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnURLRequestDestroyed(URLRequest* request) OVERRIDE {} 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnPACScriptError(int line_number, 57c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16& error) OVERRIDE { 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) got_pac_error_ = true; 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual AuthRequiredResponse OnAuthRequired( 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) URLRequest* request, 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const AuthChallengeInfo& auth_info, 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const AuthCallback& callback, 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AuthCredentials* credentials) OVERRIDE { 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return AUTH_REQUIRED_RESPONSE_NO_ACTION; 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool OnCanGetCookies(const URLRequest& request, 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CookieList& cookie_list) OVERRIDE { 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool OnCanSetCookie(const URLRequest& request, 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& cookie_line, 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CookieOptions* options) OVERRIDE { 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool OnCanAccessFile(const net::URLRequest& request, 772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& path) const OVERRIDE { 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool OnCanThrottleRequest(const URLRequest& request) const OVERRIDE { 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int OnBeforeSocketStreamConnect( 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketStream* stream, 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback) OVERRIDE { 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return OK; 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool got_pac_error_; 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Check that the OnPACScriptError method can be called from an arbitrary 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// thread. 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(NetworkDelegateErrorObserverTest, CallOnThread) { 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Thread thread("test_thread"); 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) thread.Start(); 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestNetworkDelegate network_delegate; 1007d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) NetworkDelegateErrorObserver observer( 1017d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) &network_delegate, base::MessageLoopProxy::current().get()); 1027d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) thread.message_loop() 1037d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) ->PostTask(FROM_HERE, 1047d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) base::Bind(&NetworkDelegateErrorObserver::OnPACScriptError, 1057d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) base::Unretained(&observer), 1067d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 42, 1077d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) base::string16())); 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) thread.Stop(); 10990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoop::current()->RunUntilIdle(); 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(network_delegate.got_pac_error()); 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Check that passing a NULL network delegate works. 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(NetworkDelegateErrorObserverTest, NoDelegate) { 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Thread thread("test_thread"); 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) thread.Start(); 1177d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) NetworkDelegateErrorObserver observer( 1187d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) NULL, base::MessageLoopProxy::current().get()); 1197d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) thread.message_loop() 1207d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) ->PostTask(FROM_HERE, 1217d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) base::Bind(&NetworkDelegateErrorObserver::OnPACScriptError, 1227d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) base::Unretained(&observer), 1237d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 42, 1247d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) base::string16())); 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) thread.Stop(); 12690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoop::current()->RunUntilIdle(); 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Shouldn't have crashed until here... 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 131