def test_populate_hostmap(self, resolver_mock):
# target and source lookups mocked
resolver_mock.side_effect = [
'node1.mydomain.net', 'node1.mydomain.net', 'node2.mydomain.net',
'node2.mydomain.net', 'node3.mydomain.net', 'node4.mydomain.net'
]
node_backups = list()
node_backups.append(Mock())
with open("tests/resources/restore_cluster_tokenmap.json", 'r') as f:
cluster_backup = Mock()
tokenmap = json.loads(f.read())
cluster_backup.tokenmap.return_value = tokenmap
host_list = "tests/resources/restore_cluster_host_list.txt"
restore_job = RestoreJob(cluster_backup,
self.medusa_config,
self.tmp_dir,
host_list,
None,
False,
False,
None,
version_target="4.0.0")
# mock the referenced fqdn_resolver for unit testing, takes a loong time
restore_job._populate_hostmap()
assert restore_job.host_map["node1.mydomain.net"]['source'] == [
"node1.mydomain.net"
]
assert restore_job.host_map["node2.mydomain.net"]['source'] == [
"node2.mydomain.net"
]
assert restore_job.host_map["node3.mydomain.net"]['source'] == [
"node4.mydomain.net"
]
def test_populate_ringmap(self):
node_backups = list()
node_backups.append(Mock())
with open("tests/resources/restore_cluster_tokenmap.json", 'r') as f:
cluster_backup = Mock()
tokenmap = json.loads(f.read())
cluster_backup.tokenmap.return_value = tokenmap
host_list = "tests/resources/restore_cluster_host_list.txt"
restoreJob = RestoreJob(cluster_backup, self.config, Path('/tmp'), host_list, None, False, False, None)
restoreJob._populate_hostmap()
self.assertEqual(restoreJob.host_map["node1.mydomain.net"]['source'], ["node1.mydomain.net"])
self.assertEqual(restoreJob.host_map["node2.mydomain.net"]['source'], ["node2.mydomain.net"])
self.assertEqual(restoreJob.host_map["node3.mydomain.net"]['source'], ["node4.mydomain.net"])
def test_populate_ringmap(self):
node_backups = list()
node_backups.append(Mock())
with open("tests/resources/restore_cluster_tokenmap.json", 'r') as f:
cluster_backup = Mock()
tokenmap = json.loads(f.read())
cluster_backup.tokenmap.return_value = tokenmap
host_list = "tests/resources/restore_cluster_host_list.txt"
restore_job = RestoreJob(cluster_backup, self.medusa_config, self.tmp_dir, host_list, None, False, False,
None, version_target="4.0.0")
restore_job._populate_hostmap()
assert restore_job.host_map["node1.mydomain.net"]['source'] == ["node1.mydomain.net"]
assert restore_job.host_map["node2.mydomain.net"]['source'] == ["node2.mydomain.net"]
assert restore_job.host_map["node3.mydomain.net"]['source'] == ["node4.mydomain.net"]