get and list transfer status transition added

hyperwallet/api.py

@@ -2933,3 +2933,75 @@ def lockUser(self,
             userToken,
             data
         )
+
+    '''
+
+        Get Transfer Status Transition
+
+    '''
+
+    def getTransferStatusTransition(self,
+                                    transferToken=None,
+                                    statusTransitionToken=None):
+        '''
+        Retrieve a Transfer Status Transition.
+
+        :param transferToken:
+            A token identifying the Transfer. **REQUIRED**
+        :param statusTransitionToken:
+            A token identifying the Transfer Status Transition. **REQUIRED**
+        :returns:
+            A Transfer Status Transition.
+        '''
+
+        if not transferToken:
+            raise HyperwalletException('transferToken is required')
+
+        if not statusTransitionToken:
+            raise HyperwalletException('statusTransitionToken is required')
+
+        response = self.apiClient.doGet(
+            self.__buildUrl(
+                'transfers',
+                transferToken,
+                'status-transitions',
+                statusTransitionToken
+            )
+        )
+
+        return StatusTransition(response)
+
+    '''
+
+        List Transfer Status Transition
+
+    '''
+
+    def listTransferStatusTransitions(self,
+                                      transferToken=None,
+                                      params=None):
+        '''
+        Retrieve a Transfer Status Transition.
+
+        :param transferToken:
+            A token identifying the Transfer. **REQUIRED**
+        :returns:
+            A Transfer Status Transition.
+        '''
+
+        if not transferToken:
+            raise HyperwalletException('transferToken is required')
+
+        if params and not StatusTransition.filter_array >= params.keys():
+            raise HyperwalletException('Invalid filter')
+
+        response = self.apiClient.doGet(
+            self.__buildUrl(
+                'transfers',
+                transferToken,
+                'status-transitions'
+            ),
+            params
+        )
+
+        return StatusTransition(response)

hyperwallet/models.py

@@ -682,6 +682,8 @@ class StatusTransition(HyperwalletModel):
         A dictionary containing the attributes for the Status Transition.
     '''
 
+    filter_array = {'transition'}
+
     def __init__(self, data):
         '''
         Create a new Status Transition with the provided attributes.

hyperwallet/tests/test_api.py

@@ -2344,6 +2344,56 @@ def test_create_user_activate_status_transition_success(self, mock_get):
         response = self.api.activateUser('token')
         self.assertTrue(response.token, self.data.get('token'))
 
+    '''
+
+        Get Transfer Status Transition
+
+    '''
+
+    def test_get_transfer_status_transition_fail_need_transfer_token(self):
+
+        with self.assertRaises(HyperwalletException) as exc:
+            self.api.getTransferStatusTransition()
+
+        self.assertEqual(exc.exception.message, 'transferToken is required')
+
+    def test_get_transfer_status_transition_fail_need_transition_token(self):
+
+        with self.assertRaises(HyperwalletException) as exc:
+            self.api.getTransferStatusTransition('token')
+
+        self.assertEqual(exc.exception.message, 'statusTransitionToken is required')
+
+    @mock.patch('hyperwallet.utils.ApiClient._makeRequest')
+    def test_get_transfer_status_transition_success(self, mock_get):
+
+        mock_get.return_value = self.data
+        response = self.api.getTransferStatusTransition('token', 'token')
+
+        self.assertTrue(response.token, self.data.get('token'))
+
+    '''
+
+        List Transfer Status Transition
+
+    '''
+
+    def test_list_transfer_status_transitions_fail_need_transfer_token(self):
+
+        with self.assertRaises(HyperwalletException) as exc:
+            self.api.listTransferStatusTransitions()
+
+        self.assertEqual(exc.exception.message, 'transferToken is required')
+
+    @mock.patch('hyperwallet.utils.ApiClient._makeRequest')
+    def test_list_transfer_status_transitions_success(self, mock_get):
+
+        options = {'transition': 'SCHEDULED'}
+        mock_get.return_value = self.data
+        response = self.api.listTransferStatusTransitions('token', options)
+
+        self.assertTrue(response.token, self.data.get('token'))
+
 
 if __name__ == '__main__':
     unittest.main()

hyperwallet/tests/test_encryption.py

@@ -41,7 +41,7 @@ def test_should_fail_decryption_when_wrong_private_key_is_used(self):
         with self.assertRaises(HyperwalletException) as exc:
             encryption2.decrypt(encryptedMessage)
 
-        self.assertTrue(str(exc.exception).startswith('No recipient matched the provided key["Failed: [ValueError(\'Decryption failed.\''))
+        self.assertTrue(str(exc.exception).startswith('No recipient matched the provided key'))
 
     def test_should_fail_signature_verification_when_wrong_public_key_is_used(self):