|
5 | 5 |
|
6 | 6 | import unittest |
7 | 7 | from unittest.mock import Mock |
| 8 | +from urllib.parse import urlparse |
8 | 9 |
|
9 | 10 | import ddt |
10 | 11 | from django.conf import settings |
@@ -255,21 +256,21 @@ def _post_clear_preview_lang(self): |
255 | 256 | """ |
256 | 257 | return self.client.post('/update_lang/', {'action': 'reset_preview_language'}) |
257 | 258 |
|
258 | | - def test_preview_lang_with_dark_language(self): |
| 259 | + def test_preview_lang_with_dark_language_redirect(self): |
259 | 260 | response = self._post_set_preview_lang('unrel') |
260 | 261 |
|
261 | | - # Assert redirect happened |
262 | | - self.assertEqual(response.status_code, 302) |
263 | | - self.assertEqual(response.url, f'{settings.LMS_ROOT_URL}/update_lang/') |
| 262 | + # Assert redirect happened back to the same path |
| 263 | + assert response.status_code == 302 |
| 264 | + assert urlparse(response.url).path == '/update_lang/' |
264 | 265 |
|
265 | 266 | # Test clear + set flow |
266 | 267 | response = self._post_clear_preview_lang() |
267 | | - self.assertEqual(response.status_code, 302) |
268 | | - self.assertEqual(response.url, f'{settings.LMS_ROOT_URL}/update_lang/') |
| 268 | + assert response.status_code == 302 |
| 269 | + assert urlparse(response.url).path == '/update_lang/' |
269 | 270 |
|
270 | 271 | response = self._post_set_preview_lang('unrel') |
271 | | - self.assertEqual(response.status_code, 302) |
272 | | - self.assertEqual(response.url, f'{settings.LMS_ROOT_URL}/update_lang/') |
| 272 | + assert response.status_code == 302 |
| 273 | + assert urlparse(response.url).path == '/update_lang/' |
273 | 274 |
|
274 | 275 | def test_accept_chinese_language_codes(self): |
275 | 276 | DarkLangConfig( |
|
0 commit comments