fix test for setup_payout function

Author: corentinalcoy

kkiapay/core.py

@@ -57,7 +57,8 @@ def refund_transaction(self, transaction_id):
         )
 
     def setup_payout(self, algorithm: str, destination: str, destination_type: str, roof_amount: str = None,
-                     send_notification: bool = True, rate_frequency: str = None, country_code: str = "229"):
+                     send_notification: bool = True, rate_frequency: str = DESTINATION_TYPES["1"],
+                     country_code: str = "229"):
         options = {
             "destination": country_code + destination if destination_type == "1" else destination,
             "send_notification": 1 if send_notification else 0,

tests/test_kkiapay.py

@@ -14,12 +14,55 @@ def teardown_module(module):
 
 
 class TestKkiapay(object):
-    def test_urls(self):
-        live = Kkiapay("public_key", "private_key", "secret")
-        sandbox = Kkiapay("public_key", "private_key", "secret", sandbox=True)
+    live = Kkiapay("public_key", "private_key", "secret")
+    sandbox = Kkiapay("public_key", "private_key", "secret", sandbox=True)
 
-        assert live.url == Kkiapay.BASE_URL
-        assert sandbox.url == Kkiapay.SANDBOX_URL
+    def test_urls(self):
+        assert self.live.url == Kkiapay.BASE_URL
+        assert self.sandbox.url == Kkiapay.SANDBOX_URL
 
     def test_requests(self, requests_mock):
         pass
+
+    def test_setup_payout(self, monkeypatch):
+        results = {
+            'response': {
+                'merchant': '5d34445784deb700073a0281',
+                'meta_data': '',
+                'algorithm': 'roof',
+                'rate_frequency': '',
+                'roof_amount': '50000',
+                'active': True,
+                'send_notification': True,
+                'destination_type': 'MOBILE_MONEY',
+                'destination': '22961000000',
+                'job_name': '',
+                'account': '5d34445784deb700073a0282'
+            },
+            'status_code': 200}
+
+        class MockJsonResponse:
+            status_code = 200
+
+            @staticmethod
+            def json():
+                return {
+                    'merchant': '5d34445784deb700073a0281',
+                    'meta_data': '',
+                    'algorithm': 'roof',
+                    'rate_frequency': '',
+                    'roof_amount': '50000',
+                    'active': True,
+                    'send_notification': True,
+                    'destination_type': 'MOBILE_MONEY',
+                    'destination': '22961000000',
+                    'job_name': '',
+                    'account': '5d34445784deb700073a0282'
+                }
+
+        def kkiapay_api_mock_return(*args, **kwargs):
+            return MockJsonResponse()
+
+        monkeypatch.setattr(requests, 'post', kkiapay_api_mock_return)
+        assert self.sandbox.setup_payout(algorithm="2", destination="61000000", destination_type="1") == results
+        assert self.live.setup_payout(algorithm="1", destination="61000000", destination_type="1") == results