def test_capture_release_version_from_default(self, mock_storage,
mock_abstract_storage):
# WHEN api_version is not available from either CLI or the driver.
mock_abstract_storage.driver = BaseDriver(key=Mock())
mock_storage.attach_mock(mock_abstract_storage, 'storage_driver')
restore_node.capture_release_version(mock_storage, None)
# THEN expect default release version will be captured.
self.assertEqual(HostMan.get_release_version(),
Version(HostMan.DEFAULT_RELEASE_VERSION))
def test_capture_release_version_from_driver(self, mock_storage,
mock_abstract_storage):
# WHEN api_version from driver is specified and no specified CLI version.
mock_abstract_storage.driver = BaseDriver(key=Mock(),
api_version='5.0.0')
mock_storage.attach_mock(mock_abstract_storage, 'storage_driver')
restore_node.capture_release_version(mock_storage, None)
# THEN expect the api_version from driver is captured.
self.assertEqual(HostMan.get_release_version(), Version('5.0.0'))
def test_capture_release_version_from_CLI(self, mock_storage,
mock_abstract_storage):
# WHEN api_version is specified from CLI, and api_version is available from driver.
mock_abstract_storage.driver = BaseDriver(key=Mock(),
api_version='5.0.0')
mock_storage.attach_mock(mock_abstract_storage, 'storage_driver')
restore_node.capture_release_version(mock_storage, '4.0.0')
# THEN expect the api_version from driver is captured.
self.assertEqual(HostMan.get_release_version(), Version('4.0.0'))
def test_get_release_version(self):
rv_1 = "1.2.3"
rv_2 = "5.6.7"
rv_3 = "10.11.12"
HostMan.set_release_version(rv_1)
HostMan.set_release_version(rv_2)
HostMan.set_release_version(rv_3)
self.assertEqual(HostMan.get_release_version(), Version(rv_3))
def is_node_up(config, host):
"""
If not configured with ccm, proceed with direct health check for Cassandra.
"""
try:
check_type = config.checks.health_check
logging.info('Verifying node state for host {} using check type {}'.format(host, check_type))
if int(config.cassandra.is_ccm) == 1:
logging.debug('Checking ccm health')
return is_ccm_healthy(check_type)
return is_cassandra_healthy(check_type, Cassandra(config, release_version=HostMan.get_release_version()),
host)
except Exception as e:
err_msg = 'Unable to determine if node is up for host: {}'.format(host)
logging.debug(err_msg, exc_info=e)
return False
def test_get_release_version_not_set(self):
with self.assertRaises(RuntimeError):
HostMan.get_release_version()