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 cherrypy |
10 import mox | 10 import mox |
(...skipping 106 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
117 self.static_image_dir, 'update.gz')).AndReturn(self.size) | 117 self.static_image_dir, 'update.gz')).AndReturn(self.size) |
118 autoupdate.Autoupdate.GetUpdatePayload( | 118 autoupdate.Autoupdate.GetUpdatePayload( |
119 self.hash, self.sha256, self.size, self.url, False).AndReturn( | 119 self.hash, self.sha256, self.size, self.url, False).AndReturn( |
120 self.payload) | 120 self.payload) |
121 | 121 |
122 self.mox.ReplayAll() | 122 self.mox.ReplayAll() |
123 au_mock = self._DummyAutoupdateConstructor() | 123 au_mock = self._DummyAutoupdateConstructor() |
124 self.assertEqual(au_mock.HandleUpdatePing(test_data), self.payload) | 124 self.assertEqual(au_mock.HandleUpdatePing(test_data), self.payload) |
125 self.mox.VerifyAll() | 125 self.mox.VerifyAll() |
126 | 126 |
127 def testHandleUpdatePingForArchivedBuild(self): | |
128 self.mox.StubOutWithMock(autoupdate.Autoupdate, 'GenerateImageFromZip') | |
129 | |
130 test_data = _TEST_REQUEST % self.test_dict | |
131 | |
132 autoupdate.Autoupdate.GenerateImageFromZip( | |
133 self.static_image_dir).AndReturn('update.gz') | |
134 autoupdate.Autoupdate._GetHash(os.path.join( | |
135 self.static_image_dir, 'update.gz')).AndReturn(self.hash) | |
136 autoupdate.Autoupdate._GetSHA256(os.path.join( | |
137 self.static_image_dir, 'update.gz')).AndReturn(self.sha256) | |
138 autoupdate.Autoupdate._GetSize(os.path.join( | |
139 self.static_image_dir, 'update.gz')).AndReturn(self.size) | |
140 autoupdate.Autoupdate.GetUpdatePayload( | |
141 self.hash, self.sha256, self.size, | |
142 'http://%s/static/archive/update.gz' % self.hostname, | |
143 False).AndReturn( | |
144 self.payload) | |
145 | |
146 self.mox.ReplayAll() | |
147 au_mock = self._DummyAutoupdateConstructor() | |
148 au_mock.serve_only = True | |
149 au_mock.static_urlbase = 'http://%s/static/archive' % self.hostname | |
150 self.assertEqual(au_mock.HandleUpdatePing(test_data), self.payload) | |
151 self.mox.VerifyAll() | |
152 | |
153 | 127 |
154 if __name__ == '__main__': | 128 if __name__ == '__main__': |
155 unittest.main() | 129 unittest.main() |
OLD | NEW |