Format code with Ruff

Author: bluetech

pytest_django/__init__.py

@@ -5,6 +5,6 @@
     __version__ = "unknown"
 
 
-__all__ = (
+__all__ = [
     "__version__",
-)
+]

pytest_django/asserts.py

@@ -113,15 +113,15 @@ def assertRaisesMessage(
         expected_exception: type[Exception],
         expected_message: str,
         *args,
-        **kwargs
+        **kwargs,
     ):
         ...
 
     def assertWarnsMessage(
         expected_warning: Warning,
         expected_message: str,
         *args,
-        **kwargs
+        **kwargs,
     ):
         ...
 
@@ -209,7 +209,7 @@ def assertNumQueries(
         func=...,
         *args,
         using: str = ...,
-        **kwargs
+        **kwargs,
     ):
         ...
 

pytest_django/fixtures.py

@@ -28,7 +28,7 @@
     import django
 
 
-_DjangoDbDatabases = Optional[Union[Literal['__all__'], Iterable[str]]]
+_DjangoDbDatabases = Optional[Union[Literal["__all__"], Iterable[str]]]
 _DjangoDbAvailableApps = Optional[List[str]]
 # transaction, reset_sequences, databases, serialized_rollback, available_apps
 _DjangoDb = Tuple[bool, bool, _DjangoDbDatabases, bool, _DjangoDbAvailableApps]
