mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-04-11 10:00:48 -05:00
9712ac109d
* Initial implementation of consumption templates * Frontend implementation of consumption templates Testing * Support consumption template source * order templates, automatically add permissions * Support title assignment in consumption templates * Refactoring, filters to and, show sources on list Show sources on template list, update some translation strings Make filters and minor testing * Update strings * Only update django-multiselectfield * Basic docs, document some methods * Improve testing coverage, template multi-assignment merges
44 lines
1.9 KiB
Python
44 lines
1.9 KiB
Python
from django.contrib.auth import get_user_model
|
|
|
|
from documents.tests.utils import TestMigrations
|
|
|
|
|
|
class TestMigrateConsumptionTemplate(TestMigrations):
|
|
migrate_from = "1038_sharelink"
|
|
migrate_to = "1039_consumptiontemplate"
|
|
|
|
def setUpBeforeMigration(self, apps):
|
|
User = get_user_model()
|
|
Group = apps.get_model("auth.Group")
|
|
self.Permission = apps.get_model("auth", "Permission")
|
|
self.user = User.objects.create(username="user1")
|
|
self.group = Group.objects.create(name="group1")
|
|
permission = self.Permission.objects.get(codename="add_document")
|
|
self.user.user_permissions.add(permission.id)
|
|
self.group.permissions.add(permission.id)
|
|
|
|
def test_users_with_add_documents_get_add_consumptiontemplate(self):
|
|
permission = self.Permission.objects.get(codename="add_consumptiontemplate")
|
|
self.assertTrue(self.user.has_perm(f"documents.{permission.codename}"))
|
|
self.assertTrue(permission in self.group.permissions.all())
|
|
|
|
|
|
class TestReverseMigrateConsumptionTemplate(TestMigrations):
|
|
migrate_from = "1039_consumptiontemplate"
|
|
migrate_to = "1038_sharelink"
|
|
|
|
def setUpBeforeMigration(self, apps):
|
|
User = get_user_model()
|
|
Group = apps.get_model("auth.Group")
|
|
self.Permission = apps.get_model("auth", "Permission")
|
|
self.user = User.objects.create(username="user1")
|
|
self.group = Group.objects.create(name="group1")
|
|
permission = self.Permission.objects.get(codename="add_consumptiontemplate")
|
|
self.user.user_permissions.add(permission.id)
|
|
self.group.permissions.add(permission.id)
|
|
|
|
def test_remove_consumptiontemplate_permissions(self):
|
|
permission = self.Permission.objects.get(codename="add_consumptiontemplate")
|
|
self.assertFalse(self.user.has_perm(f"documents.{permission.codename}"))
|
|
self.assertFalse(permission in self.group.permissions.all())
|