@@ -348,10 +348,6 @@ def setUpTestData(cls):
348
348
cls .main_user = baker .make (UserProfile , _fill_optional = ["email" ])
349
349
cls .other_user = baker .make (UserProfile , _fill_optional = ["email" ])
350
350
351
- # The merge candidate is created first, so the account is older.
352
- cls .
suggested_merge_candidate = baker .
make (
UserProfile ,
email = "[email protected] " )
353
- cls .
suggested_main_user = baker .
make (
UserProfile ,
email = "[email protected] " )
354
-
355
351
def test_redirection_user_merge_view (self ):
356
352
page = self .app .get (self .url , user = self .manager )
357
353
@@ -364,15 +360,19 @@ def test_redirection_user_merge_view(self):
364
360
self .assertContains (page , self .main_user .email )
365
361
self .assertContains (page , self .other_user .email )
366
362
363
+ @override_settings (INSTITUTION_EMAIL_DOMAINS = ["institution.example.com" , "student.institution.example.com" ])
367
364
def test_suggested_merge (self ):
365
+ suggested_merge_candidate = baker .
make (
UserProfile ,
email = "[email protected] " )
366
+ suggested_main_user = baker .
make (
UserProfile ,
email = "[email protected] " )
367
+
368
+ # Ensure that the merge candidate has the lower pk / seems older
369
+ if suggested_merge_candidate .pk > suggested_main_user .pk :
370
+ suggested_merge_candidate , suggested_main_user = suggested_main_user , suggested_merge_candidate
371
+
368
372
page = self .app .get (self .url , user = self .manager )
369
373
370
- expected_url = reverse (
371
- "staff:user_merge" , args = [self .suggested_main_user .id , self .suggested_merge_candidate .id ]
372
- )
373
- unexpected_url = reverse (
374
- "staff:user_merge" , args = [self .suggested_merge_candidate .id , self .suggested_main_user .id ]
375
- )
374
+ expected_url = reverse ("staff:user_merge" , args = [suggested_main_user .pk , suggested_merge_candidate .pk ])
375
+ unexpected_url = reverse ("staff:user_merge" , args = [suggested_merge_candidate .pk , suggested_main_user .pk ])
376
376
377
377
self .assertContains (page , f'<a href="{ expected_url } "' )
378
378
self .assertNotContains (page , f'<a href="{ unexpected_url } "' )
0 commit comments