1c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// Copyright (c) 2010 The Chromium Authors. All rights reserved. 2c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// Use of this source code is governed by a BSD-style license that can be 3c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// found in the LICENSE file. 4c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 521d179b334e59e9a3bfcaed4c4430bef1bc5759dKristian Monsen// A net::URLRequestJob class that simulates network errors (including https 6c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// related). 7c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// It is based on URLRequestMockHttpJob. 8c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 9c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#ifndef CHROME_BROWSER_NET_URL_REQUEST_MOCK_NET_ERROR_JOB_H_ 10c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#define CHROME_BROWSER_NET_URL_REQUEST_MOCK_NET_ERROR_JOB_H_ 113345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick#pragma once 12c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 13c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "chrome/browser/net/url_request_mock_http_job.h" 14c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 15c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochclass URLRequestMockNetErrorJob : public URLRequestMockHTTPJob { 16c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch public: 1721d179b334e59e9a3bfcaed4c4430bef1bc5759dKristian Monsen URLRequestMockNetErrorJob(net::URLRequest* request, 18c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch const std::vector<int>& errors, 19c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch net::X509Certificate* ssl_cert, 20c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch const FilePath& file_path); 21c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 22c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch virtual void Start(); 23c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch virtual void ContinueDespiteLastError(); 24c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 25c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // Add the specified URL to the list of URLs that should be mocked. When this 26c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // URL is hit, the specified |errors| will be played. If any of these errors 27c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // is a cert error, |ssl_cert| will be used as the ssl cert when notifying of 28c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // the error. |ssl_cert| can be NULL if |errors| does not contain any cert 29c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // errors. |base| is the location on disk where the file mocking the URL 30c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // contents and http-headers should be retrieved from. 31c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch static void AddMockedURL(const GURL& url, 32c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch const std::wstring& base, 33c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch const std::vector<int>& errors, 34c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch net::X509Certificate* ssl_cert); 35c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 36c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // Removes the specified |url| from the list of mocked urls. 37c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch static void RemoveMockedURL(const GURL& url); 38c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 39c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch private: 40c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch ~URLRequestMockNetErrorJob(); 41c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 4221d179b334e59e9a3bfcaed4c4430bef1bc5759dKristian Monsen static net::URLRequest::ProtocolFactory Factory; 43c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 44c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch void StartAsync(); 45c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 46c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // The errors to simulate. 47c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::vector<int> errors_; 48c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 49c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // The certificate to use for SSL errors. 50c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch scoped_refptr<net::X509Certificate> ssl_cert_; 51c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 52731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick struct MockInfo; 53c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch typedef std::map<GURL, MockInfo> URLMockInfoMap; 54c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch static URLMockInfoMap url_mock_info_map_; 55c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 56c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch DISALLOW_COPY_AND_ASSIGN(URLRequestMockNetErrorJob); 57c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch}; 58c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 59c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#endif // CHROME_BROWSER_NET_URL_REQUEST_MOCK_NET_ERROR_JOB_H_ 60