def test_method_mapping(self, apps_mock, tree_id_mock, BridgeMock):
channel_import = ChannelImport('test')
table_map = 'test_map_method'
test_map_method = Mock()
channel_import.test_map_method = test_map_method
mapper = channel_import.generate_table_mapper(table_map=table_map)
self.assertEqual(mapper, test_map_method)
def test_method_mapping(self, apps_mock, tree_id_mock, BridgeMock):
channel_import = ChannelImport('test')
mappings = {'test_attr': 'test_map_method'}
mapper = channel_import.generate_row_mapper(mappings=mappings)
record = {}
test_map_method = Mock()
test_map_method.return_value = 'test_val'
channel_import.test_map_method = test_map_method
self.assertEqual(mapper(record, 'test_attr'), 'test_val')
