Merge pull request #37 from hyperwallet/feature/HW-67903-V3-Python-SDK-for-User-Status-Transitions

added Create User Status Transition

Author: akalichety-hw

hyperwallet/__init__.py

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

hyperwallet/api.py

@@ -2791,4 +2791,146 @@ def createTransferSpendBackRefund(self,
             params
         )
 
-        return TransferRefunds(response)
+        return TransferRefunds(response)
+
+    '''
+
+        Create User Status Transition
+
+    '''
+
+    def createUserStatusTransition(self,
+                                   userToken=None,
+                                   data=None):
+        '''
+        Create User Status Transition.
+        :param userToken:
+            A token identifying the User Create UserToken. **REQUIRED**
+        :param data:
+            A dictionary containing User Status Create Data information. **REQUIRED**
+        :returns:
+            StatusTransition.
+        '''
+        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 StatusTransition(response)
+
+    def activateUser(self,
+                     userToken=None):
+
+        '''
+         Create User Activate Status Transition.
+         :param userToken:
+             A token identifying the User Create UserToken. **REQUIRED**
+         :returns:
+             StatusTransition.
+         '''
+        if not userToken:
+            raise HyperwalletException('userToken is required')
+
+        data = {
+            'transition': 'ACTIVATED'
+        }
+
+        return self.createUserStatusTransition(
+            userToken,
+            data
+        )
+
+    def deactivateUser(self,
+                       userToken=None):
+
+        '''
+         Create User DeActivate Status Transition.
+         :param userToken:
+             A token identifying the User Create UserToken. **REQUIRED**
+         :returns:
+             StatusTransition.
+         '''
+        if not userToken:
+            raise HyperwalletException('userToken is required')
+
+        data = {
+            'transition': 'DE_ACTIVATED'
+        }
+
+        return self.createUserStatusTransition(
+            userToken,
+            data
+        )
+
+    def preactivateUser(self,
+                        userToken=None):
+
+        '''
+         Create User Pre Activate Status Transition.
+         :param userToken:
+             A token identifying the User Create UserToken. **REQUIRED**
+         :returns:
+             StatusTransition.
+         '''
+        if not userToken:
+            raise HyperwalletException('userToken is required')
+        data = {
+            'transition': 'PRE-ACTIVATED'
+        }
+
+        return self.createUserStatusTransition(
+            userToken,
+            data
+        )
+
+    def freezeUser(self,
+                   userToken=None):
+
+        '''
+         Create User Freeze Status Transition.
+         :param userToken:
+             A token identifying the User Create UserToken. **REQUIRED**
+         :returns:
+             StatusTransition.
+         '''
+        if not userToken:
+            raise HyperwalletException('userToken is required')
+
+        data = {
+            'transition': 'FROZEN'
+        }
+
+        return self.createUserStatusTransition(
+            userToken,
+            data
+        )
+
+    def lockUser(self,
+                 userToken=None):
+
+        '''
+         Create User Lock Status Transition.
+         :param userToken:
+             A token identifying the User Create UserToken. **REQUIRED**
+         :returns:
+             StatusTransition.
+         '''
+        if not userToken:
+            raise HyperwalletException('userToken is required')
+
+        data = {
+            'transition': 'LOCKED'
+        }
+
+        return self.createUserStatusTransition(
+            userToken,
+            data
+        )

hyperwallet/tests/test_api.py

@@ -2267,5 +2267,82 @@ 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'))
+
+    def test_create_user_lockUser_status_transition_user_token(self):
+        with self.assertRaises(HyperwalletException) as exc:
+            self.api.lockUser()
+        self.assertEqual(exc.exception.message, 'userToken is required')
+
+    @mock.patch('hyperwallet.utils.ApiClient._makeRequest')
+    def test_create_user_lockUser_status_transition_success(self, mock_get):
+        mock_get.return_value = self.data
+        response = self.api.lockUser('token')
+        self.assertTrue(response.token, self.data.get('token'))
+
+    def test_create_user_freezeUser_status_transition_user_token(self):
+        with self.assertRaises(HyperwalletException) as exc:
+            self.api.freezeUser()
+        self.assertEqual(exc.exception.message, 'userToken is required')
+
+    @mock.patch('hyperwallet.utils.ApiClient._makeRequest')
+    def test_create_user_freezeUser_status_transition_success(self, mock_get):
+        mock_get.return_value = self.data
+        response = self.api.freezeUser('token')
+        self.assertTrue(response.token, self.data.get('token'))
+
+    def test_create_user_preactivate_status_transition_user_token(self):
+        with self.assertRaises(HyperwalletException) as exc:
+            self.api.preactivateUser()
+        self.assertEqual(exc.exception.message, 'userToken is required')
+
+    @mock.patch('hyperwallet.utils.ApiClient._makeRequest')
+    def test_create_user_preactivate_status_transition_success(self, mock_get):
+        mock_get.return_value = self.data
+        response = self.api.preactivateUser('token')
+        self.assertTrue(response.token, self.data.get('token'))
+
+    def test_create_user_deactivate_status_transition_user_token(self):
+        with self.assertRaises(HyperwalletException) as exc:
+            self.api.deactivateUser()
+        self.assertEqual(exc.exception.message, 'userToken is required')
+
+    @mock.patch('hyperwallet.utils.ApiClient._makeRequest')
+    def test_create_user_deactivate_status_transition_success(self, mock_get):
+        mock_get.return_value = self.data
+        response = self.api.deactivateUser('token')
+        self.assertTrue(response.token, self.data.get('token'))
+
+    def test_create_user_activate_status_transition_user_token(self):
+        with self.assertRaises(HyperwalletException) as exc:
+            self.api.activateUser()
+        self.assertEqual(exc.exception.message, 'userToken is required')
+
+    @mock.patch('hyperwallet.utils.ApiClient._makeRequest')
+    def test_create_user_activate_status_transition_success(self, mock_get):
+        mock_get.return_value = self.data
+        response = self.api.activateUser('token')
+        self.assertTrue(response.token, self.data.get('token'))
+
 if __name__ == '__main__':
     unittest.main()