def test_no_column_mapping(self, apps_mock, tree_id_mock, BridgeMock):
channel_import = ChannelImport('test')
mappings = {'test_attr': 'test_attr_mapped'}
mapper = channel_import.generate_row_mapper(mappings=mappings)
record = Mock(spec=['test_attr'])
with self.assertRaises(AttributeError):
mapper(record, 'test_attr')
def test_column_name_mapping(self, apps_mock, tree_id_mock, BridgeMock):
channel_import = ChannelImport('test')
mappings = {'test_attr': 'test_attr_mapped'}
mapper = channel_import.generate_row_mapper(mappings=mappings)
record = MagicMock()
record.test_attr_mapped = 'test_val'
self.assertEqual(mapper(record, 'test_attr'), 'test_val')
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')
def test_base_mapper(self, apps_mock, tree_id_mock, BridgeMock):
channel_import = ChannelImport('test')
mapper = channel_import.generate_row_mapper()
record = MagicMock()
record.test_attr = 'test_val'
self.assertEqual(mapper(record, 'test_attr'), 'test_val')