def __init__(self):
self.firmware_manager = firmware_manager_factory.get_firmware_updater()
self._required_version = None
self._firmware_update = FirmwareUpdate(self._bin_file(),
self.firmware_manager)
def test_exception_raised_if_not_supported(self, mock_sys):
mock_sys.platform = 'sun'
with self.assertRaises(Exception):
firmware.get_firmware_updater()
def test_correct_class_for_win64_platform_is_provided(self, mock_sys):
mock_sys.platform = 'winamd64'
result = firmware.get_firmware_updater()
self.assertEquals(WindowsFirmwareUpdater, type(result))
def test_correct_class_for_linux_platform_is_provided(self, mock_sys):
mock_sys.platform = 'linux'
result = firmware.get_firmware_updater()
self.assertEquals(LinuxFirmwareUpdater, type(result))
def test_correct_class_for_mac_platform_is_provided(self, mock_sys):
mock_sys.platform = 'darwin'
result = firmware.get_firmware_updater()
self.assertEquals(MacFirmwareUpdater, type(result))
