commit a387a87f14e02cdfb9ce986149d2eca56a7c4901
author Chris Sosa <sosa@chromium.org> Wed Sep 29 11:51:36 2010 -0700
committer Chris Sosa <sosa@chromium.org> Wed Sep 29 11:51:36 2010 -0700
tree 3ff505c9e2675345a4f2f5a1b27dc7c8bdf52f27
parent 5d342a27fbb2bb73dc78846dcfd15b1cbdf9a39c

Fix unit tests for sha256 and other url changes.

Change-Id: I2b1b7620f55a23db021b68f403ab809b8063e5a7

BUG=
TEST=Ran them.

Review URL: http://codereview.chromium.org/3570001

autoupdate_unittest.py

diff --git a/autoupdate_unittest.py b/autoupdate_unittest.py
index d2af440..688695e 100755
--- a/autoupdate_unittest.py
+++ b/autoupdate_unittest.py
@@ -16,6 +16,7 @@
 _TEST_REQUEST = """
 <client_test xmlns:o="http://www.google.com/update2/request" updaterversion="%(client)s" >
   <o:app version="%(version)s" track="%(track)s" board="%(board)s" />
+  <o:updatecheck />
 </client_test>"""
 
 
@@ -24,6 +25,7 @@
     mox.MoxTestBase.setUp(self)
     self.mox.StubOutWithMock(autoupdate.Autoupdate, '_GetSize')
     self.mox.StubOutWithMock(autoupdate.Autoupdate, '_GetHash')
+    self.mox.StubOutWithMock(autoupdate.Autoupdate, '_GetSHA256')
     self.mox.StubOutWithMock(autoupdate.Autoupdate, 'GetUpdatePayload')
     self.mox.StubOutWithMock(autoupdate.Autoupdate, '_GetLatestImageDir')
     self.test_board = 'test-board'
@@ -43,6 +45,7 @@
     self.size = 54321
     self.url = 'http://%s/static/update.gz' % self.hostname
     self.payload = 'My payload'
+    self.sha256 = 'SHA LA LA'
 
   def _DummyAutoupdateConstructor(self):
     """Creates a dummy autoupdater.  Used to avoid using constructor."""
@@ -77,13 +80,17 @@
     test_data = _TEST_REQUEST % self.test_dict
 
     autoupdate.Autoupdate.GenerateUpdateImage(
-        self.forced_image_path, self.static_image_dir).AndReturn(True)
+        self.forced_image_path,
+        move_to_static_dir=True,
+        static_image_dir=self.static_image_dir).AndReturn(True)
     autoupdate.Autoupdate._GetHash(os.path.join(
         self.static_image_dir, 'update.gz')).AndReturn(self.hash)
+    autoupdate.Autoupdate._GetSHA256(os.path.join(
+        self.static_image_dir, 'update.gz')).AndReturn(self.sha256)
     autoupdate.Autoupdate._GetSize(os.path.join(
         self.static_image_dir, 'update.gz')).AndReturn(self.size)
     autoupdate.Autoupdate.GetUpdatePayload(
-        self.hash, self.size, self.url).AndReturn(self.payload)
+        self.hash, self.sha256, self.size, self.url).AndReturn(self.payload)
 
     self.mox.ReplayAll()
     au_mock = self._DummyAutoupdateConstructor()
@@ -100,10 +107,12 @@
         self.test_board, 'ForcedUpdate', self.static_image_dir).AndReturn(True)
     autoupdate.Autoupdate._GetHash(os.path.join(
         self.static_image_dir, 'update.gz')).AndReturn(self.hash)
+    autoupdate.Autoupdate._GetSHA256(os.path.join(
+        self.static_image_dir, 'update.gz')).AndReturn(self.sha256)
     autoupdate.Autoupdate._GetSize(os.path.join(
         self.static_image_dir, 'update.gz')).AndReturn(self.size)
     autoupdate.Autoupdate.GetUpdatePayload(
-        self.hash, self.size, self.url).AndReturn(self.payload)
+        self.hash, self.sha256, self.size, self.url).AndReturn(self.payload)
 
     self.mox.ReplayAll()
     au_mock = self._DummyAutoupdateConstructor()
@@ -119,16 +128,19 @@
         self.static_image_dir).AndReturn(True)
     autoupdate.Autoupdate._GetHash(os.path.join(
         self.static_image_dir, 'update.gz')).AndReturn(self.hash)
+    autoupdate.Autoupdate._GetSHA256(os.path.join(
+        self.static_image_dir, 'update.gz')).AndReturn(self.sha256)
     autoupdate.Autoupdate._GetSize(os.path.join(
         self.static_image_dir, 'update.gz')).AndReturn(self.size)
     autoupdate.Autoupdate.GetUpdatePayload(
-        self.hash, self.size,
+        self.hash, self.sha256, self.size,
         'http://%s/static/archive/update.gz' % self.hostname).AndReturn(
             self.payload)
 
     self.mox.ReplayAll()
     au_mock = self._DummyAutoupdateConstructor()
     au_mock.serve_only = True
+    au_mock.static_urlbase = 'http://%s/static/archive' % self.hostname
     self.assertEqual(au_mock.HandleUpdatePing(test_data), self.payload)
     self.mox.VerifyAll()