Test cases for access token generator. #42

lepture · Sep 15, 2013 · 521c13b · 521c13b
1 parent 14cd2bf
commit 521c13b
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/oauth2/test_oauth2.py b/tests/oauth2/test_oauth2.py
@@ -270,3 +270,19 @@ class TestCredentialAuthSQLAlchemy(TestCredentialAuth):
 
     def create_oauth_provider(self, app):
         return sqlalchemy_provider(app)
+
+
+class TestTokenGenerator(OAuthSuite):
+
+    def create_oauth_provider(self, app):
+
+        def generator(request, refresh_token=False):
+            return 'foobar'
+
+        app.config['OAUTH2_PROVIDER_TOKEN_GENERATOR'] = generator
+        return default_provider(app)
+
+    def test_get_access_token(self):
+        rv = self.client.post(authorize_url, data={'confirm': 'yes'})
+        rv = self.client.get(clean_url(rv.location))
+        assert 'foobar' in u(rv.data)