chore: Lint test_commands

Author: CuriousLearner

tests/test_commands.py

@@ -5,10 +5,11 @@
 from pathlib import Path
 from unittest.mock import patch
 
-import pytest
-import yaml
 from django.core.management import call_command
 from django.core.management.base import CommandError
+
+import pytest
+import yaml
 from model_bakery import baker
 
 from django_postgres_anon.models import MaskedRole, MaskingLog, MaskingPreset, MaskingRule
@@ -744,25 +745,26 @@ def test_command_permissions():
 @pytest.mark.django_db
 def test_anon_fix_permissions_command():
     """Test anon_fix_permissions management command"""
+    from unittest.mock import MagicMock, patch
+
     from django_postgres_anon.models import MaskedRole
-    from unittest.mock import patch, MagicMock
 
     # Test with no arguments (should show error)
-    with patch('sys.stdout', new_callable=MagicMock) as mock_stdout:
+    with patch("sys.stdout", new_callable=MagicMock) as mock_stdout:
         call_command("anon_fix_permissions")
         output = str(mock_stdout.write.call_args_list)
         assert "Please specify" in output or "--role" in output
 
     # Test with --role option
-    with patch('django_postgres_anon.management.commands.anon_fix_permissions.create_masked_role') as mock_create:
+    with patch("django_postgres_anon.management.commands.anon_fix_permissions.create_masked_role") as mock_create:
         mock_create.return_value = True
-        with patch('sys.stdout', new_callable=MagicMock):
+        with patch("sys.stdout", new_callable=MagicMock):
             call_command("anon_fix_permissions", "--role", "test_role")
             mock_create.assert_called_once_with("test_role")
 
     # Test with --all option when no roles exist
     MaskedRole.objects.all().delete()
-    with patch('sys.stdout', new_callable=MagicMock) as mock_stdout:
+    with patch("sys.stdout", new_callable=MagicMock) as mock_stdout:
         call_command("anon_fix_permissions", "--all")
         # Should show warning about no roles
         output = str(mock_stdout.write.call_args_list)
@@ -772,18 +774,18 @@ def test_anon_fix_permissions_command():
     MaskedRole.objects.create(role_name="test_role_1")
     MaskedRole.objects.create(role_name="test_role_2")
 
-    with patch('django_postgres_anon.management.commands.anon_fix_permissions.create_masked_role') as mock_create:
+    with patch("django_postgres_anon.management.commands.anon_fix_permissions.create_masked_role") as mock_create:
         mock_create.return_value = True
-        with patch('sys.stdout', new_callable=MagicMock) as mock_stdout:
+        with patch("sys.stdout", new_callable=MagicMock) as mock_stdout:
             call_command("anon_fix_permissions", "--all")
             assert mock_create.call_count == 2
             mock_create.assert_any_call("test_role_1")
             mock_create.assert_any_call("test_role_2")
 
     # Test with failed permission fix
-    with patch('django_postgres_anon.management.commands.anon_fix_permissions.create_masked_role') as mock_create:
+    with patch("django_postgres_anon.management.commands.anon_fix_permissions.create_masked_role") as mock_create:
         mock_create.return_value = False
-        with patch('sys.stdout', new_callable=MagicMock) as mock_stdout:
+        with patch("sys.stdout", new_callable=MagicMock) as mock_stdout:
             call_command("anon_fix_permissions", "--role", "failing_role")
             output = str(mock_stdout.write.call_args_list)
             assert "Failed" in output or "failed" in output.lower()