Merge pull request routablehq#34 from mbierma/current_user

ej2 · web-flow · commit 7c2f03642349 · 2016-07-21T09:21:27.000-04:00
Adds ability to query Current user and reconnect an account.
diff --git a/quickbooks/client.py b/quickbooks/client.py
@@ -41,7 +41,10 @@ class QuickBooks(object):
 
     authorize_url = "https://appcenter.intuit.com/Connect/Begin"
 
+    current_user_url = "https://appcenter.intuit.com/api/v1/user/current"
+
     disconnect_url = "https://appcenter.intuit.com/api/v1/connection/disconnect"
+    reconnect_url = "https://appcenter.intuit.com/api/v1/connection/reconnect"
 
     request_token = ''
     request_token_secret = ''
@@ -158,6 +161,12 @@ def get_authorize_url(self):
         self.request_token_secret = oauth_resp['oauth_token_secret']
         return self.qbService.get_authorize_url(self.request_token)
 
+    def get_current_user(self):
+        '''Get data from the current user endpoint'''
+        url = self.current_user_url
+        result = self.make_request("GET", url)
+        return result
+
     def get_report(self, report_type, qs=None):
         '''Get data from the report endpoint'''
         if qs == None:
@@ -203,6 +212,15 @@ def disconnect_account(self):
         result = self.make_request("GET", url)
         return result
 
+    def reconnect_account(self):
+        """
+        Reconnect current account by refreshing OAuth access tokens
+        :return:
+        """
+        url = self.reconnect_url
+        result = self.make_request("GET", url)
+        return result
+
     def make_request(self, request_type, url, request_body=None, content_type='application/json',
                      params=None, file_path=None):
 
diff --git a/tests/unit/test_client.py b/tests/unit/test_client.py
@@ -167,6 +167,15 @@ def test_get_authorize_url(self):
             self.assertEqual(qb_client.request_token, 'tokenvalue')
             self.assertTrue(qb_client.request_token_secret, 'secretvalue')
 
+    @patch('quickbooks.client.QuickBooks.make_request')
+    def test_get_current_user(self, make_req):
+        qb_client = client.QuickBooks()
+        qb_client.company_id = "1234"
+
+        result = qb_client.get_current_user()
+        url = "https://appcenter.intuit.com/api/v1/user/current"
+        make_req.assert_called_with("GET", url)
+
     @patch('quickbooks.client.QuickBooks.qbService')
     def test_get_access_tokens(self, qbService):
         qb_client = client.QuickBooks()
@@ -186,6 +195,15 @@ def test_disconnect_account(self, make_req):
         url = "https://appcenter.intuit.com/api/v1/connection/disconnect"
         make_req.assert_called_with("GET", url)
 
+    @patch('quickbooks.client.QuickBooks.make_request')
+    def test_reconnect_account(self, make_req):
+        qb_client = client.QuickBooks()
+        qb_client.company_id = "1234"
+
+        result = qb_client.reconnect_account()
+        url = "https://appcenter.intuit.com/api/v1/connection/reconnect"
+        make_req.assert_called_with("GET", url)
+
     @patch('quickbooks.client.QuickBooks.make_request')
     def test_get_report(self, make_req):
         qb_client = client.QuickBooks()
