PEP8 cleanups.

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
Roberto Rosario
2017-08-27 02:08:39 -04:00
parent c27d922b3a
commit ec223dd54d
18 changed files with 15 additions and 93 deletions

View File

@@ -5,7 +5,6 @@ from django.contrib.contenttypes.models import ContentType
from document_states.tests.test_actions import ActionTestCase
from documents.permissions import permission_document_view
from ..models import AccessControlList
from ..workflow_actions import GrantAccessAction, RevokeAccessAction

View File

@@ -1,7 +1,5 @@
from __future__ import absolute_import, unicode_literals
from django.contrib.auth import get_user_model
from django.contrib.auth.models import Group
from django.core.exceptions import PermissionDenied
from django.test import override_settings
@@ -12,9 +10,6 @@ from documents.tests import (
TEST_SMALL_DOCUMENT_PATH, TEST_DOCUMENT_TYPE_LABEL,
TEST_DOCUMENT_TYPE_2_LABEL
)
from permissions.models import Role
from permissions.tests.literals import TEST_ROLE_LABEL
from user_management.tests.literals import TEST_USER_USERNAME, TEST_GROUP_NAME
from ..models import AccessControlList
@@ -46,15 +41,6 @@ class PermissionTestCase(BaseTestCase):
file_object=file_object
)
self.user = get_user_model().objects.create(
username=TEST_USER_USERNAME
)
self.group = Group.objects.create(name=TEST_GROUP_NAME)
self.role = Role.objects.create(label=TEST_ROLE_LABEL)
self.group.user_set.add(self.user)
self.role.groups.add(self.group)
def tearDown(self):
for document_type in DocumentType.objects.all():
document_type.delete()

View File

@@ -1,7 +1,6 @@
from __future__ import absolute_import, unicode_literals
import logging
from collections import OrderedDict
from django.apps import apps
from django.contrib.contenttypes.models import ContentType

View File

@@ -1,7 +1,6 @@
from __future__ import absolute_import, unicode_literals
from django.conf import settings
from django.contrib.auth import get_user_model
from django.core import mail
from django.test import override_settings
from django.urls import reverse
@@ -25,11 +24,6 @@ class UserLoginTestCase(BaseTestCase):
def setUp(self):
super(UserLoginTestCase, self).setUp()
self.admin_user = get_user_model().objects.create_superuser(
username=TEST_ADMIN_USERNAME, email=TEST_ADMIN_EMAIL,
password=TEST_ADMIN_PASSWORD
)
Namespace.invalidate_cache_all()
@override_settings(AUTHENTICATION_LOGIN_METHOD='username')

View File

@@ -1,6 +1,5 @@
from __future__ import absolute_import, unicode_literals
from documents.permissions import permission_document_view
from documents.tests.test_views import GenericDocumentViewTestCase
from ..models import Cabinet

View File

