Finish updating checkouts app to support organizations.
This commit is contained in:
@@ -25,12 +25,12 @@ from ..models import DocumentCheckout
|
||||
class DocumentCheckoutTestCase(OrganizationTestCase):
|
||||
def setUp(self):
|
||||
super(DocumentCheckoutTestCase, self).setUp()
|
||||
self.admin_user = get_user_model().objects.create_superuser(
|
||||
self.admin_user = get_user_model().on_organization.create_superuser(
|
||||
username=TEST_ADMIN_USERNAME, email=TEST_ADMIN_EMAIL,
|
||||
password=TEST_ADMIN_PASSWORD
|
||||
)
|
||||
|
||||
self.document_type = DocumentType.objects.create(
|
||||
self.document_type = DocumentType.on_organization.create(
|
||||
label=TEST_DOCUMENT_TYPE
|
||||
)
|
||||
|
||||
@@ -46,14 +46,14 @@ class DocumentCheckoutTestCase(OrganizationTestCase):
|
||||
def test_document_checkout(self):
|
||||
expiration_datetime = now() + datetime.timedelta(days=1)
|
||||
|
||||
DocumentCheckout.objects.checkout_document(
|
||||
DocumentCheckout.on_organization.checkout_document(
|
||||
document=self.document, expiration_datetime=expiration_datetime,
|
||||
user=self.admin_user, block_new_version=True
|
||||
)
|
||||
|
||||
self.assertTrue(self.document.is_checked_out())
|
||||
self.assertTrue(
|
||||
DocumentCheckout.objects.is_document_checked_out(
|
||||
DocumentCheckout.on_organization.is_document_checked_out(
|
||||
document=self.document
|
||||
)
|
||||
)
|
||||
@@ -61,7 +61,7 @@ class DocumentCheckoutTestCase(OrganizationTestCase):
|
||||
def test_version_creation_blocking(self):
|
||||
expiration_datetime = now() + datetime.timedelta(days=1)
|
||||
|
||||
DocumentCheckout.objects.checkout_document(
|
||||
DocumentCheckout.on_organization.checkout_document(
|
||||
document=self.document, expiration_datetime=expiration_datetime,
|
||||
user=self.admin_user, block_new_version=True
|
||||
)
|
||||
@@ -73,7 +73,7 @@ class DocumentCheckoutTestCase(OrganizationTestCase):
|
||||
def test_checkin_in(self):
|
||||
expiration_datetime = now() + datetime.timedelta(days=1)
|
||||
|
||||
DocumentCheckout.objects.checkout_document(
|
||||
DocumentCheckout.on_organization.checkout_document(
|
||||
document=self.document, expiration_datetime=expiration_datetime,
|
||||
user=self.admin_user, block_new_version=True
|
||||
)
|
||||
@@ -82,7 +82,7 @@ class DocumentCheckoutTestCase(OrganizationTestCase):
|
||||
|
||||
self.assertFalse(self.document.is_checked_out())
|
||||
self.assertFalse(
|
||||
DocumentCheckout.objects.is_document_checked_out(
|
||||
DocumentCheckout.on_organization.is_document_checked_out(
|
||||
document=self.document
|
||||
)
|
||||
)
|
||||
@@ -90,13 +90,13 @@ class DocumentCheckoutTestCase(OrganizationTestCase):
|
||||
def test_double_checkout(self):
|
||||
expiration_datetime = now() + datetime.timedelta(days=1)
|
||||
|
||||
DocumentCheckout.objects.checkout_document(
|
||||
DocumentCheckout.on_organization.checkout_document(
|
||||
document=self.document, expiration_datetime=expiration_datetime,
|
||||
user=self.admin_user, block_new_version=True
|
||||
)
|
||||
|
||||
with self.assertRaises(DocumentAlreadyCheckedOut):
|
||||
DocumentCheckout.objects.checkout_document(
|
||||
DocumentCheckout.on_organization.checkout_document(
|
||||
document=self.document,
|
||||
expiration_datetime=expiration_datetime, user=self.admin_user,
|
||||
block_new_version=True
|
||||
@@ -109,13 +109,13 @@ class DocumentCheckoutTestCase(OrganizationTestCase):
|
||||
def test_auto_checkin(self):
|
||||
expiration_datetime = now() + datetime.timedelta(seconds=1)
|
||||
|
||||
DocumentCheckout.objects.checkout_document(
|
||||
DocumentCheckout.on_organization.checkout_document(
|
||||
document=self.document, expiration_datetime=expiration_datetime,
|
||||
user=self.admin_user, block_new_version=True
|
||||
)
|
||||
|
||||
time.sleep(2)
|
||||
|
||||
DocumentCheckout.objects.check_in_expired_check_outs()
|
||||
DocumentCheckout.on_organization.check_in_expired_check_outs()
|
||||
|
||||
self.assertFalse(self.document.is_checked_out())
|
||||
|
||||
Reference in New Issue
Block a user