refactor: Migrate some tests to pytest

src/shared/tests/conftest.py

@@ -1,108 +1 @@
-import pytest
-
-from shared.models.cve import (
-    AffectedProduct,
-    Container,
-    CveRecord,
-    Description,
-    Metric,
-    Organization,
-    Version,
-)
-from shared.models.linkage import (
-    CVEDerivationClusterProposal,
-    DerivationClusterProposalLink,
-    ProvenanceFlags,
-)
-from shared.models.nix_evaluation import (
-    NixChannel,
-    NixDerivation,
-    NixDerivationMeta,
-    NixEvaluation,
-    NixMaintainer,
-)
-
-
-@pytest.fixture
-def cve(db: None) -> Container:
-    org = Organization.objects.create(uuid=1, short_name="test-org")
-    cve = CveRecord.objects.create(
-        cve_id="CVE-2025-0001",
-        assigner=org,
-    )
-    desc = Description.objects.create(value="Test description")
-    metric = Metric.objects.create(format="cvssV3_1", raw_cvss_json={})
-    version = Version.objects.create(status=Version.Status.AFFECTED, version="1.0")
-    affected = AffectedProduct.objects.create(package_name="dummy-package")
-    affected.versions.add(version)
-
-    container = cve.container.create(provider=org, title="Dummy Title")
-    container.affected.add(affected)
-    container.descriptions.add(desc)
-    container.metrics.add(metric)
-
-    return container
-
-
-@pytest.fixture
-def evaluation(db: None) -> NixEvaluation:
-    channel = NixChannel.objects.create(
-        staging_branch="release-24.05",
-        channel_branch="release-24.05",
-        head_sha1_commit="deadbeef",
-        state=NixChannel.ChannelState.STABLE,
-        release_version="24.05",
-        repository="https://github.com/NixOS/nixpkgs",
-    )
-
-    evaluation = NixEvaluation.objects.create(
-        channel=channel,
-        commit_sha1="deadbeef",
-        state="completed",
-    )
-    return evaluation
-
-
-@pytest.fixture
-def drv(db: None, evaluation: NixEvaluation) -> NixDerivation:
-    maintainer = NixMaintainer.objects.create(
-        github_id=123, github="testuser", name="Test User", email="test@example.com"
-    )
-
-    meta = NixDerivationMeta.objects.create(
-        description="First dummy derivation",
-        insecure=False,
-        available=True,
-        broken=False,
-        unfree=False,
-        unsupported=False,
-    )
-
-    meta.maintainers.add(maintainer)
-
-    drv = NixDerivation.objects.create(
-        attribute="foo",
-        derivation_path="/nix/store/<hash>-foo.drv",
-        name="foo-1.0",
-        metadata=meta,
-        system="x86_64-linux",
-        parent_evaluation=evaluation,
-    )
-
-    return drv
-
-
-@pytest.fixture
-def suggestion(cve: Container, drv: NixDerivation) -> CVEDerivationClusterProposal:
-    suggestion = CVEDerivationClusterProposal.objects.create(
-        status="pending",
-        cve_id=cve.pk,
-    )
-
-    DerivationClusterProposalLink.objects.create(
-        proposal=suggestion,
-        derivation=drv,
-        provenance_flags=ProvenanceFlags.PACKAGE_NAME_MATCH,
-    )
-
-    return suggestion
+pytest_plugins = ["shared.tests.fixtures"]

src/shared/tests/fixtures.py

