4
4
from oauthlib .oauth2 .rfc6749 .errors import MismatchingStateError , InvalidGrantError
5
5
from django .test .client import RequestFactory
6
6
from django .core .urlresolvers import reverse
7
+ import oauthadmin .views
7
8
8
9
9
10
SESSION_VARIABLES = ['oauth_state' , 'oauth_token' , 'uid' , 'user' ]
@@ -41,7 +42,7 @@ def test_login(app_setting, OAuth2Session, request_factory):
41
42
OAuth2Session .return_value = mock .Mock (
42
43
authorization_url = mock .Mock (return_value = ('https://foo' , 'state-variable' ))
43
44
)
44
- request = request_factory .get (reverse (' oauthadmin.views.login' ))
45
+ request = request_factory .get (reverse (oauthadmin .views .login ))
45
46
request .session = {}
46
47
request .build_absolute_uri = mock .Mock (return_value = 'https://test.com/construct-redirect' )
47
48
@@ -57,7 +58,7 @@ def test_login_redirect_uri(OAuth2Session, request_factory):
57
58
OAuth2Session .return_value = mock .Mock (
58
59
authorization_url = mock .Mock (return_value = ('https://foo' , 'state-variable' ))
59
60
)
60
- request = request_factory .get (reverse (' oauthadmin.views.login' ))
61
+ request = request_factory .get (reverse (oauthadmin .views .login ))
61
62
request .session = {}
62
63
request .build_absolute_uri = mock .Mock (return_value = 'https://test.com/construct-redirect' )
63
64
@@ -74,7 +75,7 @@ def test_login_redirect_uri_with_next_from_url(OAuth2Session, request_factory):
74
75
OAuth2Session .return_value = mock .Mock (
75
76
authorization_url = mock .Mock (return_value = ('https://foo' , 'state-variable' ))
76
77
)
77
- request = request_factory .get (reverse (' oauthadmin.views.login' ) + '?next=/admin/content/' )
78
+ request = request_factory .get (reverse (oauthadmin .views .login ) + '?next=/admin/content/' )
78
79
request .session = {}
79
80
request .build_absolute_uri = mock .Mock (return_value = 'https://test.com/construct-redirect' )
80
81
@@ -143,7 +144,7 @@ def test_callback_with_invalid_grant(import_by_path, app_setting, OAuth2Session,
143
144
@mock .patch ('oauthadmin.views.app_setting' )
144
145
@mock .patch ('oauthadmin.views.import_by_path' )
145
146
def test_callback (import_by_path , app_setting , OAuth2Session , request_factory ):
146
- request = request_factory .get (reverse (' oauthadmin.views.callback' ))
147
+ request = request_factory .get (reverse (oauthadmin .views .callback ))
147
148
request .session = {'oauth_state' : 'state-variable' }
148
149
OAuth2Session .return_value = mock .Mock (
149
150
fetch_token = mock .Mock (return_value = 'token' )
@@ -163,7 +164,7 @@ def test_callback(import_by_path, app_setting, OAuth2Session, request_factory):
163
164
@mock .patch ('oauthadmin.views.app_setting' )
164
165
@mock .patch ('oauthadmin.views.import_by_path' )
165
166
def test_callback_redirect_to_next (import_by_path , app_setting , OAuth2Session , request_factory ):
166
- request = request_factory .get (reverse (' oauthadmin.views.callback' ) + '?next=/admin/content/' )
167
+ request = request_factory .get (reverse (oauthadmin .views .callback ) + '?next=/admin/content/' )
167
168
request .session = {'oauth_state' : 'state-variable' }
168
169
OAuth2Session .return_value = mock .Mock (
169
170
fetch_token = mock .Mock (return_value = 'token' )
0 commit comments