| OLD | NEW |
| 1 #!/usr/bin/python | 1 #!/usr/bin/python |
| 2 | 2 |
| 3 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved. | 3 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved. |
| 4 # Use of this source code is governed by a BSD-style license that can be | 4 # Use of this source code is governed by a BSD-style license that can be |
| 5 # found in the LICENSE file. | 5 # found in the LICENSE file. |
| 6 | 6 |
| 7 """Unit tests for autoupdate.py.""" | 7 """Unit tests for autoupdate.py.""" |
| 8 | 8 |
| 9 import cherrypy |
| 9 import mox | 10 import mox |
| 10 import os | 11 import os |
| 11 import socket | 12 import socket |
| 12 import unittest | 13 import unittest |
| 13 | 14 |
| 14 import autoupdate | 15 import autoupdate |
| 15 | 16 |
| 16 _TEST_REQUEST = """ | 17 _TEST_REQUEST = """ |
| 17 <client_test xmlns:o="http://www.google.com/update2/request" updaterversion="%(c
lient)s" > | 18 <client_test xmlns:o="http://www.google.com/update2/request" updaterversion="%(c
lient)s" > |
| 18 <o:app version="%(version)s" track="%(track)s" board="%(board)s" /> | 19 <o:app version="%(version)s" track="%(track)s" board="%(board)s" /> |
| (...skipping 21 matching lines...) Expand all Loading... |
| 40 'track': 'unused_var', | 41 'track': 'unused_var', |
| 41 'board': self.test_board | 42 'board': self.test_board |
| 42 } | 43 } |
| 43 self.test_data = _TEST_REQUEST % self.test_dict | 44 self.test_data = _TEST_REQUEST % self.test_dict |
| 44 self.forced_image_path = '/path_to_force/chromiumos_image.bin' | 45 self.forced_image_path = '/path_to_force/chromiumos_image.bin' |
| 45 self.hash = 12345 | 46 self.hash = 12345 |
| 46 self.size = 54321 | 47 self.size = 54321 |
| 47 self.url = 'http://%s/static/update.gz' % self.hostname | 48 self.url = 'http://%s/static/update.gz' % self.hostname |
| 48 self.payload = 'My payload' | 49 self.payload = 'My payload' |
| 49 self.sha256 = 'SHA LA LA' | 50 self.sha256 = 'SHA LA LA' |
| 51 cherrypy.request.base = 'http://%s' % self.hostname |
| 52 |
| 50 | 53 |
| 51 def _DummyAutoupdateConstructor(self): | 54 def _DummyAutoupdateConstructor(self): |
| 52 """Creates a dummy autoupdater. Used to avoid using constructor.""" | 55 """Creates a dummy autoupdater. Used to avoid using constructor.""" |
| 53 dummy = autoupdate.Autoupdate(root_dir=None, | 56 dummy = autoupdate.Autoupdate(root_dir=None, |
| 54 static_dir=self.static_image_dir, | 57 static_dir=self.static_image_dir, |
| 55 port=self.port) | 58 port=self.port) |
| 56 dummy.client_prefix = 'ChromeOSUpdateEngine' | 59 dummy.client_prefix = 'ChromeOSUpdateEngine' |
| 57 return dummy | 60 return dummy |
| 58 | 61 |
| 59 def testGenerateLatestUpdateImageWithForced(self): | 62 def testGenerateLatestUpdateImageWithForced(self): |
| (...skipping 79 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 139 self.mox.ReplayAll() | 142 self.mox.ReplayAll() |
| 140 au_mock = self._DummyAutoupdateConstructor() | 143 au_mock = self._DummyAutoupdateConstructor() |
| 141 au_mock.serve_only = True | 144 au_mock.serve_only = True |
| 142 au_mock.static_urlbase = 'http://%s/static/archive' % self.hostname | 145 au_mock.static_urlbase = 'http://%s/static/archive' % self.hostname |
| 143 self.assertEqual(au_mock.HandleUpdatePing(test_data), self.payload) | 146 self.assertEqual(au_mock.HandleUpdatePing(test_data), self.payload) |
| 144 self.mox.VerifyAll() | 147 self.mox.VerifyAll() |
| 145 | 148 |
| 146 | 149 |
| 147 if __name__ == '__main__': | 150 if __name__ == '__main__': |
| 148 unittest.main() | 151 unittest.main() |
| OLD | NEW |