diff --git a/smash/web/tests/functions.py b/smash/web/tests/functions.py index a3d0b3b4917115ed3405df49ed3ca53ebf35f372..7f158d8b08e6fac59f6b29646504d00fe1efff64 100644 --- a/smash/web/tests/functions.py +++ b/smash/web/tests/functions.py @@ -6,7 +6,7 @@ from django.contrib.auth.models import User from web.models import Location, AppointmentType, StudySubject, Worker, Visit, Appointment, ConfigurationItem, \ Language, ContactAttempt, FlyingTeam, Availability, Subject, Study, StudyColumns, StudyNotificationParameters, \ - VoucherType, VoucherTypePrice, Voucher, Room, Item, WorkerStudyRole + VoucherType, VoucherTypePrice, Voucher, Room, Item, WorkerStudyRole, StudyRedCapColumns from web.models.constants import REDCAP_TOKEN_CONFIGURATION_TYPE, REDCAP_BASE_URL_CONFIGURATION_TYPE, \ SEX_CHOICES_MALE, SUBJECT_TYPE_CHOICES_CONTROL, CONTACT_TYPES_PHONE, \ MONDAY_AS_DAY_OF_WEEK, COUNTRY_AFGHANISTAN_ID, VOUCHER_STATUS_NEW, GLOBAL_STUDY_ID, DEFAULT_LOCALE_NAME @@ -68,7 +68,8 @@ def create_empty_study_columns(): def create_study(name="test"): study_columns = StudyColumns.objects.create() notification_parameters = StudyNotificationParameters.objects.create() - return Study.objects.create(name=name, columns=study_columns, notification_parameters=notification_parameters) + redcap_columns = StudyRedCapColumns.objects.create() + return Study.objects.create(name=name, columns=study_columns, notification_parameters=notification_parameters, redcap_columns=redcap_columns) TEST_ID_COUNTER = 0