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