OLD | NEW |
1 # Copyright (C) 2011 Google Inc. All rights reserved. | 1 # Copyright (C) 2011 Google Inc. All rights reserved. |
2 # | 2 # |
3 # Redistribution and use in source and binary forms, with or without | 3 # Redistribution and use in source and binary forms, with or without |
4 # modification, are permitted provided that the following conditions are | 4 # modification, are permitted provided that the following conditions are |
5 # met: | 5 # met: |
6 # | 6 # |
7 # * Redistributions of source code must retain the above copyright | 7 # * Redistributions of source code must retain the above copyright |
8 # notice, this list of conditions and the following disclaimer. | 8 # notice, this list of conditions and the following disclaimer. |
9 # * Redistributions in binary form must reproduce the above | 9 # * Redistributions in binary form must reproduce the above |
10 # copyright notice, this list of conditions and the following disclaimer | 10 # copyright notice, this list of conditions and the following disclaimer |
(...skipping 12 matching lines...) Expand all Loading... |
23 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | 23 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
24 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | 24 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
25 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | 25 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
26 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | 26 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
27 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | 27 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
28 | 28 |
29 import StringIO | 29 import StringIO |
30 | 30 |
31 | 31 |
32 class MockWeb(object): | 32 class MockWeb(object): |
| 33 |
33 def __init__(self, urls=None): | 34 def __init__(self, urls=None): |
34 self.urls = urls or {} | 35 self.urls = urls or {} |
35 self.urls_fetched = [] | 36 self.urls_fetched = [] |
36 | 37 |
37 def get_binary(self, url, convert_404_to_None=False): | 38 def get_binary(self, url, convert_404_to_None=False): |
38 self.urls_fetched.append(url) | 39 self.urls_fetched.append(url) |
39 if url in self.urls: | 40 if url in self.urls: |
40 return self.urls[url] | 41 return self.urls[url] |
41 return "MOCK Web result, convert 404 to None=%s" % convert_404_to_None | 42 return "MOCK Web result, convert 404 to None=%s" % convert_404_to_None |
42 | 43 |
43 | 44 |
44 # FIXME: Classes which are using Browser probably want to use Web instead. | 45 # FIXME: Classes which are using Browser probably want to use Web instead. |
45 class MockBrowser(object): | 46 class MockBrowser(object): |
46 params = {} | 47 params = {} |
47 | 48 |
48 def open(self, url): | 49 def open(self, url): |
49 pass | 50 pass |
50 | 51 |
51 def select_form(self, name): | 52 def select_form(self, name): |
52 pass | 53 pass |
53 | 54 |
54 def __setitem__(self, key, value): | 55 def __setitem__(self, key, value): |
55 self.params[key] = value | 56 self.params[key] = value |
56 | 57 |
57 def submit(self): | 58 def submit(self): |
58 return StringIO.StringIO() | 59 return StringIO.StringIO() |
OLD | NEW |