@@ -0,0 +1,143 @@
+import pytest
+from allauth.socialaccount.models import SocialAccount
+from django.contrib.auth.models import AbstractBaseUser
+from django.test import Client
+
+from shared.listeners.cache_suggestions import cache_new_suggestions
+from shared.models.cve import (
+    AffectedProduct,
+    Container,
+    CveRecord,
+    Description,
+    Metric,
+    Organization,
+    Version,
+)
+from shared.models.linkage import (
+    CVEDerivationClusterProposal,
+    DerivationClusterProposalLink,
+    ProvenanceFlags,
+)
+from shared.models.nix_evaluation import (
+    NixChannel,
+    NixDerivation,
+    NixDerivationMeta,
+    NixEvaluation,
+    NixMaintainer,
+)
+
+
+@pytest.fixture
+def cve(db: None) -> Container:
+    org = Organization.objects.create(uuid=1, short_name="test-org")
+    cve = CveRecord.objects.create(
+        cve_id="CVE-2025-0001",
+        assigner=org,
+    )
+    desc = Description.objects.create(value="Test description")
+    metric = Metric.objects.create(format="cvssV3_1", raw_cvss_json={})
+    version = Version.objects.create(status=Version.Status.AFFECTED, version="1.0")
+    affected = AffectedProduct.objects.create(package_name="dummy-package")
+    affected.versions.add(version)
+
+    container = cve.container.create(provider=org, title="Dummy Title")
+    container.affected.add(affected)
+    container.descriptions.add(desc)
+    container.metrics.add(metric)
+
+    return container
+
+
+@pytest.fixture
+def evaluation(db: None) -> NixEvaluation:
+    channel = NixChannel.objects.create(
+        staging_branch="release-24.05",
+        channel_branch="release-24.05",
+        head_sha1_commit="deadbeef",
+        state=NixChannel.ChannelState.STABLE,
+        release_version="24.05",
+        repository="https://github.com/NixOS/nixpkgs",
+    )
+
+    evaluation = NixEvaluation.objects.create(
+        channel=channel,
+        commit_sha1="deadbeef",
+        state="completed",
+    )
+    return evaluation
+
+
+@pytest.fixture
+def drv(db: None, evaluation: NixEvaluation) -> NixDerivation:
+    maintainer = NixMaintainer.objects.create(
+        github_id=123, github="testuser", name="Test User", email="test@example.com"
+    )
+
+    meta = NixDerivationMeta.objects.create(
+        description="First dummy derivation",
+        insecure=False,
+        available=True,
+        broken=False,
+        unfree=False,
+        unsupported=False,
+    )
+
+    meta.maintainers.add(maintainer)
+
+    drv = NixDerivation.objects.create(
+        attribute="foo",
+        derivation_path="/nix/store/<hash>-foo.drv",
+        name="foo-1.0",
+        metadata=meta,
+        system="x86_64-linux",
+        parent_evaluation=evaluation,
+    )
+
+    return drv
+
+
+@pytest.fixture
+def suggestion(cve: Container, drv: NixDerivation) -> CVEDerivationClusterProposal:
+    suggestion = CVEDerivationClusterProposal.objects.create(
+        status="pending",
+        cve=cve.cve,
+    )
+
+    DerivationClusterProposalLink.objects.create(
+        proposal=suggestion,
+        derivation=drv,
+        provenance_flags=ProvenanceFlags.PACKAGE_NAME_MATCH,
+    )
+
+    return suggestion
+
+
+@pytest.fixture
+def cached_suggestion(
+    suggestion: CVEDerivationClusterProposal,
+) -> CVEDerivationClusterProposal:
+    cache_new_suggestions(suggestion)
+
+    return suggestion
+
+
+@pytest.fixture
+def authenticated_client(
+    client: Client, django_user_model: type[AbstractBaseUser]
+) -> Client:
+    user = django_user_model.objects.create_user(
+        username="testuser",
+        is_staff=True,
+    )
+    SocialAccount.objects.get_or_create(
+        user=user,
+        provider="github",
+        uid="123456",
+        extra_data={"login": user.username},
+    )
+    # https://docs.djangoproject.com/en/6.0/topics/testing/tools/#django.test.Client.force_login
+    client.force_login(
+        user,
+        backend="allauth.account.auth_backends.AuthenticationBackend",
+    )
+    return client

src/webview/tests/conftest.py

@@ -0,0 +1 @@
+pytest_plugins = ["shared.tests.fixtures"]