From b71ebe45f9ef407ee26f5f97b1c341cfb7a8618b Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Thu, 26 May 2016 14:52:25 -0400 Subject: [PATCH] Initial sample a organization view test. --- .../folders/tests/test_organization_views.py | 39 +++++++++++++++++++ 1 file changed, 39 insertions(+) create mode 100644 mayan/apps/folders/tests/test_organization_views.py diff --git a/mayan/apps/folders/tests/test_organization_views.py b/mayan/apps/folders/tests/test_organization_views.py new file mode 100644 index 0000000000..29c6c443b8 --- /dev/null +++ b/mayan/apps/folders/tests/test_organization_views.py @@ -0,0 +1,39 @@ +from __future__ import absolute_import, unicode_literals + +from documents.tests.test_views import GenericDocumentViewTestCase +from organizations.models import Organization + +from ..models import Folder +from .literals import TEST_FOLDER_LABEL, TEST_FOLDER_EDITED_LABEL + + +class FolderOrganizationViewTestCase(GenericDocumentViewTestCase): + def setUp(self): + super(FolderOrganizationViewTestCase, self).setUp() + # Create two organizations + self.organization_a = Organization.objects.create( + label='Organization A' + ) + self.organization_b = Organization.objects.create( + label='Organization B' + ) + + # Create an organization admin for organization B + username, password = self.organization_b.create_admin() + with self.settings(ORGANIZATION_ID=self.organization_b.pk): + # Login organization admin for organization B + self.login(username=username, password=password) + + def test_folder_view_view(self): + # Create a folder for organization A + folder = Folder.objects.create( + organization=self.organization_a, label=TEST_FOLDER_LABEL + ) + + with self.settings(ORGANIZATION_ID=self.organization_b.pk): + # Make sure admin for organization B cannot find the folder for + # organization A + response = self.get( + 'folders:folder_view', args=(folder.pk,), + ) + self.assertEqual(response.status_code, 404)