Added test for download_pdf unauthorized condition

ej2 · ej2 · commit cb7c83ad75e1 · 2018-10-18T12:03:08.000-07:00
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -1,8 +1,10 @@
 Changelog
-========
+=========
 
-* 0.7.5 (May 11th, 2018)
+* 0.7.5 (October 18th, 2018)
     * Fixed bug with reporting authentication failure when attempting to download PDF (previously the error details were "lost")
+    * Added get_new_access_tokens to Oauth2SessionManager
+    *
 
 * 0.7.4 (March 26th, 2018)
     * Fixed bug in SendMixin send method.
diff --git a/tests/unit/test_client.py b/tests/unit/test_client.py
@@ -6,7 +6,7 @@
     from unittest.mock import patch
 
 from quickbooks.auth import Oauth1SessionManager
-from quickbooks.exceptions import QuickbooksException, SevereException
+from quickbooks.exceptions import QuickbooksException, SevereException, AuthorizationException
 from quickbooks import client
 from quickbooks.objects.salesreceipt import SalesReceipt
 
@@ -237,6 +237,17 @@ def test_download_nonexistent_pdf(self):
         receipt.Id = 666
         self.assertRaises(QuickbooksException, receipt.download_pdf)
 
+    @patch('quickbooks.client.QuickBooks.process_request')
+    def test_download_pdf_not_authorized(self, process_request):
+        qb_client = client.QuickBooks(sandbox=True)
+        qb_client.company_id = "1234"
+        receipt = SalesReceipt()
+        receipt.Id = 1
+
+        process_request.return_value = MockUnauthorizedResponse()
+
+        self.assertRaises(AuthorizationException, receipt.download_pdf, qb_client)
+
 
 class MockResponse(object):
     @property
@@ -258,6 +269,20 @@ def content(self):
         return ''
 
 
+class MockUnauthorizedResponse(object):
+    @property
+    def text(self):
+        return "UNAUTHORIZED"
+
+    @property
+    def status_code(self):
+        try:
+            import httplib  # python 2
+        except ImportError:
+            import http.client as httplib  # python 3
+        return httplib.UNAUTHORIZED
+
+
 class MockPdfResponse(object):
     @property
     def status_code(self):
