def test_lookup_work_by_instance(self, mock_iden):
mock_session = MagicMock()
mock_session.query().filter().one.return_value = (1, )
mock_session.query().get.return_value = 'test_work'
mock_iden.getByIdentifier.side_effect = [None, 'test_id']
testID = Work.lookupWork(mock_session, ['id1'], None)
self.assertEqual(testID, 'test_work')
def test_lookup_work_not_found(self, mock_iden):
mock_session = MagicMock()
mock_session.query().get().work.uuid = 'test_uuid'
mock_iden.getByIdentifier.side_effect = [None, None]
testID = Work.lookupWork(mock_session, ['id1'], None)
self.assertEqual(testID, None)
def test_lookup_work(self, mock_iden):
mock_session = MagicMock()
mock_session.query().get.return_value = 'test_work'
mock_iden.getByIdentifier.return_value = 'test_id'
testID = Work.lookupWork(mock_session, ['id1'], None)
self.assertEqual(testID, 'test_work')
def test_lookup_uuid(self, mock_get_uuid):
testID = Work.lookupWork('session', ['id1'], {
'type': 'uuid',
'identifier': 'test_uuid'
})
self.assertEqual(testID, 'test_id')