def test_get_channel_info_by_peer_target(self):
# GIVEN
default_CHANNEL_MANAGE_DATA_PATH = conf.CHANNEL_MANAGE_DATA_PATH
conf.CHANNEL_MANAGE_DATA_PATH = os.path.join(conf.LOOPCHAIN_ROOT_PATH,
"testcase/unittest/channel_manage_data_for_test.json")
admin_manager = AdminManager("station")
peer_target1 = '1.1.1.1:1111'
peer_target2 = '2.2.2.2:2222'
peer_target3 = '3.3.3.3:3333'
peer_target4 = '4.4.4.4:4444'
channel1 = 'loopchain_default'
channel2 = 'loopchain_test'
# WHEN
channel_infos1 = json.loads(admin_manager.get_channel_infos_by_peer_target(peer_target1))
channel_infos2 = json.loads(admin_manager.get_channel_infos_by_peer_target(peer_target2))
channel_infos3 = json.loads(admin_manager.get_channel_infos_by_peer_target(peer_target3))
channel_infos4 = json.loads(admin_manager.get_channel_infos_by_peer_target(peer_target4))
# THEN
self.assertEqual(list(channel_infos1.keys()), [channel1, channel2])
self.assertEqual(list(channel_infos2.keys()), [channel1])
self.assertEqual(list(channel_infos3.keys()), [channel2])
self.assertEqual(list(channel_infos4.keys()), [])
# CLEAR
conf.CHANNEL_MANAGE_DATA_PATH = default_CHANNEL_MANAGE_DATA_PATH
def test_delete_peer_target(self):
# GIVEN
default_CHANNEL_MANAGE_DATA_PATH = conf.CHANNEL_MANAGE_DATA_PATH
conf.CHANNEL_MANAGE_DATA_PATH = os.path.join(conf.LOOPCHAIN_ROOT_PATH,
"testcase/unittest/channel_manage_data_for_test.json")
admin_manager = AdminManager("station")
i = 0
remove_peer_target = '2.2.2.2:2222'
loaded_data = admin_manager.json_data
filtered_channel_infos = admin_manager.get_channel_infos_by_peer_target(remove_peer_target)
filtered_list = list(json.loads(filtered_channel_infos))
# WHEN
modified_data = admin_manager.delete_peer_target(
loaded_data, remove_peer_target, filtered_list, i)
second_peer_target_list = modified_data[filtered_list[0]]["peers"]
# THEN
self.assertEqual(len(second_peer_target_list), 1)
# CLEAR
conf.CHANNEL_MANAGE_DATA_PATH = default_CHANNEL_MANAGE_DATA_PATH
