tested OAuthLibCoreBackend's _get_extra_credentials method overriding

Author: outsbart

oauth2_provider/oauth2_backends.py

@@ -43,7 +43,6 @@ def _get_extra_credentials(self, request):
         """
         return None
 
-
     def _extract_params(self, request):
         """
         Extract parameters from the Django request object. Such parameters will then be passed to

oauth2_provider/tests/test_oauth2_backends.py

@@ -1,4 +1,5 @@
 import json
+import mock
 
 from django.test import TestCase, RequestFactory
 
@@ -34,6 +35,33 @@ def test_application_json_extract_params(self):
         self.assertNotIn("password=123456", body)
 
 
+class TestCustomOAuthLibCoreBackend(TestCase):
+    """
+    Tests that the public API behaves as expected when we override
+    the OAuthLibCoreBackend core methods.
+    """
+    class MyOAuthLibCore(OAuthLibCore):
+        def _get_extra_credentials(self, request):
+            return 1
+
+    def setUp(self):
+        self.factory = RequestFactory()
+
+    def test_create_token_response_gets_extra_credentials(self):
+        """
+        Make sures that extra_credentials parameter is passed to oauthlib
+        """
+        payload = "grant_type=password&username=john&password=123456"
+        request = self.factory.post("/o/token/", payload, content_type="application/x-www-form-urlencoded")
+
+        with mock.patch('oauthlib.oauth2.Server.create_token_response') as create_token_response:
+            mocked = mock.MagicMock()
+            create_token_response.return_value = mocked, mocked, mocked
+            core = self.MyOAuthLibCore()
+            core.create_token_response(request)
+            self.assertTrue(create_token_response.call_args[0][4] == 1)
+
+
 class TestJSONOAuthLibCoreBackend(TestCase):
     def setUp(self):
         self.factory = RequestFactory()