added Create User Status Transition

Author: mgovindhasamy

hyperwallet/__init__.py

@@ -32,7 +32,8 @@
     StatusTransition,                                                    # noqa
     TransferMethodConfiguration,                                         # noqa
     Webhook,                                                             # noqa
-    TransferRefunds                                                      # naqa
+    TransferRefunds,                                                     # naqa
+    CreateUserStatusTransition                                           # naqa
 )
 
 from .api import Api                                                     # noqa

hyperwallet/api.py

@@ -23,7 +23,8 @@
     StatusTransition,
     TransferMethodConfiguration,
     Webhook,
-    TransferRefunds
+    TransferRefunds,
+    CreateUserStatusTransition
 )
 
 
@@ -2791,4 +2792,41 @@ def createTransferSpendBackRefund(self,
             params
         )
 
-        return TransferRefunds(response)
+        return TransferRefunds(response)
+
+    '''
+
+    Create User Transition
+
+    '''
+
+    def createUserStatusTransition(self,
+                                   userToken=None,
+                             data=None):
+        '''
+        Create User Transition.
+
+        :param data:
+            A dictionary containing User Create UserToken information. **REQUIRED**
+        :param data:
+            A dictionary containing User Create Data information. **REQUIRED**
+        :returns:
+            CreateUserStatusTransition.
+        '''
+
+        if not userToken:
+            raise HyperwalletException('userToken is required')
+
+        if not data:
+            raise HyperwalletException('data is required')
+
+        response = self.apiClient.doPost(
+            self.__buildUrl(
+                'users',
+                userToken,
+                'status-transitions'
+            ),
+            data
+        )
+
+        return CreateUserStatusTransition(response)

hyperwallet/models.py

@@ -845,4 +845,43 @@ def __repr__(self):
         return "TransferRefunds({date}, {token})".format(
             date=self.createdOn,
             token=self.token
+        )
+'''
+
+   Create User Transition
+
+'''
+
+class CreateUserStatusTransition(HyperwalletModel):
+    '''
+    The StatusTransition Model.
+
+    :param data:
+        A dictionary containing the attributes for the Status Transition.
+    '''
+
+    def __init__(self, data):
+        '''
+        Create a new Status Transition with the provided attributes.
+        '''
+
+        super(CreateUserStatusTransition, self).__init__(data)
+
+        self.defaults = {
+            'createdOn': None,
+            'fromStatus': None,
+            'notes': None,
+            'statusCode': None,
+            'token': None,
+            'toStatus': None,
+            'transition': None
+        }
+
+        for (param, default) in self.defaults.items():
+            setattr(self, param, data.get(param, default))
+
+    def __repr__(self):
+        return "CreateUserStatusTransition({date}, {token})".format(
+            date=self.createdOn,
+            token=self.token
         )

hyperwallet/tests/test_api.py

@@ -2267,5 +2267,34 @@ def test_create_transfer_spend_back_refunds_success(self, mock_get):
 
         self.assertTrue(response.token, self.data.get('token'))
 
+    '''
+
+    Create User Status Transition
+
+    '''
+
+    def test_create_user_status_transition_user_token(self):
+
+        with self.assertRaises(HyperwalletException) as exc:
+            self.api.createUserStatusTransition()
+
+        self.assertEqual(exc.exception.message, 'userToken is required')
+
+    def test_create_user_status_transition_data(self):
+
+        with self.assertRaises(HyperwalletException) as exc:
+            self.api.createUserStatusTransition('token')
+
+        self.assertEqual(exc.exception.message, 'data is required')
+
+    @mock.patch('hyperwallet.utils.ApiClient._makeRequest')
+    def test_create_user_status_transition_success(self, mock_get):
+
+        mock_get.return_value = self.data
+        response = self.api.createUserStatusTransition('token', 'token')
+
+        self.assertTrue(response.token, self.data.get('token'))
+
+
 if __name__ == '__main__':
     unittest.main()

hyperwallet/tests/test_models.py

@@ -23,7 +23,8 @@
     StatusTransition,
     TransferMethodConfiguration,
     Webhook,
-    TransferRefunds
+    TransferRefunds,
+    CreateUserStatusTransition
 )
 
 
@@ -554,5 +555,28 @@ def test_transfer_refunds(self):
             )
         )
 
+
+    '''
+    
+    Create User Status Transition
+
+    '''
+
+    def test_create_user_status_transition(self):
+
+        status_transfer_refunds_data = {
+            "transition" : "LOCKED"
+        }
+
+        status_create_user_status_transition = CreateUserStatusTransition(status_transfer_refunds_data)
+
+        self.assertEqual(
+            status_create_user_status_transition.__repr__(),
+            'CreateUserStatusTransition({date}, {token})'.format(
+                date=status_transfer_refunds_data.get('createdOn'),
+                token=status_transfer_refunds_data.get('token')
+            )
+        )
+
 if __name__ == '__main__':
     unittest.main()