From 0eec7bd70716b1d61d0117db437435087ad4404e Mon Sep 17 00:00:00 2001
From: Sander Roosendaal <roosendaalsander@gmail.com>
Date: Wed, 31 Mar 2021 11:01:26 +0200
Subject: [PATCH] updating new user registration test

---
 rowers/tests/test_newusers.py | 34 +++++++++++++++++++++++++++-------
 1 file changed, 27 insertions(+), 7 deletions(-)

diff --git a/rowers/tests/test_newusers.py b/rowers/tests/test_newusers.py
index bdfae610..3c53115a 100644
--- a/rowers/tests/test_newusers.py
+++ b/rowers/tests/test_newusers.py
@@ -42,20 +42,38 @@ class NewUserRegistrationTest(TestCase):
         self.assertTrue(form.is_valid())
 
         response = self.c.post('/rowers/register/', form_data, follow=True)
+        self.assertEqual(response.status_code,200)
 
-        self.assertRedirects(response,
-                             expected_url='/rowers/me/gdpr-optin/?next=/rowers/list-workouts/',
-                             status_code=302,target_status_code=200)
+        # set opt-in
+        user = User.objects.get(username='janderoeiert')
+        user.rower.gdpr_optin = True
+        user.set_password('aapindewei2')
+        user.is_active = True
+        user.save()
+        user.rower.save()
+
+        login = self.c.login(username=user.username,password='aapindewei2')
+        self.assertTrue(login)
 
 
-        url = '/rowers/me/gdpr-optin-confirm/?next=/rowers/list-workouts/'
+        url = '/rowers/list-workouts/'
 
         response = self.c.get(url)
 
+        expected = '/rowers/me/gdpr-optin/?next=/rowers/list-workouts/'
+
         self.assertRedirects(response,
-                             expected_url='/rowers/list-workouts/',
-                             status_code=302,target_status_code=200,
-                             )
+                            expected_url=expected,
+                            status_code=302,target_status_code=200)
+
+        url = '/rowers/me/gdpr-optin-confirm/?next=/rowers/list-workouts/'
+        response = self.c.get(url)
+
+        expected = '/rowers/list-workouts/'
+        self.assertRedirects(response,
+                            expected_url=expected,
+                            status_code=302,target_status_code=200)
+
 
         url = '/rowers/exportallworkouts/'
 
@@ -74,6 +92,8 @@ class NewUserRegistrationTest(TestCase):
 
         self.assertTrue(response.status_code,200)
 
+
+
         url = '/rowers/me/delete/'
 
         form_data = {