From a54fe3f1317898b3346932a49ab8568ee3a1a3fa Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Wed, 8 Jul 2015 01:25:20 -0400 Subject: [PATCH] Add trash, restore and deletion test to document models. --- mayan/apps/documents/test_models.py | 32 ++++++++++++++++++++++++++--- 1 file changed, 29 insertions(+), 3 deletions(-) diff --git a/mayan/apps/documents/test_models.py b/mayan/apps/documents/test_models.py index 7dd4a0eb90..9bd668ac91 100644 --- a/mayan/apps/documents/test_models.py +++ b/mayan/apps/documents/test_models.py @@ -15,7 +15,7 @@ from django.test import TestCase from rest_framework import status from rest_framework.test import APIClient -from .models import Document, DocumentType +from .models import DeletedDocument, Document, DocumentType TEST_ADMIN_PASSWORD = 'test_admin_password' TEST_ADMIN_USERNAME = 'test_admin' @@ -43,6 +43,9 @@ class DocumentTestCase(TestCase): with open(TEST_DOCUMENT_PATH) as file_object: self.document = self.document_type.new_document(file_object=File(file_object), label='mayan_11_1.pdf') + def tearDown(self): + self.document_type.delete() + def test_document_creation(self): self.failUnlessEqual(self.document_type.label, TEST_DOCUMENT_TYPE) @@ -55,6 +58,7 @@ class DocumentTestCase(TestCase): self.failUnlessEqual(self.document.checksum, 'c637ffab6b8bb026ed3784afdb07663fddc60099853fae2be93890852a69ecf3') self.failUnlessEqual(self.document.page_count, 47) + def test_version_creation(self): with open(TEST_SMALL_DOCUMENT_PATH) as file_object: self.document.new_version(file_object=File(file_object)) @@ -63,6 +67,28 @@ class DocumentTestCase(TestCase): self.failUnlessEqual(self.document.versions.count(), 3) - def tearDown(self): + def test_restoring_documents(self): + self.assertEqual(Document.objects.count(), 1) + + # Trash the document self.document.delete() - self.document_type.delete() + self.assertEqual(DeletedDocument.objects.count(), 1) + self.assertEqual(Document.objects.count(), 0) + + # Restore the document + self.document.restore() + self.assertEqual(DeletedDocument.objects.count(), 0) + self.assertEqual(Document.objects.count(), 1) + + def test_trashing_documents(self): + self.assertEqual(Document.objects.count(), 1) + + # Trash the document + self.document.delete() + self.assertEqual(DeletedDocument.objects.count(), 1) + self.assertEqual(Document.objects.count(), 0) + + # Delete the document + self.document.delete() + self.assertEqual(DeletedDocument.objects.count(), 0) + self.assertEqual(Document.objects.count(), 0)