Added apiclient tests with encryption, updated changelog version

Author: aseveryn-hw

CHANGELOG.rst

@@ -1,9 +1,13 @@
 Changelog
 =========
 
-1.2.1 (current)
+1.2.2 (current)
 ------------------
 
+1.2.1 (2019-01-17)
+------------------
+
+- FIX: Resolved issue with restricted "Accept" & "Content-Type" headers to support only "application/json" or "application/jose+json"
 
 1.2.0 (2018-12-18)
 ------------------

hyperwallet/tests/test_client.py

@@ -3,10 +3,12 @@
 import mock
 import json
 import unittest
+import os.path
 
 from hyperwallet.utils import ApiClient
 from hyperwallet.config import SERVER
 from hyperwallet.exceptions import HyperwalletAPIException
+from hyperwallet.utils.encryption import Encryption
 
 
 class ApiClientTest(unittest.TestCase):
@@ -19,6 +21,17 @@ def setUp(self):
             SERVER
         )
 
+        localDir = os.path.abspath(os.path.dirname(__file__))
+        clientPath = os.path.join(localDir, 'resources', 'private-jwkset1')
+        hyperwalletPath = os.path.join(localDir, 'resources', 'public-jwkset1')
+
+        self.clientWithEncryption = ApiClient(
+            'test-user',
+            'test-pass',
+            SERVER,
+            {'clientPrivateKeySetLocation': clientPath, 'hyperwalletKeySetLocation': hyperwalletPath}
+        )
+
     def test_failed_connection(self):
 
         with self.assertRaises(HyperwalletAPIException) as exc:
@@ -120,6 +133,125 @@ def test_receive_valid_json_response(self, session_mock):
             json.loads(encoded)
         )
 
+    @mock.patch('requests.Session.request')
+    def test_request_with_encryption_successful(self, session_mock):
+
+        data = {
+            'key': 'value'
+        }
+
+        localDir = os.path.abspath(os.path.dirname(__file__))
+        clientPath = os.path.join(localDir, 'resources', 'private-jwkset1')
+        hyperwalletPath = os.path.join(localDir, 'resources', 'public-jwkset1')
+        encryption = Encryption(clientPath, hyperwalletPath)
+        encryptedMessage = encryption.encrypt(json.dumps(data))
+
+        session_mock.return_value = mock.MagicMock(
+            status_code=200,
+            content=encryptedMessage,
+            headers={
+                "Content-Type": "application/jose+json"
+            }
+        )
+
+        encoded = json.dumps(data)
+        if hasattr(encoded, 'decode'):  # Python 2
+            encoded = encoded.decode('utf-8')
+
+        self.assertEqual(
+            self.clientWithEncryption._makeRequest(),
+            json.loads(encoded)
+        )
+
+    @mock.patch('requests.Session.request')
+    def test_request_with_encryption_when_content_type_contains_charset(self, session_mock):
+
+        data = {
+            'key': 'value'
+        }
+
+        localDir = os.path.abspath(os.path.dirname(__file__))
+        clientPath = os.path.join(localDir, 'resources', 'private-jwkset1')
+        hyperwalletPath = os.path.join(localDir, 'resources', 'public-jwkset1')
+        encryption = Encryption(clientPath, hyperwalletPath)
+        encryptedMessage = encryption.encrypt(json.dumps(data))
+
+        session_mock.return_value = mock.MagicMock(
+            status_code=200,
+            content=encryptedMessage,
+            headers={
+                "Content-Type": "application/jose+json;charset=utf-8"
+            }
+        )
+
+        encoded = json.dumps(data)
+        if hasattr(encoded, 'decode'):  # Python 2
+            encoded = encoded.decode('utf-8')
+
+        self.assertEqual(
+            self.clientWithEncryption._makeRequest(),
+            json.loads(encoded)
+        )
+
+    @mock.patch('requests.Session.request')
+    def test_request_with_encryption_when_content_type_contains_charset_ahead(self, session_mock):
+
+        data = {
+            'key': 'value'
+        }
+
+        localDir = os.path.abspath(os.path.dirname(__file__))
+        clientPath = os.path.join(localDir, 'resources', 'private-jwkset1')
+        hyperwalletPath = os.path.join(localDir, 'resources', 'public-jwkset1')
+        encryption = Encryption(clientPath, hyperwalletPath)
+        encryptedMessage = encryption.encrypt(json.dumps(data))
+
+        session_mock.return_value = mock.MagicMock(
+            status_code=200,
+            content=encryptedMessage,
+            headers={
+                "Content-Type": "charset=utf-8;application/jose+json"
+            }
+        )
+
+        encoded = json.dumps(data)
+        if hasattr(encoded, 'decode'):  # Python 2
+            encoded = encoded.decode('utf-8')
+
+        self.assertEqual(
+            self.clientWithEncryption._makeRequest(),
+            json.loads(encoded)
+        )
+
+    @mock.patch('requests.Session.request')
+    def test_request_with_encryption_when_response_content_type_is_not_valid(self, session_mock):
+
+        data = {
+            'key': 'value'
+        }
+
+        localDir = os.path.abspath(os.path.dirname(__file__))
+        clientPath = os.path.join(localDir, 'resources', 'private-jwkset1')
+        hyperwalletPath = os.path.join(localDir, 'resources', 'public-jwkset1')
+        encryption = Encryption(clientPath, hyperwalletPath)
+        encryptedMessage = encryption.encrypt(json.dumps(data))
+
+        session_mock.return_value = mock.MagicMock(
+            status_code=200,
+            content=encryptedMessage,
+            headers={
+                "Content-Type": "wrongContentType"
+            }
+        )
+
+        with self.assertRaises(HyperwalletAPIException) as exc:
+            self.clientWithEncryption._makeRequest()
+
+        self.assertEqual(
+            exc.exception.message,
+            'Invalid Content-Type specified in Response Header'
+        )
+
     @mock.patch('requests.Session.request')
     def test_receive_valid_json_response_when_content_type_contains_charset(self, session_mock):