def test_node_name_by_id(self):
cluster_config = {
'1': {'name': 'node1', 'zone': 'a'},
'2': {'name': 'node2', 'zone': 'b'},
'3': {'name': 'node3', 'zone': 'a'},
'4': {'name': 'node4', 'zone': 'b'},
'5': {'name': 'node5', 'zone': 'c'},
'6': {'name': 'node6', 'zone': 'c'},
}
cluster = Cluster(cluster_config)
self.assertEqual('node1', cluster.node_name('1'))
self.assertEqual('node2', cluster.node_name('2'))
self.assertEqual('node3', cluster.node_name('3'))
self.assertEqual('node4', cluster.node_name('4'))
self.assertEqual('node5', cluster.node_name('5'))
self.assertEqual('node6', cluster.node_name('6'))
self.assertEqual(None, cluster.node_name('7'))