@@ -134,7 +134,7 @@ def django_db_setup(
         db_cfg = setup_databases(
             verbosity=request.config.option.verbose,
             interactive=False,
-            **setup_databases_args
+            **setup_databases_args,
         )
 
     yield
@@ -145,9 +145,7 @@ def django_db_setup(
                 teardown_databases(db_cfg, verbosity=request.config.option.verbose)
             except Exception as exc:  # noqa: BLE001
                 request.node.warn(
-                    pytest.PytestWarning(
-                        f"Error when trying to teardown test databases: {exc!r}"
-                    )
+                    pytest.PytestWarning(f"Error when trying to teardown test databases: {exc!r}")
                 )
 
 
@@ -181,13 +179,12 @@ def _django_db_helper(
             available_apps,
         ) = False, False, None, False, None
 
-    transactional = transactional or reset_sequences or (
-        "transactional_db" in request.fixturenames
-        or "live_server" in request.fixturenames
-    )
-    reset_sequences = reset_sequences or (
-        "django_db_reset_sequences" in request.fixturenames
+    transactional = (
+        transactional
+        or reset_sequences
+        or ("transactional_db" in request.fixturenames or "live_server" in request.fixturenames)
     )
+    reset_sequences = reset_sequences or ("django_db_reset_sequences" in request.fixturenames)
     serialized_rollback = serialized_rollback or (
         "django_db_serialized_rollback" in request.fixturenames
     )
@@ -229,6 +226,7 @@ class PytestDjangoTestCase(test_case_class):  # type: ignore[misc,valid-type]
         # functionality to these methods, in which case skipping them completely
         # would not be desirable. Let's cross that bridge when we get there...
         if not transactional:
+
             @classmethod
             def setUpClass(cls) -> None:
                 super(django.test.TestCase, cls).setUpClass()
@@ -558,9 +556,11 @@ def live_server(request: pytest.FixtureRequest):
     """
     skip_if_no_django()
 
-    addr = request.config.getvalue("liveserver") or os.getenv(
-        "DJANGO_LIVE_TEST_SERVER_ADDRESS"
-    ) or "localhost"
+    addr = (
+        request.config.getvalue("liveserver")
+        or os.getenv("DJANGO_LIVE_TEST_SERVER_ADDRESS")
+        or "localhost"
+    )
 
     server = live_server_helper.LiveServer(addr)
     yield server

pytest_django/live_server_helper.py

@@ -45,7 +45,7 @@ def __init__(self, addr: str, *, start: bool = True) -> None:
         self.thread = LiveServerThread(host, **liveserver_kwargs)
 
         self._live_server_modified_settings = modify_settings(
-            ALLOWED_HOSTS={"append": host}
+            ALLOWED_HOSTS={"append": host},
         )
         # `_live_server_modified_settings` is enabled and disabled by
         # `_live_server_helper`.

pytest_django/plugin.py

@@ -112,15 +112,17 @@ def pytest_addoption(parser: pytest.Parser) -> None:
         help="Enable Django migrations on test setup",
     )
     parser.addini(
-        CONFIGURATION_ENV, "django-configurations class to use by pytest-django."
+        CONFIGURATION_ENV,
+        "django-configurations class to use by pytest-django.",
     )
     group.addoption(
         "--liveserver",
         default=None,
         help="Address and port for the live_server fixture.",
     )
     parser.addini(
-        SETTINGS_MODULE_ENV, "Django settings module to use by pytest-django."
+        SETTINGS_MODULE_ENV,
+        "Django settings module to use by pytest-django.",
     )
 
     parser.addini(
@@ -131,8 +133,7 @@ def pytest_addoption(parser: pytest.Parser) -> None:
     )
     parser.addini(
         "django_debug_mode",
-        "How to set the Django DEBUG setting (default `False`). "
-        "Use `keep` to not override.",
+        "How to set the Django DEBUG setting (default `False`). " "Use `keep` to not override.",
         default="False",
     )
     group.addoption(
@@ -308,9 +309,7 @@ def pytest_load_initial_conftests(
 
     if (
         options.itv
-        or _get_boolean_value(
-            os.environ.get(INVALID_TEMPLATE_VARS_ENV), INVALID_TEMPLATE_VARS_ENV
-        )
+        or _get_boolean_value(os.environ.get(INVALID_TEMPLATE_VARS_ENV), INVALID_TEMPLATE_VARS_ENV)
         or early_config.getini(INVALID_TEMPLATE_VARS_ENV)
     ):
         os.environ[INVALID_TEMPLATE_VARS_ENV] = "true"
@@ -370,6 +369,7 @@ def pytest_report_header(config: pytest.Config) -> list[str] | None:
 
     if "django" in sys.modules:
         import django
+
         report_header.insert(0, f"version: {django.get_version()}")
 
     if report_header:
@@ -534,6 +534,7 @@ def _django_setup_unittest(
     # Before pytest 5.4: https://github.com/pytest-dev/pytest/issues/5991
     # After pytest 5.4: https://github.com/pytest-dev/pytest-django/issues/824
     from _pytest.unittest import TestCaseFunction
+
     original_runtest = TestCaseFunction.runtest
 
     def non_debugging_runtest(self) -> None:
@@ -707,7 +708,7 @@ def _template_string_if_invalid_marker(
     request: pytest.FixtureRequest,
 ) -> None:
     """Apply the @pytest.mark.ignore_template_errors marker,
-     internal to pytest-django."""
+    internal to pytest-django."""
     marker = request.keywords.get("ignore_template_errors", None)
     if os.environ.get(INVALID_TEMPLATE_VARS_ENV, "false") == "true":
         if marker and django_settings_is_configured():

pytest_django_test/app/migrations/0001_initial.py

@@ -6,7 +6,6 @@
 
 
 class Migration(migrations.Migration):
-
     initial = True
 
     dependencies: tuple[tuple[str, str], ...] = ()

pytest_django_test/db_helpers.py

@@ -30,8 +30,8 @@
         # An explicit test db name was given, is that as the base name
         TEST_DB_NAME = f"{TEST_DB_NAME}_inner"
 
-    SECOND_DB_NAME = DB_NAME + '_second' if DB_NAME is not None else None
-    SECOND_TEST_DB_NAME = TEST_DB_NAME + '_second' if DB_NAME is not None else None
+    SECOND_DB_NAME = DB_NAME + "_second" if DB_NAME is not None else None
+    SECOND_TEST_DB_NAME = TEST_DB_NAME + "_second" if DB_NAME is not None else None
 
 
 def get_db_engine() -> str:
@@ -87,10 +87,7 @@ def run_mysql(*args: str) -> CmdResult:
 
 
 def skip_if_sqlite_in_memory() -> None:
-    if (
-        _settings["ENGINE"] == "django.db.backends.sqlite3"
-        and _settings["TEST"]["NAME"] is None
-    ):
+    if _settings["ENGINE"] == "django.db.backends.sqlite3" and _settings["TEST"]["NAME"] is None:
         pytest.skip("Do not test db reuse since database does not support it")
 
 
@@ -107,9 +104,7 @@ def drop_database(db_suffix: str | None = None) -> None:
 
     if db_engine == "postgresql":
         r = run_psql("postgres", "-c", f"DROP DATABASE {name}")
-        assert "DROP DATABASE" in force_str(
-            r.std_out
-        ) or "does not exist" in force_str(r.std_err)
+        assert "DROP DATABASE" in force_str(r.std_out) or "does not exist" in force_str(r.std_err)
         return
 
     if db_engine == "mysql":
@@ -136,8 +131,7 @@ def db_exists(db_suffix: str | None = None) -> bool:
         return r.status_code == 0
 
     assert db_engine == "sqlite3", f"{db_engine} cannot be tested properly!"
-    assert TEST_DB_NAME != ":memory:", (
-        "sqlite in-memory database cannot be checked for existence!")
+    assert TEST_DB_NAME != ":memory:", "sqlite in-memory database cannot be checked for existence!"
     return os.path.exists(name)
 
 
@@ -155,8 +149,7 @@ def mark_database() -> None:
         return
 
     assert db_engine == "sqlite3", f"{db_engine} cannot be tested properly!"
-    assert TEST_DB_NAME != ":memory:", (
-        "sqlite in-memory database cannot be marked!")
+    assert TEST_DB_NAME != ":memory:", "sqlite in-memory database cannot be marked!"
 
     conn = sqlite3.connect(TEST_DB_NAME)
     try:
@@ -180,8 +173,7 @@ def mark_exists() -> bool:
         return r.status_code == 0
 
     assert db_engine == "sqlite3", f"{db_engine} cannot be tested properly!"
-    assert TEST_DB_NAME != ":memory:", (
-        "sqlite in-memory database cannot be checked for mark!")
+    assert TEST_DB_NAME != ":memory:", "sqlite in-memory database cannot be checked for mark!"
 
     conn = sqlite3.connect(TEST_DB_NAME)
     try:

pytest_django_test/db_router.py

@@ -1,14 +1,14 @@
 class DbRouter:
     def db_for_read(self, model, **hints):
-        if model._meta.app_label == 'app' and model._meta.model_name == 'seconditem':
-            return 'second'
+        if model._meta.app_label == "app" and model._meta.model_name == "seconditem":
+            return "second"
         return None
 
     def db_for_write(self, model, **hints):
-        if model._meta.app_label == 'app' and model._meta.model_name == 'seconditem':
-            return 'second'
+        if model._meta.app_label == "app" and model._meta.model_name == "seconditem":
+            return "second"
         return None
 
     def allow_migrate(self, db, app_label, model_name=None, **hints):
-        if app_label == 'app' and model_name == 'seconditem':
-            return db == 'second'
+        if app_label == "app" and model_name == "seconditem":
+            return db == "second"

pytest_django_test/settings_base.py

@@ -28,6 +28,6 @@
     }
 ]
 
-DATABASE_ROUTERS = ['pytest_django_test.db_router.DbRouter']
+DATABASE_ROUTERS = ["pytest_django_test.db_router.DbRouter"]
 
 USE_TZ = True

pytest_django_test/urls_overridden.py

@@ -3,5 +3,5 @@
 
 
 urlpatterns = [
-    path("overridden_url/", lambda r: HttpResponse("Overridden urlconf works!"))
+    path("overridden_url/", lambda r: HttpResponse("Overridden urlconf works!")),
 ]

tests/conftest.py

@@ -20,7 +20,8 @@
 
 def pytest_configure(config: pytest.Config) -> None:
     config.addinivalue_line(
-        "markers", "django_project: options for the django_pytester fixture"
+        "markers",
+        "django_project: options for the django_pytester fixture",
     )
 
 
@@ -68,9 +69,8 @@ def django_pytester(
         db_settings["second"]["NAME"] = SECOND_DB_NAME
         db_settings["second"].setdefault("TEST", {})["NAME"] = SECOND_TEST_DB_NAME
 
-    test_settings = (
-        dedent(
-            """
+    test_settings = dedent(
+        """
         import django
 
         # Pypy compatibility
@@ -109,13 +109,11 @@ def django_pytester(
         ]
 
         %(extra_settings)s
-    """
-        )
-        % {
-            "db_settings": repr(db_settings),
-            "extra_settings": dedent(options["extra_settings"]),
-        }
-    )
+        """
+    ) % {
+        "db_settings": repr(db_settings),
+        "extra_settings": dedent(options["extra_settings"]),
+    }
 
     if options["project_root"]:
         project_root = pytester.mkdir(options["project_root"])

tests/test_asserts.py

@@ -19,17 +19,20 @@ def _get_actual_assertions_names() -> list[str]:
 
     from django.test import TestCase as DjangoTestCase
 
-    obj = DjangoTestCase('run')
+    obj = DjangoTestCase("run")
 
     def is_assert(func) -> bool:
-        return func.startswith('assert') and '_' not in func
+        return func.startswith("assert") and "_" not in func
 
-    base_methods = [name for name, member in
-                    inspect.getmembers(DefaultTestCase)
-                    if is_assert(name)]
+    base_methods = [
+        name for name, member in inspect.getmembers(DefaultTestCase) if is_assert(name)
+    ]
 
-    return [name for name, member in inspect.getmembers(obj)
-            if is_assert(name) and name not in base_methods]
+    return [
+        name
+        for name, member in inspect.getmembers(obj)
+        if is_assert(name) and name not in base_methods
+    ]
 
 
 def test_django_asserts_available() -> None:
@@ -47,11 +50,11 @@ def test_sanity() -> None:
 
     from pytest_django.asserts import assertContains, assertNumQueries
 
-    response = HttpResponse('My response')
+    response = HttpResponse("My response")
 
-    assertContains(response, 'My response')
+    assertContains(response, "My response")
     with pytest.raises(AssertionError):
-        assertContains(response, 'Not my response')
+        assertContains(response, "Not my response")
 
     assertNumQueries(0, lambda: 1 + 1)
     with assertNumQueries(0):