expose auth responses

Author: kennethkoontz

AUTHORS.md

@@ -33,4 +33,5 @@ Add yourself as a contributor!
 * [Kieron Briggs](https://github.com/kbriggs)
 * [Tatsuji Tsuchiya](https://github.com/ta2xeo)
 * [Chris McGraw](https://github.com/mitgr81)
+* [Ken Koontz](https://github.com/kennethkoontz)
 * (your name here)

rauth/service.py

@@ -156,6 +156,10 @@ def __init__(self,
         #: Object used to construct signatures with.
         self.signature_obj = signature_obj
 
+        #: Request and access token responses.
+        self.request_token_response = None
+        self.access_token_response = None
+
         super(OAuth1Service, self).__init__(name,
                                             base_url,
                                             authorize_url)
@@ -206,7 +210,10 @@ def get_raw_request_token(self, method='GET', **kwargs):
             raise TypeError('request_token_url must not be None')
 
         session = self.get_session()
-        return session.request(method, self.request_token_url, **kwargs)
+        self.request_token_response = session.request(method,
+                                                      self.request_token_url,
+                                                      **kwargs)
+        return self.request_token_response
 
     def get_request_token(self,
                           method='GET',
@@ -279,7 +286,10 @@ def get_raw_access_token(self,
             raise TypeError('access_token_url must not be None')
 
         session = self.get_session((request_token, request_token_secret))
-        return session.request(method, self.access_token_url, **kwargs)
+        self.access_token_response = session.request(method,
+                                                     self.access_token_url,
+                                                     **kwargs)
+        return self.access_token_response
 
     def get_access_token(self,
                          request_token,
@@ -347,7 +357,14 @@ def get_auth_session(self,
                                       request_token_secret,
                                       method=method,
                                       **kwargs)
-        return self.get_session(token)
+        session = self.get_session(token)
+
+        if self.request_token_response:
+            session.request_token_response = self.request_token_response
+        if self.access_token_response:
+            session.access_token_response = self.access_token_response
+
+        return session
 
 
 class OAuth2Service(Service):
@@ -435,6 +452,9 @@ def __init__(self,
         #: Object used to construct sessions with.
         self.session_obj = session_obj or OAuth2Session
 
+        #: Access token response.
+        self.access_token_response = None
+
         super(OAuth2Service, self).__init__(name,
                                             base_url,
                                             authorize_url)
@@ -493,7 +513,10 @@ def get_raw_access_token(self, method='POST', **kwargs):
                             'client_secret': self.client_secret})
 
         session = self.get_session()
-        return session.request(method, self.access_token_url, **kwargs)
+        self.access_token_response = session.request(method,
+                                                     self.access_token_url,
+                                                     **kwargs)
+        return self.access_token_response
 
     def get_access_token(self,
                          method='POST',
@@ -530,7 +553,12 @@ def get_auth_session(self, method='POST', **kwargs):
         :param \*\*kwargs: Optional arguments. Same as Requests.
         :type \*\*kwargs: dict
         '''
-        return self.get_session(self.get_access_token(method, **kwargs))
+        session = self.get_session(self.get_access_token(method, **kwargs))
+
+        if self.access_token_response:
+            session.access_token_response = self.access_token_response
+
+        return session
 
 
 class OflyService(Service):

tests/test_service_oauth1.py

@@ -308,6 +308,13 @@ def test_get_auth_session(self):
         s = self.service.get_auth_session('foo', 'bar')
         self.assertIsInstance(s, OAuth1Session)
 
+    def test_get_auth_session_with_request_token_response(self):
+        resp = 'oauth_token=foo&oauth_token_secret=bar'
+        self.response.content = resp
+        self.service.request_token_response = 'ok'
+        s = self.service.get_auth_session('foo', 'bar')
+        self.assertEqual(s.request_token_response, 'ok')
+
     def test_pickle_session(self):
         session = pickle.loads(pickle.dumps(self.session))
 

tests/test_service_oauth2.py

@@ -138,6 +138,12 @@ def test_get_auth_session(self):
         s = self.service.get_auth_session()
         self.assertIsInstance(s, OAuth2Session)
 
+    def test_get_auth_session_with_access_token_response(self):
+        self.response.content = \
+            'access_token=123&expires_in=3600&refresh_token=456'
+        s = self.service.get_auth_session()
+        self.assertIsNotNone(s.access_token_response)
+
     def test_pickle_session(self):
         session = pickle.loads(pickle.dumps(self.session))