update tests

clinicedc · Jan 7, 2024 · 6fba11b · 6fba11b
1 parent a437bf2
commit 6fba11b
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 3 deletions.
diff --git a/edc_action_item/tests/models.py b/edc_action_item/tests/models.py
@@ -13,10 +13,19 @@
 )
 from edc_model.models import BaseUuidModel, HistoricalRecords
 from edc_sites.model_mixins import SiteModelMixin
+from edc_visit_schedule.model_mixins import OffScheduleModelMixin, OnScheduleModelMixin
 
 from ..models import ActionModelMixin
 
 
+class OnSchedule(SiteModelMixin, OnScheduleModelMixin, BaseUuidModel):
+    pass
+
+
+class OffSchedule(SiteModelMixin, OffScheduleModelMixin, BaseUuidModel):
+    pass
+
+
 class SubjectConsent(
     ConsentModelMixin,
     SiteModelMixin,

diff --git a/edc_action_item/tests/test_case_mixin.py b/edc_action_item/tests/test_case_mixin.py
@@ -30,7 +30,9 @@ def enroll(subject_identifier=None):
             dob=get_utcnow() - relativedelta(years=25),
         )
         RegisteredSubject.objects.create(subject_identifier=subject_identifier)
-        _, schedule = site_visit_schedules.get_by_onschedule_model("edc_metadata.onschedule")
+        _, schedule = site_visit_schedules.get_by_onschedule_model(
+            "edc_visit_schedule.onschedule"
+        )
         schedule.put_on_schedule(
             subject_identifier=subject_consent.subject_identifier,
             onschedule_datetime=subject_consent.consent_datetime,

diff --git a/edc_action_item/tests/visit_schedule.py b/edc_action_item/tests/visit_schedule.py
@@ -39,8 +39,8 @@
 
 schedule = Schedule(
     name="schedule",
-    onschedule_model="edc_metadata.onschedule",
-    offschedule_model="edc_metadata.offschedule",
+    onschedule_model="edc_visit_schedule.onschedule",
+    offschedule_model="edc_visit_schedule.offschedule",
     consent_model="edc_action_item.subjectconsent",
     appointment_model="edc_appointment.appointment",
 )