diff options
Diffstat (limited to 'tests/auth_tests/test_migrations.py')
| -rw-r--r-- | tests/auth_tests/test_migrations.py | 138 |
1 files changed, 82 insertions, 56 deletions
diff --git a/tests/auth_tests/test_migrations.py b/tests/auth_tests/test_migrations.py index a91b9a9146..7fa0487029 100644 --- a/tests/auth_tests/test_migrations.py +++ b/tests/auth_tests/test_migrations.py @@ -9,14 +9,16 @@ from django.test.utils import captured_stdout from .models import Proxy, UserProxy -update_proxy_permissions = import_module('django.contrib.auth.migrations.0011_update_proxy_permissions') +update_proxy_permissions = import_module( + "django.contrib.auth.migrations.0011_update_proxy_permissions" +) class ProxyModelWithDifferentAppLabelTests(TransactionTestCase): available_apps = [ - 'auth_tests', - 'django.contrib.auth', - 'django.contrib.contenttypes', + "auth_tests", + "django.contrib.auth", + "django.contrib.contenttypes", ] def setUp(self): @@ -29,19 +31,25 @@ class ProxyModelWithDifferentAppLabelTests(TransactionTestCase): self.concrete_content_type = ContentType.objects.get_for_model(UserProxy) self.default_permission = Permission.objects.create( content_type=self.concrete_content_type, - codename='add_userproxy', - name='Can add userproxy', + codename="add_userproxy", + name="Can add userproxy", ) self.custom_permission = Permission.objects.create( content_type=self.concrete_content_type, - codename='use_different_app_label', - name='May use a different app label', + codename="use_different_app_label", + name="May use a different app label", ) def test_proxy_model_permissions_contenttype(self): - proxy_model_content_type = ContentType.objects.get_for_model(UserProxy, for_concrete_model=False) - self.assertEqual(self.default_permission.content_type, self.concrete_content_type) - self.assertEqual(self.custom_permission.content_type, self.concrete_content_type) + proxy_model_content_type = ContentType.objects.get_for_model( + UserProxy, for_concrete_model=False + ) + self.assertEqual( + self.default_permission.content_type, self.concrete_content_type + ) + self.assertEqual( + self.custom_permission.content_type, self.concrete_content_type + ) with connection.schema_editor() as editor: update_proxy_permissions.update_proxy_model_permissions(apps, editor) self.default_permission.refresh_from_db() @@ -54,47 +62,51 @@ class ProxyModelWithDifferentAppLabelTests(TransactionTestCase): user.user_permissions.add(self.default_permission) user.user_permissions.add(self.custom_permission) for permission in [self.default_permission, self.custom_permission]: - self.assertTrue(user.has_perm('auth.' + permission.codename)) - self.assertFalse(user.has_perm('auth_tests.' + permission.codename)) + self.assertTrue(user.has_perm("auth." + permission.codename)) + self.assertFalse(user.has_perm("auth_tests." + permission.codename)) with connection.schema_editor() as editor: update_proxy_permissions.update_proxy_model_permissions(apps, editor) # Reload user to purge the _perm_cache. user = User._default_manager.get(pk=user.pk) for permission in [self.default_permission, self.custom_permission]: - self.assertFalse(user.has_perm('auth.' + permission.codename)) - self.assertTrue(user.has_perm('auth_tests.' + permission.codename)) + self.assertFalse(user.has_perm("auth." + permission.codename)) + self.assertTrue(user.has_perm("auth_tests." + permission.codename)) def test_migrate_backwards(self): with connection.schema_editor() as editor: update_proxy_permissions.update_proxy_model_permissions(apps, editor) update_proxy_permissions.revert_proxy_model_permissions(apps, editor) self.default_permission.refresh_from_db() - self.assertEqual(self.default_permission.content_type, self.concrete_content_type) + self.assertEqual( + self.default_permission.content_type, self.concrete_content_type + ) self.custom_permission.refresh_from_db() - self.assertEqual(self.custom_permission.content_type, self.concrete_content_type) + self.assertEqual( + self.custom_permission.content_type, self.concrete_content_type + ) def test_user_keeps_same_permissions_after_migrating_backward(self): user = User.objects.create() user.user_permissions.add(self.default_permission) user.user_permissions.add(self.custom_permission) for permission in [self.default_permission, self.custom_permission]: - self.assertTrue(user.has_perm('auth.' + permission.codename)) - self.assertFalse(user.has_perm('auth_tests.' + permission.codename)) + self.assertTrue(user.has_perm("auth." + permission.codename)) + self.assertFalse(user.has_perm("auth_tests." + permission.codename)) with connection.schema_editor() as editor: update_proxy_permissions.update_proxy_model_permissions(apps, editor) update_proxy_permissions.revert_proxy_model_permissions(apps, editor) # Reload user to purge the _perm_cache. user = User._default_manager.get(pk=user.pk) for permission in [self.default_permission, self.custom_permission]: - self.assertTrue(user.has_perm('auth.' + permission.codename)) - self.assertFalse(user.has_perm('auth_tests.' + permission.codename)) + self.assertTrue(user.has_perm("auth." + permission.codename)) + self.assertFalse(user.has_perm("auth_tests." + permission.codename)) class ProxyModelWithSameAppLabelTests(TransactionTestCase): available_apps = [ - 'auth_tests', - 'django.contrib.auth', - 'django.contrib.contenttypes', + "auth_tests", + "django.contrib.auth", + "django.contrib.contenttypes", ] def setUp(self): @@ -107,19 +119,25 @@ class ProxyModelWithSameAppLabelTests(TransactionTestCase): self.concrete_content_type = ContentType.objects.get_for_model(Proxy) self.default_permission = Permission.objects.create( content_type=self.concrete_content_type, - codename='add_proxy', - name='Can add proxy', + codename="add_proxy", + name="Can add proxy", ) self.custom_permission = Permission.objects.create( content_type=self.concrete_content_type, - codename='display_proxys', - name='May display proxys information', + codename="display_proxys", + name="May display proxys information", ) def test_proxy_model_permissions_contenttype(self): - proxy_model_content_type = ContentType.objects.get_for_model(Proxy, for_concrete_model=False) - self.assertEqual(self.default_permission.content_type, self.concrete_content_type) - self.assertEqual(self.custom_permission.content_type, self.concrete_content_type) + proxy_model_content_type = ContentType.objects.get_for_model( + Proxy, for_concrete_model=False + ) + self.assertEqual( + self.default_permission.content_type, self.concrete_content_type + ) + self.assertEqual( + self.custom_permission.content_type, self.concrete_content_type + ) with connection.schema_editor() as editor: update_proxy_permissions.update_proxy_model_permissions(apps, editor) self.default_permission.refresh_from_db() @@ -132,36 +150,40 @@ class ProxyModelWithSameAppLabelTests(TransactionTestCase): user.user_permissions.add(self.default_permission) user.user_permissions.add(self.custom_permission) for permission in [self.default_permission, self.custom_permission]: - self.assertTrue(user.has_perm('auth_tests.' + permission.codename)) + self.assertTrue(user.has_perm("auth_tests." + permission.codename)) with connection.schema_editor() as editor: update_proxy_permissions.update_proxy_model_permissions(apps, editor) # Reload user to purge the _perm_cache. user = User._default_manager.get(pk=user.pk) for permission in [self.default_permission, self.custom_permission]: - self.assertTrue(user.has_perm('auth_tests.' + permission.codename)) + self.assertTrue(user.has_perm("auth_tests." + permission.codename)) def test_migrate_backwards(self): with connection.schema_editor() as editor: update_proxy_permissions.update_proxy_model_permissions(apps, editor) update_proxy_permissions.revert_proxy_model_permissions(apps, editor) self.default_permission.refresh_from_db() - self.assertEqual(self.default_permission.content_type, self.concrete_content_type) + self.assertEqual( + self.default_permission.content_type, self.concrete_content_type + ) self.custom_permission.refresh_from_db() - self.assertEqual(self.custom_permission.content_type, self.concrete_content_type) + self.assertEqual( + self.custom_permission.content_type, self.concrete_content_type + ) def test_user_keeps_same_permissions_after_migrating_backward(self): user = User.objects.create() user.user_permissions.add(self.default_permission) user.user_permissions.add(self.custom_permission) for permission in [self.default_permission, self.custom_permission]: - self.assertTrue(user.has_perm('auth_tests.' + permission.codename)) + self.assertTrue(user.has_perm("auth_tests." + permission.codename)) with connection.schema_editor() as editor: update_proxy_permissions.update_proxy_model_permissions(apps, editor) update_proxy_permissions.revert_proxy_model_permissions(apps, editor) # Reload user to purge the _perm_cache. user = User._default_manager.get(pk=user.pk) for permission in [self.default_permission, self.custom_permission]: - self.assertTrue(user.has_perm('auth_tests.' + permission.codename)) + self.assertTrue(user.has_perm("auth_tests." + permission.codename)) def test_migrate_with_existing_target_permission(self): """ @@ -172,48 +194,52 @@ class ProxyModelWithSameAppLabelTests(TransactionTestCase): Output a reminder to audit relevant permissions. """ - proxy_model_content_type = ContentType.objects.get_for_model(Proxy, for_concrete_model=False) + proxy_model_content_type = ContentType.objects.get_for_model( + Proxy, for_concrete_model=False + ) Permission.objects.create( content_type=proxy_model_content_type, - codename='add_proxy', - name='Can add proxy', + codename="add_proxy", + name="Can add proxy", ) Permission.objects.create( content_type=proxy_model_content_type, - codename='display_proxys', - name='May display proxys information', + codename="display_proxys", + name="May display proxys information", ) with captured_stdout() as stdout: with connection.schema_editor() as editor: update_proxy_permissions.update_proxy_model_permissions(apps, editor) - self.assertIn('A problem arose migrating proxy model permissions', stdout.getvalue()) + self.assertIn( + "A problem arose migrating proxy model permissions", stdout.getvalue() + ) class MultiDBProxyModelAppLabelTests(TransactionTestCase): - databases = {'default', 'other'} + databases = {"default", "other"} available_apps = [ - 'auth_tests', - 'django.contrib.auth', - 'django.contrib.contenttypes', + "auth_tests", + "django.contrib.auth", + "django.contrib.contenttypes", ] def setUp(self): ContentType.objects.all().delete() - Permission.objects.using('other').delete() - concrete_content_type = ContentType.objects.db_manager( - 'other' - ).get_for_model(Proxy) - self.permission = Permission.objects.using('other').create( + Permission.objects.using("other").delete() + concrete_content_type = ContentType.objects.db_manager("other").get_for_model( + Proxy + ) + self.permission = Permission.objects.using("other").create( content_type=concrete_content_type, - codename='add_proxy', - name='Can add proxy', + codename="add_proxy", + name="Can add proxy", ) def test_migrate_other_database(self): proxy_model_content_type = ContentType.objects.db_manager( - 'other' + "other" ).get_for_model(Proxy, for_concrete_model=False) - with connections['other'].schema_editor() as editor: + with connections["other"].schema_editor() as editor: update_proxy_permissions.update_proxy_model_permissions(apps, editor) self.permission.refresh_from_db() self.assertEqual(self.permission.content_type, proxy_model_content_type) |