class RemoteDeviceUpdaterTest(cros_test_lib.MockTempDirTestCase):
"""Test the flow of flash.Flash() with RemoteDeviceUpdater."""
IMAGE = '/path/to/image'
DEVICE = commandline.Device(scheme=commandline.DEVICE_SCHEME_SSH,
hostname='1.1.1.1')
def setUp(self):
"""Patches objects."""
self.updater_mock = self.StartPatcher(RemoteDeviceUpdaterMock())
self.PatchObject(dev_server_wrapper,
'GenerateXbuddyRequest',
return_value='xbuddy/local/latest')
self.PatchObject(dev_server_wrapper, 'DevServerWrapper')
self.PatchObject(
dev_server_wrapper,
'GetImagePathWithXbuddy',
return_value=('taco-paladin/R36/chromiumos_test_image.bin',
'remote/taco-paladin/R36/test'))
self.PatchObject(dev_server_wrapper, 'GetUpdatePayloads')
self.PatchObject(remote_access, 'CHECK_INTERVAL', new=0)
self.PatchObject(remote_access, 'ChromiumOSDevice')
def testUpdateAll(self):
"""Tests that update methods are called correctly."""
with mock.patch('os.path.exists', return_value=True):
flash.Flash(self.DEVICE, self.IMAGE)
self.assertTrue(self.updater_mock.patched['UpdateStateful'].called)
self.assertTrue(self.updater_mock.patched['UpdateRootfs'].called)
def testUpdateStateful(self):
"""Tests that update methods are called correctly."""
with mock.patch('os.path.exists', return_value=True):
flash.Flash(self.DEVICE, self.IMAGE, rootfs_update=False)
self.assertTrue(self.updater_mock.patched['UpdateStateful'].called)
self.assertFalse(self.updater_mock.patched['UpdateRootfs'].called)
def testUpdateRootfs(self):
"""Tests that update methods are called correctly."""
with mock.patch('os.path.exists', return_value=True):
flash.Flash(self.DEVICE, self.IMAGE, stateful_update=False)
self.assertFalse(
self.updater_mock.patched['UpdateStateful'].called)
self.assertTrue(self.updater_mock.patched['UpdateRootfs'].called)
def testMissingPayloads(self):
"""Tests we raise FlashError when payloads are missing."""
with mock.patch('os.path.exists', return_value=False):
self.assertRaises(auto_updater.ChromiumOSUpdateError, flash.Flash,
self.DEVICE, self.IMAGE)
class RemoteDeviceUpdaterTest(cros_test_lib.MockTempDirTestCase):
"""Test the flow of flash.Flash() with RemoteDeviceUpdater."""
IMAGE = '/path/to/image'
DEVICE = commandline.Device(scheme=commandline.DEVICE_SCHEME_SSH,
hostname=remote_access.TEST_IP)
def setUp(self):
"""Patches objects."""
self.updater_mock = self.StartPatcher(RemoteDeviceUpdaterMock())
self.PatchObject(dev_server_wrapper,
'GenerateXbuddyRequest',
return_value='xbuddy/local/latest')
self.PatchObject(
dev_server_wrapper,
'GetImagePathWithXbuddy',
return_value=('taco-paladin/R36/chromiumos_test_image.bin',
'remote/taco-paladin/R36/test'))
self.PatchObject(paygen_payload_lib, 'GenerateUpdatePayload')
self.PatchObject(paygen_stateful_payload_lib,
'GenerateStatefulPayload')
self.PatchObject(remote_access, 'CHECK_INTERVAL', new=0)
self.PatchObject(remote_access.ChromiumOSDevice,
'Pingable',
return_value=True)
m = self.StartPatcher(RemoteAccessMock())
m.AddCmdResult(['cat', '/etc/lsb-release'],
stdout='CHROMEOS_RELEASE_BOARD=board')
m.SetDefaultCmdResult()
def testUpdateAll(self):
"""Tests that update methods are called correctly."""
with mock.patch('os.path.exists', return_value=True):
flash.Flash(self.DEVICE, self.IMAGE)
self.assertTrue(self.updater_mock.patched['UpdateStateful'].called)
self.assertTrue(self.updater_mock.patched['UpdateRootfs'].called)
def testUpdateStateful(self):
"""Tests that update methods are called correctly."""
with mock.patch('os.path.exists', return_value=True):
flash.Flash(self.DEVICE, self.IMAGE, rootfs_update=False)
self.assertTrue(self.updater_mock.patched['UpdateStateful'].called)
self.assertFalse(self.updater_mock.patched['UpdateRootfs'].called)
def testUpdateRootfs(self):
"""Tests that update methods are called correctly."""
with mock.patch('os.path.exists', return_value=True):
flash.Flash(self.DEVICE, self.IMAGE, stateful_update=False)
self.assertFalse(
self.updater_mock.patched['UpdateStateful'].called)
self.assertTrue(self.updater_mock.patched['UpdateRootfs'].called)
def testMissingPayloads(self):
"""Tests we raise FlashError when payloads are missing."""
with mock.patch('os.path.exists', return_value=False):
self.assertRaises(auto_updater_transfer.ChromiumOSTransferError,
flash.Flash, self.DEVICE, self.IMAGE)
def testFullPayload(self):
"""Tests that we download full_payload and stateful using xBuddy."""
with mock.patch.object(
dev_server_wrapper,
'GetImagePathWithXbuddy',
return_value=('translated/xbuddy/path',
'resolved/xbuddy/path')) as mock_xbuddy:
with mock.patch('os.path.exists', return_value=True):
flash.Flash(self.DEVICE, self.IMAGE)
# Call to download full_payload and stateful. No other calls.
mock_xbuddy.assert_has_calls([
mock.call('/path/to/image/full_payload',
mock.ANY,
static_dir=flash.DEVSERVER_STATIC_DIR,
silent=True),
mock.call('/path/to/image/stateful',
mock.ANY,
static_dir=flash.DEVSERVER_STATIC_DIR,
silent=True)
])
self.assertEqual(mock_xbuddy.call_count, 2)
def testTestImage(self):
"""Tests that we download the test image when the full payload fails."""
with mock.patch.object(
dev_server_wrapper,
'GetImagePathWithXbuddy',
side_effect=(dev_server_wrapper.ImagePathError,
('translated/xbuddy/path',
'resolved/xbuddy/path'))) as mock_xbuddy:
with mock.patch('os.path.exists', return_value=True):
flash.Flash(self.DEVICE, self.IMAGE)
# Call to download full_payload and image. No other calls.
mock_xbuddy.assert_has_calls([
mock.call('/path/to/image/full_payload',
mock.ANY,
static_dir=flash.DEVSERVER_STATIC_DIR,
silent=True),
mock.call('/path/to/image',
mock.ANY,
static_dir=flash.DEVSERVER_STATIC_DIR)
])
self.assertEqual(mock_xbuddy.call_count, 2)
def Device(self, path):
"""Create a USB device for passing to flash.Flash()."""
return commandline.Device(scheme=commandline.DEVICE_SCHEME_USB,
path=path)
class RemoteDeviceUpdaterTest(cros_test_lib.MockTempDirTestCase):
"""Test the flow of flash.Flash() with RemoteDeviceUpdater."""
IMAGE = '/path/to/image'
DEVICE = commandline.Device(scheme=commandline.DEVICE_SCHEME_SSH,
hostname=remote_access.TEST_IP)
def setUp(self):
"""Patches objects."""
self.updater_mock = self.StartPatcher(RemoteDeviceUpdaterMock())
self.PatchObject(dev_server_wrapper,
'GenerateXbuddyRequest',
return_value='xbuddy/local/latest')
self.PatchObject(
dev_server_wrapper,
'GetImagePathWithXbuddy',
return_value=('taco-paladin/R36/chromiumos_test_image.bin',
'remote/taco-paladin/R36/test'))
self.PatchObject(paygen_payload_lib, 'GenerateUpdatePayload')
self.PatchObject(paygen_stateful_payload_lib,
'GenerateStatefulPayload')
self.PatchObject(remote_access, 'CHECK_INTERVAL', new=0)
self.PatchObject(remote_access.ChromiumOSDevice,
'Pingable',
return_value=True)
m = self.StartPatcher(RemoteAccessMock())
m.AddCmdResult(['cat', '/etc/lsb-release'],
stdout='CHROMEOS_RELEASE_BOARD=board')
m.SetDefaultCmdResult()
def _ExistsMock(self, path, ret=True):
"""Mock function for os.path.exists.
os.path.exists is used a lot; we only want to mock it for devserver/static,
and actually check if the file exists in all other cases (using os.access).
Args:
path: path to check.
ret: return value of mock.
Returns:
ret for paths under devserver/static, and the expected value of
os.path.exists otherwise.
"""
if path.startswith(dev_server_wrapper.DEFAULT_STATIC_DIR):
return ret
return os.access(path, os.F_OK)
def testUpdateAll(self):
"""Tests that update methods are called correctly."""
with mock.patch('os.path.exists', side_effect=self._ExistsMock):
flash.Flash(self.DEVICE, self.IMAGE)
self.assertTrue(self.updater_mock.patched['UpdateStateful'].called)
self.assertTrue(self.updater_mock.patched['UpdateRootfs'].called)
def testUpdateStateful(self):
"""Tests that update methods are called correctly."""
with mock.patch('os.path.exists', side_effect=self._ExistsMock):
flash.Flash(self.DEVICE, self.IMAGE, rootfs_update=False)
self.assertTrue(self.updater_mock.patched['UpdateStateful'].called)
self.assertFalse(self.updater_mock.patched['UpdateRootfs'].called)
def testUpdateRootfs(self):
"""Tests that update methods are called correctly."""
with mock.patch('os.path.exists', side_effect=self._ExistsMock):
flash.Flash(self.DEVICE, self.IMAGE, stateful_update=False)
self.assertFalse(
self.updater_mock.patched['UpdateStateful'].called)
self.assertTrue(self.updater_mock.patched['UpdateRootfs'].called)
def testMissingPayloads(self):
"""Tests we raise FlashError when payloads are missing."""
with mock.patch('os.path.exists',
side_effect=lambda p: self._ExistsMock(p, ret=False)):
self.assertRaises(auto_updater_transfer.ChromiumOSTransferError,
flash.Flash, self.DEVICE, self.IMAGE)
def testFullPayload(self):
"""Tests that we download full_payload and stateful using xBuddy."""
with mock.patch.object(
dev_server_wrapper,
'GetImagePathWithXbuddy',
return_value=('translated/xbuddy/path',
'resolved/xbuddy/path')) as mock_xbuddy:
with mock.patch('os.path.exists', side_effect=self._ExistsMock):
flash.Flash(self.DEVICE, self.IMAGE)
# Call to download full_payload and stateful. No other calls.
mock_xbuddy.assert_has_calls([
mock.call('/path/to/image/full_payload',
'board',
None,
silent=True),
mock.call('/path/to/image/stateful',
'board',
None,
silent=True)
])
self.assertEqual(mock_xbuddy.call_count, 2)
def testTestImage(self):
"""Tests that we download the test image when the full payload fails."""
with mock.patch.object(
dev_server_wrapper,
'GetImagePathWithXbuddy',
side_effect=(dev_server_wrapper.ImagePathError,
('translated/xbuddy/path',
'resolved/xbuddy/path'))) as mock_xbuddy:
with mock.patch('os.path.exists', side_effect=self._ExistsMock):
flash.Flash(self.DEVICE, self.IMAGE)
# Call to download full_payload and image. No other calls.
mock_xbuddy.assert_has_calls([
mock.call('/path/to/image/full_payload',
'board',
None,
silent=True),
mock.call('/path/to/image', 'board', None)
])
self.assertEqual(mock_xbuddy.call_count, 2)
