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 #include "net/proxy/mock_proxy_script_fetcher.h" 6 7 #include "base/logging.h" 8 #include "base/message_loop/message_loop.h" 9 #include "base/strings/string16.h" 10 #include "base/strings/utf_string_conversions.h" 11 #include "net/base/net_errors.h" 12 13 namespace net { 14 MockProxyScriptFetcher()15MockProxyScriptFetcher::MockProxyScriptFetcher() 16 : pending_request_text_(NULL), 17 waiting_for_fetch_(false) { 18 } 19 ~MockProxyScriptFetcher()20MockProxyScriptFetcher::~MockProxyScriptFetcher() {} 21 22 // ProxyScriptFetcher implementation. Fetch(const GURL & url,base::string16 * text,const CompletionCallback & callback)23int MockProxyScriptFetcher::Fetch(const GURL& url, base::string16* text, 24 const CompletionCallback& callback) { 25 DCHECK(!has_pending_request()); 26 27 // Save the caller's information, and have them wait. 28 pending_request_url_ = url; 29 pending_request_callback_ = callback; 30 pending_request_text_ = text; 31 32 if (waiting_for_fetch_) 33 base::MessageLoop::current()->Quit(); 34 35 return ERR_IO_PENDING; 36 } 37 NotifyFetchCompletion(int result,const std::string & ascii_text)38void MockProxyScriptFetcher::NotifyFetchCompletion( 39 int result, const std::string& ascii_text) { 40 DCHECK(has_pending_request()); 41 *pending_request_text_ = ASCIIToUTF16(ascii_text); 42 CompletionCallback callback = pending_request_callback_; 43 pending_request_callback_.Reset(); 44 callback.Run(result); 45 } 46 Cancel()47void MockProxyScriptFetcher::Cancel() { 48 } 49 GetRequestContext() const50URLRequestContext* MockProxyScriptFetcher::GetRequestContext() const { 51 return NULL; 52 } 53 pending_request_url() const54const GURL& MockProxyScriptFetcher::pending_request_url() const { 55 return pending_request_url_; 56 } 57 has_pending_request() const58bool MockProxyScriptFetcher::has_pending_request() const { 59 return !pending_request_callback_.is_null(); 60 } 61 WaitUntilFetch()62void MockProxyScriptFetcher::WaitUntilFetch() { 63 DCHECK(!has_pending_request()); 64 waiting_for_fetch_ = true; 65 base::MessageLoop::current()->Run(); 66 waiting_for_fetch_ = false; 67 } 68 69 } // namespace net 70