From e1387ba7bc157fc8c06288779f7d8019a6d00ef5 Mon Sep 17 00:00:00 2001
From: Piotr Gawron <piotr.gawron@uni.lu>
Date: Wed, 29 Nov 2017 16:58:00 +0100
Subject: [PATCH] test for correct redirection when save and continue clicked
 on subject edit

---
 smash/web/tests/view/test_subjects.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/smash/web/tests/view/test_subjects.py b/smash/web/tests/view/test_subjects.py
index 4617ccab..72f070dc 100644
--- a/smash/web/tests/view/test_subjects.py
+++ b/smash/web/tests/view/test_subjects.py
@@ -67,6 +67,16 @@ class SubjectsViewTests(LoggedInWithWorkerTestCase):
         self.assertTrue(updated_study_subject.subject.dead)
         self.assertTrue(updated_study_subject.resigned)
 
+    def test_save_subject_edit_with_continue(self):
+        form_data = self.create_edit_form_data_for_study_subject()
+        form_data['_continue'] = True
+
+        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.assertTrue("edit" in response.url)
+
     def create_edit_form_data_for_study_subject(self):
         form_study_subject = StudySubjectEditForm(instance=self.study_subject, prefix="study_subject")
         form_subject = SubjectEditForm(instance=self.study_subject.subject, prefix="subject")
-- 
GitLab