@@ -88,8 +88,11 @@ def setUp(self):
88
88
super (TestOAuth2Middleware , self ).setUp ()
89
89
self .anon_user = AnonymousUser ()
90
90
91
+ def dummy_get_response (request ):
92
+ return None
93
+
91
94
def test_middleware_wrong_headers (self ):
92
- m = OAuth2TokenMiddleware ()
95
+ m = OAuth2TokenMiddleware (self . dummy_get_response )
93
96
request = self .factory .get ("/a-resource" )
94
97
self .assertIsNone (m .process_request (request ))
95
98
auth_headers = {
@@ -99,7 +102,7 @@ def test_middleware_wrong_headers(self):
99
102
self .assertIsNone (m .process_request (request ))
100
103
101
104
def test_middleware_user_is_set (self ):
102
- m = OAuth2TokenMiddleware ()
105
+ m = OAuth2TokenMiddleware (self . dummy_get_response )
103
106
auth_headers = {
104
107
"HTTP_AUTHORIZATION" : "Bearer " + "tokstr" ,
105
108
}
@@ -110,7 +113,7 @@ def test_middleware_user_is_set(self):
110
113
self .assertIsNone (m .process_request (request ))
111
114
112
115
def test_middleware_success (self ):
113
- m = OAuth2TokenMiddleware ()
116
+ m = OAuth2TokenMiddleware (self . dummy_get_response )
114
117
auth_headers = {
115
118
"HTTP_AUTHORIZATION" : "Bearer " + "tokstr" ,
116
119
}
@@ -119,7 +122,7 @@ def test_middleware_success(self):
119
122
self .assertEqual (request .user , self .user )
120
123
121
124
def test_middleware_response (self ):
122
- m = OAuth2TokenMiddleware ()
125
+ m = OAuth2TokenMiddleware (self . dummy_get_response )
123
126
auth_headers = {
124
127
"HTTP_AUTHORIZATION" : "Bearer " + "tokstr" ,
125
128
}
@@ -129,7 +132,7 @@ def test_middleware_response(self):
129
132
self .assertIs (response , processed )
130
133
131
134
def test_middleware_response_header (self ):
132
- m = OAuth2TokenMiddleware ()
135
+ m = OAuth2TokenMiddleware (self . dummy_get_response )
133
136
auth_headers = {
134
137
"HTTP_AUTHORIZATION" : "Bearer " + "tokstr" ,
135
138
}
0 commit comments