Finish updating checkouts app to support organizations.

This commit is contained in:
Roberto Rosario
2016-06-08 16:58:57 -04:00
parent 7ae0917564
commit a2f8e8b8d8
10 changed files with 159 additions and 42 deletions

View File

@@ -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())