Revert get and retrive User Status Transition

Author: akreisman-epam

hyperwallet/api.py

@@ -155,6 +155,65 @@ def listUsers(self,
 
         return [User(x) for x in response.get('data', [])]
 
+    def getUserStatusTransition(self,
+                                userToken=None,
+                                statusTransitionToken=None):
+        '''
+        Retrieve a User Status Transition.
+
+        :param userToken:
+            A token identifying the User. **REQUIRED**
+        :param statusTransitionToken:
+            A token identifying the User Status Transition. **REQUIRED**
+        :returns:
+            A User Status Transition.
+        '''
+
+        if not userToken:
+            raise HyperwalletException('userToken is required')
+
+        if not statusTransitionToken:
+            raise HyperwalletException('statusTransitionToken is required')
+
+        response = self.apiClient.doGet(
+            os.path.join(
+                'users',
+                userToken,
+                'status-transitions',
+                statusTransitionToken
+            )
+        )
+
+        return StatusTransition(response)
+
+    def listUserStatusTransitions(self,
+                                  userToken=None,
+                                  params=None):
+        '''
+        List User Status Transitions.
+
+        :param userToken:
+            A token identifying the User. **REQUIRED**
+        :param params:
+            A dictionary containing query parameters.
+        :returns:
+            An array of User Status Transitions.
+        '''
+
+        if not userToken:
+            raise HyperwalletException('userToken is required')
+
+        response = self.apiClient.doGet(
+            os.path.join(
+                'users',
+                userToken,
+                'status-transitions'
+            ),
+            params
+        )
+
+        return [StatusTransition(x) for x in response.get('data', [])]
+
     '''
 
     Bank Accounts

hyperwallet/tests/test_api.py

@@ -131,6 +131,43 @@ def test_list_users_success(self, mock_get):
 
         self.assertEqual(response[0].token, self.data.get('token'))
 
+    def test_get_user_status_transition_fail_need_user_token(self):
+
+        with self.assertRaises(HyperwalletException) as exc:
+            self.api.getUserStatusTransition()
+
+        self.assertEqual(exc.exception.message, 'userToken is required')
+
+    def test_get_user_status_transition_fail_need_transition_token(self):
+
+        with self.assertRaises(HyperwalletException) as exc:
+            self.api.getUserStatusTransition('token')
+
+        self.assertEqual(exc.exception.message, 'statusTransitionToken is required')
+
+    @mock.patch('hyperwallet.utils.ApiClient._makeRequest')
+    def test_get_user_status_transition_success(self, mock_get):
+
+        mock_get.return_value = self.data
+        response = self.api.getUserStatusTransition('token', 'token')
+
+        self.assertTrue(response.token, self.data.get('token'))
+
+    def test_list_user_status_transitions_fail_need_user_token(self):
+
+        with self.assertRaises(HyperwalletException) as exc:
+            self.api.listUserStatusTransitions()
+
+        self.assertEqual(exc.exception.message, 'userToken is required')
+
+    @mock.patch('hyperwallet.utils.ApiClient._makeRequest')
+    def test_list_user_status_transitions_success(self, mock_get):
+
+        mock_get.return_value = {'data': [self.data]}
+        response = self.api.listUserStatusTransitions('token')
+
+        self.assertTrue(response[0].token, self.data.get('token'))
+
     '''
 
     Bank Accounts