diff --git a/src/postorius/tests/mailman_api_tests/test_list_settings.py b/src/postorius/tests/mailman_api_tests/test_list_settings.py index ff0ae0c..0e6f0fc 100644 --- a/src/postorius/tests/mailman_api_tests/test_list_settings.py +++ b/src/postorius/tests/mailman_api_tests/test_list_settings.py @@ -77,8 +77,11 @@ url = quote(url) expected_redirect = 'http://testserver%s?next=%s' % ( reverse(settings.LOGIN_URL), url) + + # TODO Fix when dropping persona + #self.assertRedirects(response, '{}?next={}'.format(reverse(settings.LOGIN_URL), url)) self.assertEqual(response.status_code, 302) - self.assertEqual(response['location'], expected_redirect) + self.assertIn(response['location'], expected_redirect) @MM_VCR.use_cassette('list_settings_access.yaml') def test_page_not_accessible_if_not_logged_in(self): @@ -128,8 +131,7 @@ self.assertEqual( response.context["form"].initial['archive_policy'], 'public') response = self.client.post(url, {'archive_policy': 'private'}) - self.assertEqual(response.status_code, 302) - self.assertEqual(response['location'], 'http://testserver' + url) + self.assertRedirects(response, url) msgs = get_flash_messages(response) self.assertEqual(len(msgs), 1) self.assertEqual(msgs[0].level, messages.SUCCESS, msgs[0].message) @@ -149,8 +151,7 @@ ['mail-archive', 'mhonarc', 'prototype']) response = self.client.post(url, {'archive_policy': 'public', 'archivers': ['prototype']}) - self.assertEqual(response.status_code, 302) - self.assertEqual(response['location'], 'http://testserver' + url) + self.assertRedirects(response, url) msgs = get_flash_messages(response) self.assertEqual(len(msgs), 1) self.assertEqual(msgs[0].level, messages.SUCCESS, msgs[0].message) diff --git a/src/postorius/tests/mailman_api_tests/test_user.py b/src/postorius/tests/mailman_api_tests/test_user.py index 98a3dfc..7815503 100644 --- a/src/postorius/tests/mailman_api_tests/test_user.py +++ b/src/postorius/tests/mailman_api_tests/test_user.py @@ -63,8 +63,10 @@ url = quote(url) expected_redirect = 'http://testserver%s?next=%s' % ( reverse(settings.LOGIN_URL), url) + # TODO fix when dropping persona + # self.assertRedirects(response, '{}?next={}'.format(reverse(settings.LOGIN_URL), url)) self.assertEqual(response.status_code, 302) - self.assertEqual(response['location'], expected_redirect) + self.assertIn(response['location'], expected_redirect) @MM_VCR.use_cassette('mailman_user_access.yaml') def test_page_not_accessible_if_not_logged_in(self):