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"]