227 lines
7.5 KiB
Python
227 lines
7.5 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from ..models import DeletedDocument, Document
|
|
from ..permissions import (
|
|
permission_document_delete, permission_document_restore,
|
|
permission_document_trash, permission_document_view
|
|
)
|
|
|
|
from .base import GenericDocumentViewTestCase
|
|
|
|
|
|
class TrashedDocumentTestMixin(object):
|
|
def _request_document_restore_get_view(self):
|
|
return self.get(
|
|
viewname='documents:document_restore', kwargs={
|
|
'pk': self.test_document.pk
|
|
}
|
|
)
|
|
|
|
def _request_document_restore_post_view(self):
|
|
return self.post(
|
|
viewname='documents:document_restore', kwargs={
|
|
'pk': self.test_document.pk
|
|
}
|
|
)
|
|
|
|
def _request_document_trash_get_view(self):
|
|
return self.get(
|
|
viewname='documents:document_trash', kwargs={
|
|
'pk': self.test_document.pk
|
|
}
|
|
)
|
|
|
|
def _request_document_trash_post_view(self):
|
|
return self.post(
|
|
viewname='documents:document_trash', kwargs={
|
|
'pk': self.test_document.pk
|
|
}
|
|
)
|
|
|
|
def _request_trashed_document_delete_get_view(self):
|
|
return self.get(
|
|
viewname='documents:document_delete', kwargs={
|
|
'pk': self.test_document.pk
|
|
}
|
|
)
|
|
|
|
def _request_trashed_document_delete_post_view(self):
|
|
return self.post(
|
|
viewname='documents:document_delete', kwargs={
|
|
'pk': self.test_document.pk
|
|
}
|
|
)
|
|
|
|
def _request_trashed_document_list_view(self):
|
|
return self.get(viewname='documents:document_list_deleted')
|
|
|
|
|
|
class TrashedDocumentTestCase(
|
|
TrashedDocumentTestMixin, GenericDocumentViewTestCase
|
|
):
|
|
def test_document_restore_get_view_no_permission(self):
|
|
self.test_document.delete()
|
|
self.assertEqual(Document.objects.count(), 0)
|
|
|
|
document_count = Document.objects.count()
|
|
|
|
response = self._request_document_restore_get_view()
|
|
self.assertEqual(response.status_code, 404)
|
|
|
|
self.assertEqual(Document.objects.count(), document_count)
|
|
|
|
def test_document_restore_get_view_with_access(self):
|
|
self.test_document.delete()
|
|
self.assertEqual(Document.objects.count(), 0)
|
|
|
|
self.grant_access(
|
|
obj=self.test_document, permission=permission_document_restore
|
|
)
|
|
|
|
document_count = Document.objects.count()
|
|
|
|
response = self._request_document_restore_get_view()
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
self.assertEqual(Document.objects.count(), document_count)
|
|
|
|
def test_document_restore_post_view_no_permission(self):
|
|
self.test_document.delete()
|
|
self.assertEqual(Document.objects.count(), 0)
|
|
|
|
response = self._request_document_restore_post_view()
|
|
self.assertEqual(response.status_code, 404)
|
|
|
|
self.assertEqual(DeletedDocument.objects.count(), 1)
|
|
self.assertEqual(Document.objects.count(), 0)
|
|
|
|
def test_document_restore_post_view_with_access(self):
|
|
self.test_document.delete()
|
|
self.assertEqual(Document.objects.count(), 0)
|
|
|
|
self.grant_access(
|
|
obj=self.test_document, permission=permission_document_restore
|
|
)
|
|
|
|
response = self._request_document_restore_post_view()
|
|
self.assertEqual(response.status_code, 302)
|
|
|
|
self.assertEqual(DeletedDocument.objects.count(), 0)
|
|
self.assertEqual(Document.objects.count(), 1)
|
|
|
|
def test_document_trash_get_view_no_permissions(self):
|
|
document_count = Document.objects.count()
|
|
|
|
response = self._request_document_trash_get_view()
|
|
self.assertEqual(response.status_code, 404)
|
|
|
|
self.assertEqual(Document.objects.count(), document_count)
|
|
|
|
def test_document_trash_get_view_with_access(self):
|
|
self.grant_access(
|
|
obj=self.test_document, permission=permission_document_trash
|
|
)
|
|
|
|
document_count = Document.objects.count()
|
|
|
|
response = self._request_document_trash_get_view()
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
self.assertEqual(Document.objects.count(), document_count)
|
|
|
|
def test_document_trash_post_view_no_permissions(self):
|
|
response = self._request_document_trash_post_view()
|
|
self.assertEqual(response.status_code, 404)
|
|
|
|
self.assertEqual(DeletedDocument.objects.count(), 0)
|
|
self.assertEqual(Document.objects.count(), 1)
|
|
|
|
def test_document_trash_post_view_with_access(self):
|
|
self.grant_access(
|
|
obj=self.test_document, permission=permission_document_trash
|
|
)
|
|
|
|
response = self._request_document_trash_post_view()
|
|
self.assertEqual(response.status_code, 302)
|
|
|
|
self.assertEqual(DeletedDocument.objects.count(), 1)
|
|
self.assertEqual(Document.objects.count(), 0)
|
|
|
|
def test_document_delete_get_view_no_permissions(self):
|
|
self.test_document.delete()
|
|
self.assertEqual(Document.objects.count(), 0)
|
|
self.assertEqual(DeletedDocument.objects.count(), 1)
|
|
|
|
trashed_document_count = DeletedDocument.objects.count()
|
|
|
|
response = self._request_trashed_document_delete_get_view()
|
|
self.assertEqual(response.status_code, 404)
|
|
|
|
self.assertEqual(
|
|
DeletedDocument.objects.count(), trashed_document_count
|
|
)
|
|
|
|
def test_document_delete_get_view_with_access(self):
|
|
self.test_document.delete()
|
|
self.assertEqual(Document.objects.count(), 0)
|
|
self.assertEqual(DeletedDocument.objects.count(), 1)
|
|
|
|
self.grant_access(
|
|
obj=self.test_document, permission=permission_document_delete
|
|
)
|
|
|
|
trashed_document_count = DeletedDocument.objects.count()
|
|
|
|
response = self._request_trashed_document_delete_get_view()
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
self.assertEqual(
|
|
DeletedDocument.objects.count(), trashed_document_count
|
|
)
|
|
|
|
def test_document_delete_post_view_no_permissions(self):
|
|
self.test_document.delete()
|
|
self.assertEqual(Document.objects.count(), 0)
|
|
self.assertEqual(DeletedDocument.objects.count(), 1)
|
|
|
|
response = self._request_trashed_document_delete_post_view()
|
|
self.assertEqual(response.status_code, 404)
|
|
|
|
self.assertEqual(Document.objects.count(), 0)
|
|
self.assertEqual(DeletedDocument.objects.count(), 1)
|
|
|
|
def test_document_delete_post_view_with_access(self):
|
|
self.test_document.delete()
|
|
self.assertEqual(Document.objects.count(), 0)
|
|
self.assertEqual(DeletedDocument.objects.count(), 1)
|
|
|
|
self.grant_access(
|
|
obj=self.test_document, permission=permission_document_delete
|
|
)
|
|
|
|
response = self._request_trashed_document_delete_post_view()
|
|
self.assertEqual(response.status_code, 302)
|
|
|
|
self.assertEqual(DeletedDocument.objects.count(), 0)
|
|
self.assertEqual(Document.objects.count(), 0)
|
|
|
|
def test_deleted_document_list_view_no_permissions(self):
|
|
self.test_document.delete()
|
|
|
|
response = self._request_trashed_document_list_view()
|
|
self.assertNotContains(
|
|
response=response, text=self.test_document.label, status_code=200
|
|
)
|
|
|
|
def test_deleted_document_list_view_with_access(self):
|
|
self.test_document.delete()
|
|
|
|
self.grant_access(
|
|
obj=self.test_document, permission=permission_document_view
|
|
)
|
|
|
|
response = self._request_trashed_document_list_view()
|
|
self.assertContains(
|
|
response=response, text=self.test_document.label, status_code=200
|
|
)
|