Merge pull request ej2#84 from nbhambhani/user_agent

ej2 · web-flow · commit 02d17ac66639 · 2017-08-17T09:26:31.000-07:00
Update user-agent
diff --git a/quickbooks/client.py b/quickbooks/client.py
@@ -247,7 +247,8 @@ def make_request(self, request_type, url, request_body=None, content_type='appli
             self.create_session()
         headers = {
             'Content-Type': content_type,
-            'Accept': 'application/json'
+            'Accept': 'application/json',
+            'User-Agent': 'python-quickbooks V3 library'
         }
 
         if file_path:
@@ -257,7 +258,7 @@ def make_request(self, request_type, url, request_body=None, content_type='appli
             headers.update({
                 'Content-Type': 'multipart/form-data; boundary=%s' % boundary,
                 'Accept-Encoding': 'gzip;q=1.0,deflate;q=0.6,identity;q=0.3',
-                'User-Agent': 'OAuth gem v0.4.7',
+                'User-Agent': 'python-quickbooks V3 library',
                 'Accept': 'application/json',
                 'Connection': 'close'
             })
@@ -378,6 +379,7 @@ def download_pdf(self, qbbo, item_id):
         headers = {
             'Content-Type': 'application/pdf',
             'Accept': 'application/pdf, application/json',
+            'User-Agent': 'python-quickbooks V3 library'
         }
 
         response = self.session.request("GET", url, True, self.company_id, headers=headers)
diff --git a/tests/unit/test_client.py b/tests/unit/test_client.py
@@ -253,7 +253,7 @@ def test_make_request(self, qb_session):
 
         qb_session.request.assert_called_with(
                 "GET", url, True, "1234", data={},
-                headers={'Content-Type': 'application/json', 'Accept': 'application/json'}, params={})
+                headers={'Content-Type': 'application/json', 'Accept': 'application/json', 'User-Agent': 'python-quickbooks V3 library'}, params={})
 
     def test_make_request_create_session(self):
         receipt = SalesReceipt()
@@ -298,7 +298,7 @@ def test_download_pdf(self, qb_session):
         url = "https://sandbox-quickbooks.api.intuit.com/v3/company/1234/salesreceipt/1/pdf"
         qb_session.request.assert_called_with(
             "GET", url, True, "1234",
-            headers={'Content-Type': 'application/pdf', 'Accept': 'application/pdf, application/json'})
+            headers={'Content-Type': 'application/pdf', 'Accept': 'application/pdf, application/json', 'User-Agent': 'python-quickbooks V3 library'})
 
         qb_session.request.return_value = MockPdfResponse()
         response = receipt.download_pdf(qb_client)
