Files
mayan-edms/mayan/apps/checkouts/tests/mixins.py
Roberto Rosario 0267c79b07 Backport test improvements
Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
2019-04-20 02:39:57 -04:00

26 lines
719 B
Python

from __future__ import unicode_literals
import datetime
from django.utils.timezone import now
from ..models import DocumentCheckout
class DocumentCheckoutTestMixin(object):
_test_document_check_out_seconds = 0.1
def _check_out_test_document(self, user=None):
if not user:
user = self._test_case_user
self._check_out_expiration_datetime = now() + datetime.timedelta(
seconds=self._test_document_check_out_seconds
)
self.test_check_out = DocumentCheckout.objects.check_out_document(
document=self.test_document,
expiration_datetime=self._check_out_expiration_datetime,
block_new_version=True, user=user
)