@@ -4,7 +4,6 @@ import datetime
import logging
import time
from django.contrib.auth import get_user_model
from django.test import override_settings
from django.utils.timezone import now
@@ -13,9 +12,6 @@ from documents.models import DocumentType
from documents.tests.literals import (
TEST_DOCUMENT_TYPE_LABEL, TEST_SMALL_DOCUMENT_PATH
)
from user_management.tests.literals import (
TEST_ADMIN_USERNAME, TEST_ADMIN_EMAIL, TEST_ADMIN_PASSWORD
)
from ..exceptions import (
DocumentAlreadyCheckedOut, DocumentNotCheckedOut,
@@ -28,11 +24,6 @@ from ..models import DocumentCheckout, NewVersionBlock
class DocumentCheckoutTestCase(BaseTestCase):
def setUp(self):
super(DocumentCheckoutTestCase, self).setUp()
self.admin_user = get_user_model().objects.create_superuser(
username=TEST_ADMIN_USERNAME, email=TEST_ADMIN_EMAIL,
password=TEST_ADMIN_PASSWORD
)
self.document_type = DocumentType.objects.create(
label=TEST_DOCUMENT_TYPE_LABEL
)

View File

@@ -2,7 +2,6 @@ from __future__ import unicode_literals
from django.apps import apps
from django.contrib.contenttypes.fields import GenericRelation
from django.conf import settings
from django.db import models

View File

@@ -9,4 +9,3 @@ namespace = PermissionNamespace('common', _('Common'))
permission_error_log_view = namespace.add_permission(
name='error_log_view', label=_('View errorr log')
)

View File

@@ -2,18 +2,14 @@ from __future__ import absolute_import, unicode_literals
from django.conf.urls import url
from django.contrib.auth import get_user_model
from django.contrib.auth.models import Group
from django.http import HttpResponse
from django.template import Context, Template
from django.test.utils import ContextList
from django.urls import clear_url_caches, reverse
from acls.models import AccessControlList
from permissions.models import Role
from permissions.tests.literals import TEST_ROLE_LABEL
from user_management.tests import (
TEST_ADMIN_PASSWORD, TEST_ADMIN_USERNAME, TEST_ADMIN_EMAIL,
TEST_GROUP_NAME, TEST_USER_EMAIL, TEST_USER_USERNAME, TEST_USER_PASSWORD
TEST_ADMIN_PASSWORD, TEST_ADMIN_USERNAME, TEST_USER_USERNAME,
TEST_USER_PASSWORD
)
from .base import BaseTestCase

View File

@@ -76,10 +76,6 @@ class WorkflowStateActionDynamicForm(DynamicModelForm):
action_data[field_name] = action_data[field_name].pk
data['action_data'] = action_data
#data = self.instance.get_class().clean(
# form_data=data, request=self.request
#)
data = import_string(self.action_path).clean(
form_data=data, request=self.request
)

View File

@@ -6,6 +6,7 @@ from documents.tests.test_models import GenericDocumentTestCase
class ActionTestCase(GenericDocumentTestCase):
def setUp(self):
super(ActionTestCase, self).setUp()
class MockWorkflowInstance(object):
document = self.document

View File

@@ -1,25 +1,17 @@
from __future__ import unicode_literals
from django.contrib.auth import get_user_model
from django.test import override_settings
from common.tests import BaseTestCase
from documents.models import DocumentType
from documents.search import document_search
from documents.tests import TEST_DOCUMENT_TYPE_LABEL, TEST_SMALL_DOCUMENT_PATH
from user_management.tests import (
TEST_ADMIN_PASSWORD, TEST_ADMIN_USERNAME, TEST_ADMIN_EMAIL
)
@override_settings(OCR_AUTO_OCR=False)
class DocumentSearchTestCase(BaseTestCase):
def setUp(self):
super(DocumentSearchTestCase, self).setUp()
self.admin_user = get_user_model().objects.create_superuser(
username=TEST_ADMIN_USERNAME, email=TEST_ADMIN_EMAIL,
password=TEST_ADMIN_PASSWORD
)
self.document_type = DocumentType.objects.create(
label=TEST_DOCUMENT_TYPE_LABEL
)

View File

@@ -1,7 +1,5 @@
from __future__ import unicode_literals
from collections import OrderedDict
from django.utils.translation import ugettext_lazy as _
from .classes import MailerBackend

View File

@@ -1,27 +1,16 @@
from __future__ import unicode_literals
from django.contrib.auth import get_user_model
from django.contrib.auth.models import Group
from django.core.exceptions import PermissionDenied
from common.tests import BaseTestCase
from user_management.tests import TEST_GROUP_NAME, TEST_USER_USERNAME
from ..classes import Permission
from ..models import Role
from ..permissions import permission_role_view
from .literals import TEST_ROLE_LABEL
class PermissionTestCase(BaseTestCase):
def setUp(self):
super(PermissionTestCase, self).setUp()
self.user = get_user_model().objects.create(
username=TEST_USER_USERNAME
)
self.group = Group.objects.create(name=TEST_GROUP_NAME)
self.role = Role.objects.create(label=TEST_ROLE_LABEL)
def test_no_permissions(self):
with self.assertRaises(PermissionDenied):

View File

@@ -1,13 +1,10 @@
from __future__ import unicode_literals
from django.contrib.auth import get_user_model
from django.test.client import Client
from django.urls import reverse
from common.tests import BaseTestCase
from user_management.tests import (
TEST_ADMIN_PASSWORD, TEST_ADMIN_USERNAME, TEST_ADMIN_EMAIL
)
from user_management.tests import TEST_ADMIN_PASSWORD, TEST_ADMIN_USERNAME
from ..models import Role
@@ -17,10 +14,6 @@ from .literals import TEST_ROLE_LABEL, TEST_ROLE_LABEL_EDITED
class PermissionsViewsTestCase(BaseTestCase):
def setUp(self):
super(PermissionsViewsTestCase, self).setUp()
self.admin_user = get_user_model().objects.create_superuser(
username=TEST_ADMIN_USERNAME, email=TEST_ADMIN_EMAIL,
password=TEST_ADMIN_PASSWORD
)
self.client = Client()
# Login the admin user
logged_in = self.client.login(
@@ -30,6 +23,8 @@ class PermissionsViewsTestCase(BaseTestCase):
self.assertTrue(self.admin_user.is_authenticated)
def test_role_creation_view(self):
self.role.delete()
response = self.client.post(
reverse(
'permissions:role_create',
@@ -44,11 +39,9 @@ class PermissionsViewsTestCase(BaseTestCase):
self.assertEqual(Role.objects.first().label, TEST_ROLE_LABEL)
def test_role_delete_view(self):
role = Role.objects.create(label=TEST_ROLE_LABEL)
response = self.client.post(
reverse(
'permissions:role_delete', args=(role.pk,),
'permissions:role_delete', args=(self.role.pk,),
), follow=True
)
@@ -57,11 +50,9 @@ class PermissionsViewsTestCase(BaseTestCase):
self.assertEqual(Role.objects.count(), 0)
def test_role_edit_view(self):
role = Role.objects.create(label=TEST_ROLE_LABEL)
response = self.client.post(
reverse(
'permissions:role_edit', args=(role.pk,),
'permissions:role_edit', args=(self.role.pk,),
), data={
'label': TEST_ROLE_LABEL_EDITED,
}, follow=True

View File

@@ -2,7 +2,6 @@ from __future__ import unicode_literals
import shutil
from django.contrib.auth import get_user_model
from django.test import override_settings
from common.utils import mkdtemp
@@ -13,9 +12,6 @@ from documents.tests import (
TEST_NON_ASCII_DOCUMENT_FILENAME, TEST_NON_ASCII_DOCUMENT_PATH,
TEST_NON_ASCII_COMPRESSED_DOCUMENT_PATH
)
from user_management.tests import (
TEST_ADMIN_EMAIL, TEST_ADMIN_PASSWORD, TEST_ADMIN_USERNAME
)
from ..literals import SOURCE_UNCOMPRESS_CHOICE_Y
from ..models import WatchFolderSource, WebFormSource
@@ -33,14 +29,8 @@ class UploadDocumentTestCase(BaseTestCase):
label=TEST_DOCUMENT_TYPE_LABEL
)
self.admin_user = get_user_model().objects.create_superuser(
username=TEST_ADMIN_USERNAME, email=TEST_ADMIN_EMAIL,
password=TEST_ADMIN_PASSWORD
)
def tearDown(self):
self.document_type.delete()
self.admin_user.delete()
super(UploadDocumentTestCase, self).tearDown()
def test_issue_gh_163(self):

View File

@@ -15,7 +15,8 @@ logger = logging.getLogger(__name__)
class AttachTagAction(WorkflowAction):
fields = {
'tags': {'label': _('Tags'),
'tags': {
'label': _('Tags'),
'class': 'django.forms.ModelMultipleChoiceField', 'kwargs': {
'help_text': _('Tags to attach to the document'),
'queryset': Tag.objects.none(), 'required': False
@@ -31,13 +32,14 @@ class AttachTagAction(WorkflowAction):
}
}
}
permission = permission_tag_attach
def get_form_schema(self, request):
user = request.user
logger.debug('user: %s', user)
queryset = AccessControlList.objects.filter_by_access(
permission_tag_attach, user, queryset=Tag.objects.all()
self.permission, user, queryset=Tag.objects.all()
)
self.fields['tags']['kwargs']['queryset'] = queryset
@@ -69,6 +71,7 @@ class RemoveTagAction(AttachTagAction):
},
}
label = _('Remove tag')
permission = permission_tag_remove
def execute(self, context):
for tag in self.get_tags():