class HostUpgradeTestCase(unittest.TestCase):
TIMEOUT = 6000
def setUp(self):
self.datastore_manager = MagicMock()
self.datastore_manager.get_datastore_ids.return_value = [
"datastore1", "datastore2"
]
self.host_upgrade = HostUpgrade(self.datastore_manager)
self.timeout = self.TIMEOUT
def tearDown(self):
self.host_upgrade._task_runner.stop()
@patch("host.upgrade.softlink_generator.SoftLinkGenerator.process")
def test_lifecycle(self, process):
assert_that(not self.host_upgrade.in_progress())
self.host_upgrade.start(self.timeout)
self.host_upgrade._task_runner.wait_for_task_end()
assert_that(not self.host_upgrade.in_progress())
self.assertEqual(process.call_count, 1)
@patch("host.upgrade.softlink_generator.SoftLinkGenerator.process")
def test_upgrade_exception(self, process):
process.side_effect = Exception("create_symlinks failed.")
assert_that(not self.host_upgrade.in_progress())
self.host_upgrade.start(self.timeout)
self.host_upgrade._task_runner.wait_for_task_end()
assert_that(not self.host_upgrade.in_progress())
self.assertEqual(process.call_count, 1)
class HostUpgradeTestCase(unittest.TestCase):
TIMEOUT = 6000
def setUp(self):
self.datastore_manager = MagicMock()
self.datastore_manager.get_datastore_ids.return_value = ["datastore1", "datastore2"]
self.host_upgrade = HostUpgrade(self.datastore_manager)
self.timeout = self.TIMEOUT
def tearDown(self):
self.host_upgrade._task_runner.stop()
def test_lifecycle(self):
assert_that(not self.host_upgrade.in_progress())
self.host_upgrade.start(self.timeout)
self.host_upgrade._task_runner.wait_for_task_end()
assert_that(not self.host_upgrade.in_progress())
