check if data provenance is provided when changing type of study subject

smash/web/tests/view/test_subjects.py

@@ -5,12 +5,12 @@ from django.contrib.auth.models import Permission
 from django.core.files.uploadedfile import SimpleUploadedFile
 from django.urls import reverse
 
-from web.models.custom_data import CustomStudySubjectField
-from web.models.custom_data.custom_study_subject_field import get_study_subject_field_id
 from web.forms import SubjectAddForm, SubjectEditForm, StudySubjectAddForm, StudySubjectEditForm
-from web.models import MailTemplate, StudySubject, StudyColumns, Visit
+from web.models import MailTemplate, StudySubject, StudyColumns, Visit, Provenance
 from web.models.constants import SEX_CHOICES_MALE, SUBJECT_TYPE_CHOICES_CONTROL, SUBJECT_TYPE_CHOICES_PATIENT, \
     COUNTRY_AFGHANISTAN_ID, COUNTRY_OTHER_ID, MAIL_TEMPLATE_CONTEXT_SUBJECT, CUSTOM_FIELD_TYPE_FILE
+from web.models.custom_data import CustomStudySubjectField
+from web.models.custom_data.custom_study_subject_field import get_study_subject_field_id
 from web.tests import LoggedInWithWorkerTestCase
 from web.tests.functions import create_study_subject, create_visit, create_appointment, get_test_location, \
     create_language, get_resource_path, get_test_study, format_form_field
@@ -140,7 +140,7 @@ class SubjectsViewTests(LoggedInWithWorkerTestCase):
         self.assertEqual(response.status_code, 302)
         self.assertTrue("edit" in response.url)
 
-    def create_edit_form_data_for_study_subject(self, instance=None):
+    def create_edit_form_data_for_study_subject(self, instance: StudySubject = None):
         if instance is None:
             instance = self.study_subject
         form_study_subject = StudySubjectEditForm(instance=instance, prefix="study_subject")
@@ -303,3 +303,15 @@ class SubjectsViewTests(LoggedInWithWorkerTestCase):
 
         response = self.client.get(reverse('web.views.subject_require_contact'))
         self.assertEqual(response.status_code, 200)
+
+    def test_save_subject_edit_when_type_changed(self):
+        form_data = self.create_edit_form_data_for_study_subject(self.study_subject)
+
+        count = Provenance.objects.all().count()
+        form_data["study_subject-type"] = SUBJECT_TYPE_CHOICES_PATIENT
+        response = self.client.post(
+            reverse('web.views.subject_edit', kwargs={'id': self.study_subject.id}), data=form_data)
+
+        self.assertEqual(response.status_code, 302)
+
+        self.assertEqual(count + 1, Provenance.objects.all().count())