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_no_column_mapping(self, apps_mock, tree_id_mock, BridgeMock):
channel_import = ChannelImport('test')
table_map = 'test_map_method'
with self.assertRaises(AttributeError):
channel_import.generate_table_mapper(table_map=table_map)
def test_base_mapper(self, apps_mock, tree_id_mock, BridgeMock):
channel_import = ChannelImport('test')
mapper = channel_import.generate_table_mapper()
self.assertEqual(mapper, channel_import.base_table_mapper)
