update tests

edc_consent/tests/test_case_mixins.py

@@ -0,0 +1,60 @@
+from __future__ import annotations
+
+from datetime import date, datetime
+from typing import TYPE_CHECKING
+
+from dateutil.relativedelta import relativedelta
+from django.conf import settings
+from django.contrib.sites.models import Site
+from edc_constants.constants import FEMALE
+from edc_utils import get_utcnow, get_utcnow_as_date
+from model_bakery import baker
+
+from edc_consent.utils import get_consent_model_name
+
+if TYPE_CHECKING:
+    from edc_screening.tests.models import SubjectScreening
+
+
+class ConsentTestCaseMixin:
+    @staticmethod
+    def get_subject_consent(
+        subject_screening: SubjectScreening | None = None,
+        consent_datetime: datetime | None = None,
+        site_id: int | None = None,
+        dob: date | None = None,
+        screening_identifier: str | None = None,
+        gender: str | None = None,
+        initials: str | None = None,
+        age_in_years: int | None = None,
+        report_datetime: datetime | None = None,
+    ):
+        now = get_utcnow()
+        now_as_date = get_utcnow_as_date()
+        screening_identifier = (
+            subject_screening.screening_identifier
+            if subject_screening
+            else (screening_identifier or "ABCDEF")
+        )
+        initials = subject_screening.initials if subject_screening else (initials or "XX")
+        gender = subject_screening.gender if subject_screening else (gender or FEMALE)
+        age_in_years = (
+            subject_screening.age_in_years if subject_screening else (age_in_years or 25)
+        )
+        dob = dob or (now_as_date - relativedelta(years=age_in_years))
+        report_datetime = (
+            subject_screening.report_datetime if subject_screening else report_datetime or now
+        )
+        consent_datetime = consent_datetime or report_datetime
+
+        return baker.make_recipe(
+            get_consent_model_name(),
+            user_created="erikvw",
+            user_modified="erikvw",
+            screening_identifier=screening_identifier,
+            initials=initials,
+            gender=gender,
+            dob=dob,
+            site=Site.objects.get(id=site_id or settings.SITE_ID),
+            consent_datetime=consent_datetime,
+        )