diff --git a/mayan/apps/checkouts/managers.py b/mayan/apps/checkouts/managers.py index 064c4bda37..6d795729ce 100644 --- a/mayan/apps/checkouts/managers.py +++ b/mayan/apps/checkouts/managers.py @@ -19,7 +19,7 @@ logger = logging.getLogger(__name__) class DocumentCheckoutManager(models.Manager): def checkout_document(self, document, expiration_datetime, user, block_new_version=True): - self.create( + return self.create( document=document, expiration_datetime=expiration_datetime, user=user, block_new_version=block_new_version ) diff --git a/mayan/apps/checkouts/tests/test_api.py b/mayan/apps/checkouts/tests/test_api.py index 67b41a365f..2453ab432c 100644 --- a/mayan/apps/checkouts/tests/test_api.py +++ b/mayan/apps/checkouts/tests/test_api.py @@ -40,18 +40,19 @@ class CheckoutsAPITestCase(BaseAPITestCase): def _request_checkedout_document_view(self): return self.get( viewname='rest_api:checkedout-document-view', - args=(self.document.pk,) + args=(self.checkout.pk,) ) def _checkout_document(self): expiration_datetime = now() + datetime.timedelta(days=1) - DocumentCheckout.objects.checkout_document( + self.checkout = DocumentCheckout.objects.checkout_document( document=self.document, expiration_datetime=expiration_datetime, user=self.admin_user, block_new_version=True ) def test_checkedout_document_view_no_access(self): + self._checkout_document() response = self._request_checkedout_document_view() self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND)