| 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 17 matching lines...) Expand all Loading... |
| 28 | 28 |
| 29 import StringIO | 29 import StringIO |
| 30 | 30 |
| 31 | 31 |
| 32 class MockWeb(object): | 32 class MockWeb(object): |
| 33 | 33 |
| 34 def __init__(self, urls=None): | 34 def __init__(self, urls=None): |
| 35 self.urls = urls or {} | 35 self.urls = urls or {} |
| 36 self.urls_fetched = [] | 36 self.urls_fetched = [] |
| 37 | 37 |
| 38 def get_binary(self, url, convert_404_to_None=False): | 38 def get_binary(self, url, return_none_on_404=False): |
| 39 self.urls_fetched.append(url) | 39 self.urls_fetched.append(url) |
| 40 if url in self.urls: | 40 if url in self.urls: |
| 41 return self.urls[url] | 41 return self.urls[url] |
| 42 return "MOCK Web result, convert 404 to None=%s" % convert_404_to_None | 42 return "MOCK Web result, convert 404 to None=%s" % return_none_on_404 |
| 43 | 43 |
| 44 | 44 |
| 45 # 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. |
| 46 class MockBrowser(object): | 46 class MockBrowser(object): |
| 47 params = {} | 47 params = {} |
| 48 | 48 |
| 49 def open(self, url): | 49 def open(self, url): |
| 50 pass | 50 pass |
| 51 | 51 |
| 52 def select_form(self, name): | 52 def select_form(self, name): |
| 53 pass | 53 pass |
| 54 | 54 |
| 55 def __setitem__(self, key, value): | 55 def __setitem__(self, key, value): |
| 56 self.params[key] = value | 56 self.params[key] = value |
| 57 | 57 |
| 58 def submit(self): | 58 def submit(self): |
| 59 return StringIO.StringIO() | 59 return StringIO.StringIO() |
| OLD | NEW |