Add additional token auth test

miguelgrinberg · Mar 13, 2016 · 29edb19 · 29edb19
1 parent 96b5764
commit 29edb19
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/test_token.py b/tests/test_token.py
@@ -72,6 +72,14 @@ def test_token_auth_login_invalid_scheme(self):
         self.assertEqual(response.headers['WWW-Authenticate'],
                          'MyToken realm="Foo"')
 
+    def test_token_auth_login_invalid_header(self):
+        response = self.client.get(
+            '/protected', headers={'Authorization': 'this-is-a-bad-header'})
+        self.assertEqual(response.status_code, 401)
+        self.assertTrue('WWW-Authenticate' in response.headers)
+        self.assertEqual(response.headers['WWW-Authenticate'],
+                         'MyToken realm="Foo"')
+
     def test_token_auth_login_invalid_no_callback(self):
         token_auth2 = HTTPTokenAuth('Token', realm='foo')