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')