| 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 22 matching lines...) Expand all Loading... |
| 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, return_none_on_404=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" % return_none_on_404 | 42 return "MOCK Web result, convert 404 to None=%s" % return_none_on_404 |
| 43 | |
| 44 | |
| 45 # FIXME: Classes which are using Browser probably want to use Web instead. | |
| 46 class MockBrowser(object): | |
| 47 params = {} | |
| 48 | |
| 49 def open(self, url): | |
| 50 pass | |
| 51 | |
| 52 def select_form(self, name): | |
| 53 pass | |
| 54 | |
| 55 def __setitem__(self, key, value): | |
| 56 self.params[key] = value | |
| 57 | |
| 58 def submit(self): | |
| 59 return StringIO.StringIO() | |
| OLD | NEW |