@@ -15,6 +15,8 @@ def setup_method(self):
1515 self .request .GET = {'next' : self .next_url }
1616 self .request .POST = {'next' : self .next_url }
1717 self .empty_request = HttpRequest ()
18+ self .external_request = HttpRequest ()
19+ self .external_request .GET = {'next' : 'https://test.com' }
1820 self .adapter = AccountAdapter ()
1921 self .social_adapter = SocialAccountAdapter ()
2022
@@ -28,6 +30,10 @@ def test_get_login_redirect_url_no_next(self):
2830 assert '/user' == self .adapter \
2931 .get_login_redirect_url (self .empty_request )
3032
33+ def test_get_login_redirect_url_external (self ):
34+ assert '/user' == self .adapter \
35+ .get_login_redirect_url (self .external_request )
36+
3137 def test_get_logout_redirect_url (self ):
3238 assert self .next_url == self .adapter \
3339 .get_logout_redirect_url (self .request )
@@ -36,6 +42,10 @@ def test_get_logout_redirect_url_no_next(self):
3642 assert '/' == self .adapter \
3743 .get_logout_redirect_url (self .empty_request )
3844
45+ def test_get_logout_redirect_url_external (self ):
46+ assert '/' == self .adapter \
47+ .get_logout_redirect_url (self .external_request )
48+
3949
4050class TestSocialAccountAdapter (UserAdapterTestBase ):
4151 def setup_method (self ):
@@ -45,9 +55,16 @@ def setup_method(self):
4555 )
4656
4757 def test_get_connect_redirect_url (self ):
48- assert self .next_url == self .social_adapter \
49- .get_connect_redirect_url (self .request , self .social_account )
58+ assert self .next_url == self .social_adapter .get_connect_redirect_url (
59+ self .request , self .social_account
60+ )
5061
5162 def test_get_connect_redirect_url_no_next (self ):
52- assert '/user' == self .social_adapter \
53- .get_connect_redirect_url (self .empty_request , self .social_account )
63+ assert '/user' == self .social_adapter .get_connect_redirect_url (
64+ self .empty_request , self .social_account
65+ )
66+
67+ def test_get_connect_redirect_url_external (self ):
68+ assert '/user' == self .social_adapter .get_connect_redirect_url (
69+ self .external_request , self .social_account
70+ )
0 commit comments