def test_get_by_uuid_missing(self, mock_uuid):
mock_session = MagicMock()
mock_session.query().filter().one.side_effect = NoResultFound
with self.assertRaises(DBError):
Work.getByUUID(mock_session, 'uuid')
def test_get_by_uuid(self, mock_uuid):
mock_session = MagicMock()
mock_session.query().filter().one.return_value = 'exist_uuid'
testUUID = Work.getByUUID(mock_session, 'uuid')
self.assertEqual(testUUID, 'exist_uuid')
