Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1920)

Unified Diff: autoupdate_unittest.py

Issue 3459002: Revert devserver update to fix factory install. (Closed) Base URL: http://git.chromium.org/git/dev-util.git
Patch Set: Created 10 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « autoupdate.py ('k') | devserver.py » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: autoupdate_unittest.py
diff --git a/autoupdate_unittest.py b/autoupdate_unittest.py
deleted file mode 100755
index d2af440b06282d940459ec0644ad4355a22187c0..0000000000000000000000000000000000000000
--- a/autoupdate_unittest.py
+++ /dev/null
@@ -1,137 +0,0 @@
-#!/usr/bin/python
-
-# Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
-# Use of this source code is governed by a BSD-style license that can be
-# found in the LICENSE file.
-
-"""Unit tests for autoupdate.py."""
-
-import mox
-import os
-import unittest
-import web
-
-import autoupdate
-
-_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" />
-</client_test>"""
-
-
-class AutoupdateTest(mox.MoxTestBase):
- def setUp(self):
- mox.MoxTestBase.setUp(self)
- self.mox.StubOutWithMock(autoupdate.Autoupdate, '_GetSize')
- self.mox.StubOutWithMock(autoupdate.Autoupdate, '_GetHash')
- self.mox.StubOutWithMock(autoupdate.Autoupdate, 'GetUpdatePayload')
- self.mox.StubOutWithMock(autoupdate.Autoupdate, '_GetLatestImageDir')
- self.test_board = 'test-board'
- self.build_root = '/src_path/build/images'
- self.latest_dir = '12345_af_12-a1'
- self.latest_verision = '12345_af_12'
- self.static_image_dir = '/tmp/static-dir/'
- self.hostname = 'fake-host'
- self.test_dict = { 'client': 'ChromeOSUpdateEngine-1.0',
- 'version': 'ForcedUpdate',
- 'track': 'unused_var',
- 'board': self.test_board
- }
- self.test_data = _TEST_REQUEST % self.test_dict
- self.forced_image_path = '/path_to_force/chromiumos_image.bin'
- self.hash = 12345
- self.size = 54321
- self.url = 'http://%s/static/update.gz' % self.hostname
- self.payload = 'My payload'
-
- def _DummyAutoupdateConstructor(self):
- """Creates a dummy autoupdater. Used to avoid using constructor."""
- dummy = autoupdate.Autoupdate(root_dir=None,
- static_dir=self.static_image_dir)
- dummy.client_prefix = 'ChromeOSUpdateEngine'
-
- # Set to fool the web.
- web.ctx.host = self.hostname
- return dummy
-
- def testGenerateLatestUpdateImageWithForced(self):
- self.mox.StubOutWithMock(autoupdate.Autoupdate, 'GenerateUpdateImage')
- autoupdate.Autoupdate._GetLatestImageDir(self.test_board).AndReturn(
- '%s/%s/%s' % (self.build_root, self.test_board, self.latest_dir))
- autoupdate.Autoupdate.GenerateUpdateImage(
- '%s/%s/%s/chromiumos_image.bin' % (self.build_root, self.test_board,
- self.latest_dir),
- move_to_static_dir=True,
- static_image_dir=self.static_image_dir).AndReturn(True)
-
- self.mox.ReplayAll()
- au_mock = self._DummyAutoupdateConstructor()
- self.assertTrue(au_mock.GenerateLatestUpdateImage(self.test_board,
- 'ForcedUpdate',
- self.static_image_dir))
- self.mox.VerifyAll()
-
- def testHandleUpdatePingForForcedImage(self):
- self.mox.StubOutWithMock(autoupdate.Autoupdate, 'GenerateUpdateImage')
-
- test_data = _TEST_REQUEST % self.test_dict
-
- autoupdate.Autoupdate.GenerateUpdateImage(
- self.forced_image_path, self.static_image_dir).AndReturn(True)
- autoupdate.Autoupdate._GetHash(os.path.join(
- self.static_image_dir, 'update.gz')).AndReturn(self.hash)
- 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.mox.ReplayAll()
- au_mock = self._DummyAutoupdateConstructor()
- au_mock.forced_image = self.forced_image_path
- self.assertEqual(au_mock.HandleUpdatePing(test_data), self.payload)
- self.mox.VerifyAll()
-
- def testHandleUpdatePingForLatestImage(self):
- self.mox.StubOutWithMock(autoupdate.Autoupdate, 'GenerateLatestUpdateImage')
-
- test_data = _TEST_REQUEST % self.test_dict
-
- autoupdate.Autoupdate.GenerateLatestUpdateImage(
- 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._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.mox.ReplayAll()
- au_mock = self._DummyAutoupdateConstructor()
- self.assertEqual(au_mock.HandleUpdatePing(test_data), self.payload)
- self.mox.VerifyAll()
-
- def testHandleUpdatePingForArchivedBuild(self):
- self.mox.StubOutWithMock(autoupdate.Autoupdate, 'GenerateImageFromZip')
-
- test_data = _TEST_REQUEST % self.test_dict
-
- autoupdate.Autoupdate.GenerateImageFromZip(
- self.static_image_dir).AndReturn(True)
- autoupdate.Autoupdate._GetHash(os.path.join(
- self.static_image_dir, 'update.gz')).AndReturn(self.hash)
- autoupdate.Autoupdate._GetSize(os.path.join(
- self.static_image_dir, 'update.gz')).AndReturn(self.size)
- autoupdate.Autoupdate.GetUpdatePayload(
- self.hash, 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
- self.assertEqual(au_mock.HandleUpdatePing(test_data), self.payload)
- self.mox.VerifyAll()
-
-
-if __name__ == '__main__':
- unittest.main()
« no previous file with comments | « autoupdate.py ('k') | devserver.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698