def test_constructor_inputArgs_populate_manager_attributes(self):
c = MagicMock()
m = update.UpdateManager(card=c)
self.assertEqual(m._card, c)
s = MagicMock()
m = update.UpdateManager(statusUpdater=s)
self.assertEqual(m._statusUpdater, s)
def test_extract_generatesFileForEachItem(self, mockOpen):
card = MagicMock()
m = update.UpdateManager(card=card)
item1 = generateTarItemBin(name="mydir1/")
item2 = generateTarItemBin(name="myfile2", content=b'content 2')
item3 = generateTarItemBin(name="myfile3", content=b'content 3')
emptyBlock = generateTarItemBin()
f = io.BytesIO(item1 + item2 + item3 + emptyBlock)
mockOpen.return_value = f
m._writeTarItemToFile = MagicMock()
m._writeTarItemAsDir = MagicMock()
m.extract("myfile")
mockOpen.assert_called_once_with("myfile", "rb")
self.assertEqual(m._writeTarItemAsDir.call_args[0][0].Name, "mydir1/")
args = m._writeTarItemToFile.call_args_list
firstCallArg = args[0][0][0]
secondCallArg = args[1][0][0]
self.assertEqual(firstCallArg.Name, "myfile2")
self.assertEqual(secondCallArg.Name, "myfile3")
def test__writeTarItemAsDir_calls_os_to_make_directory(self, mockMkDir):
card = MagicMock()
m = update.UpdateManager(card=card)
i = generateTarItem("myItem", b'hello content')
m._writeTarItemAsDir(i)
mockMkDir.assert_called_once_with("myItem")
def test_migrateAndInstall_noUpdateAvailable_provides_status_message(self):
card = MagicMock()
s = MagicMock()
m = update.UpdateManager(card=card, statusUpdater=s)
m._dfu = MagicMock() # type: ignore
m._dfu.isUpdateAvailable.return_value = False
m.migrateAndInstall()
s.assert_called_once_with("No update available", None)
def test_gather_getInfo_returns_none_raise_exception(self):
card = MagicMock()
m = update.UpdateManager(card=card)
m._dfu = MagicMock() # type: ignore
m._dfu.getUpdateInfo.return_value = None
with self.assertRaises(Exception) as context: # type: ignore
m.gather()
self.assertEqual(str(context.exception), 'failed to get update info')
def test__writeTarItemToFile_multipleTarItemBlocks(self, mockOpen):
card = MagicMock()
m = update.UpdateManager(card=card)
data = bytearray([1] * (utar.BLOCK_LENGTH_BYTES + 1))
item = generateTarItem(name="myfile", content=data)
m._writeTarItemToFile(item)
mockOpen.assert_called_once_with("myfile", "wb")
mockWrite = mockOpen.return_value.__enter__().write
mockWrite.assert_any_call(data[0:utar.BLOCK_LENGTH_BYTES])
mockWrite.assert_any_call(data[utar.BLOCK_LENGTH_BYTES:])
def test_migrateAndInstall_restart_disabledByDefault(self):
card = MagicMock()
s = MagicMock()
m = update.UpdateManager(card=card, statusUpdater=s)
m._dfu = MagicMock() # type: ignore
m._dfu.isUpdateAvailable.return_value = True
m.gather = MagicMock(return_value="myupdatefile")
m.extract = MagicMock()
m.restart = MagicMock()
m.migrateAndInstall()
m.restart.assert_not_called()
def test_migrateAndInstall_gatherFails_provides_status_message(self):
card = MagicMock()
s = MagicMock()
m = update.UpdateManager(card=card, statusUpdater=s)
m._dfu = MagicMock() # type: ignore
m._dfu.isUpdateAvailable.return_value = True
m.gather = MagicMock(side_effect=FileExistsError)
m.migrateAndInstall()
s.assert_any_call("migrate file")
s.assert_any_call("failed to migrate file")
m._dfu.setUpdateError.assert_called_once_with(
card, "failed to migrate file")
def test_migrateAndInstall_restart_enabled(self):
card = MagicMock()
s = MagicMock()
m = update.UpdateManager(card=card, statusUpdater=s)
m._dfu = MagicMock() # type: ignore
m._dfu.isUpdateAvailable.return_value = True
m.gather = MagicMock(return_value="myupdatefile")
m.extract = MagicMock()
m.restart = MagicMock()
m.migrateAndInstall(restart=True)
m.restart.assert_called_once()
s.assert_any_call("restart")
def test_gather_update_file_open_fails(self, mockOpen):
card = MagicMock()
s = MagicMock()
m = update.UpdateManager(card=card, statusUpdater=s)
m._dfu = MagicMock() # type: ignore
m._dfu.isUpdateAvailable.return_value = True
fileName = "myfile.tar"
m._dfu.getUpdateInfo.return_value = {"source": fileName}
mockOpen.side_effect = FileExistsError
f = m.gather()
self.assertEqual(f, "")
s.assert_any_call("Failed to migrate update", None)
m._dfu.setUpdateError.assert_called_with(
card, "failed to copy image to file")
def test_gather_updateAvailable_copiesImageToFile_providesStatusMessages(
self, mockOpen):
card = MagicMock()
s = MagicMock()
m = update.UpdateManager(card=card, statusUpdater=s)
m._dfu = MagicMock() # type: ignore
m._dfu.isUpdateAvailable.return_value = True
fileName = "myfile.tar"
m._dfu.getUpdateInfo.return_value = {"source": fileName}
f = m.gather()
mockOpen.assert_called_once_with(fileName, "wb")
args = m._dfu.copyImageToWriter.call_args
self.assertEqual(args[0][0], card)
p = args[1]['progressUpdater']
percentComplete = 13
p(percentComplete)
s.assert_called_with("Migrating", percentComplete)
s.assert_any_call("Successful copy of update", None)
self.assertEqual(f, fileName)
def test_reset_generates_error_by_default(self):
card = MagicMock()
m = update.UpdateManager(card=card)
self.assertRaises(NotImplementedError, m.restart)