Merge pull request #21 from Leop0ld/master

비인증 결제예약/예약 취소 기능 및 테스트코드 추가

Author: perhapsspy

README.rst

@@ -150,6 +150,62 @@ Python 사용자를 위한 아임포트 REST API 연동 모듈입니다.
         pass
 
 
+정기 결제를 예약합니다.
+
+.. code-block:: python
+
+	# 테스트용 값
+	payload = {
+		'customer_uid': '{고객 아이디}',
+		'schedules': [
+            {
+                'merchant_uid': 'test_merchant_01',
+                # UNIX timestamp
+                'schedule_at': 1478150985,
+                'amount': 1004
+            },
+            {
+                'merhcant_uid': 'test_merchant_02',
+                # UNIX timestamp
+                'schedule_at': 1478150985,
+                'amount': 5000,
+                'name': '{주문명}',
+                'buyer_name': '{주문자명}',
+                'buyer_email': '{주문자 이메일}',
+                'buyer_tel': '{주문자 전화번호}',
+                'buyer_addr': '{주문자 주소}',
+                'buyer_postcode': '{주문자 우편번호}',
+            },
+		]
+	}
+	try:
+		reponse = iamport.pay_schedule(**payload)
+	except KeyError:
+        # 필수 값이 없을때 에러 처리
+        pass
+	except Iamport.ResponseError as e:
+		# 응답 에러 처리
+		pass
+
+정기 결제 예약을 취소합니다.
+
+.. code-block:: python
+
+	# 테스트용 값 (merchant_uid 가 누락되면 customer_uid 에 대한 결제예약정보 일괄취소)
+	payload = {
+		'customer_uid': '{고객 아이디}',
+		'merchant_uid': 'test_merchant_01',
+	}
+	try:
+		response = iamport.pay_unschedule(**payload)
+	except KeyError:
+        # 필수 값이 없을때 에러 처리
+        pass
+	except Iamport.ResponseError as e:
+		# 응답 에러 처리
+		pass
+
+
 결제 사전 검증
 ----------------
 

iamport/client.py

@@ -1,4 +1,6 @@
 # -*- coding: utf-8 -*-
+import json
+
 import requests
 
 __all__ = ['IAMPORT_API_URL', 'Iamport']
@@ -96,6 +98,27 @@ def pay_foreign(self, **kwargs):
 
         return self._post(url, kwargs)
 
+    def pay_schedule(self, **kwargs):
+        headers = self.get_headers()
+        headers['Content-Type'] = 'application/json'
+        url = '{}subscribe/payments/schedule'.format(self.imp_url)
+        if 'customer_uid' not in kwargs:
+            raise KeyError('customer_uid is required')
+        for key in ['merchant_uid', 'schedule_at', 'amount']:
+            for schedules in kwargs['schedules']:
+                if key not in schedules:
+                    raise KeyError('Essential parameter is missing!: %s' % key)
+
+        response = self.requests_session.post(url, headers=headers, data=json.dumps(kwargs))
+        return self.get_response(response)
+
+    def pay_unschedule(self, **kwargs):
+        url = '{}subscribe/payments/unschedule'.format(self.imp_url)
+        if 'customer_uid' not in kwargs:
+            raise KeyError('customer_uid is required')
+
+        return self._post(url, kwargs)
+
     def cancel_by_merchant_uid(self, merchant_uid, reason, **kwargs):
         payload = {'merchant_uid': merchant_uid, 'reason': reason}
         if kwargs:

tests/test_pay_again.py

@@ -4,7 +4,7 @@
 def test_pay_again(iamport):
     # Without 'customer_uid'
     payload_notEnough = {
-        'merchant_uid': '00000000',
+        'merchant_uid': '1234qwer',
         'amount': 5000,
     }
 
@@ -15,7 +15,7 @@ def test_pay_again(iamport):
 
     payload_full = {
         'customer_uid': '00000000',
-        'merchant_uid': '00000000',
+        'merchant_uid': '1234qwer',
         'amount': 5000,
     }
 

tests/test_pay_onetime.py

@@ -4,7 +4,7 @@
 def test_pay_onetime(iamport):
     # Without 'card_number'
     payload_notEnough = {
-        'merchant_uid': '00000000',
+        'merchant_uid': 'qwer1234',
         'amount': 5000,
         'expiry': '2019-03',
         'birth': '500203',
@@ -17,7 +17,7 @@ def test_pay_onetime(iamport):
         assert "Essential parameter is missing!: card_number" in str(e)
 
     payload_full = {
-        'merchant_uid': '00000000',
+        'merchant_uid': 'qwer1234',
         'amount': 5000,
         'card_number': '4092-0230-1234-1234',
         'expiry': '2019-03',

tests/test_pay_schedule.py

@@ -0,0 +1,73 @@
+# -*- coding: utf-8 -*-
+import time
+
+
+def test_pay_schedule(iamport):
+    schedule_at = int(time.time() + 1000)
+
+    payload_without_customer_uid = {
+        # without 'customer_uid'
+        'schedules': [
+            {
+                'merchant_uid': 'pay_schedule_%s' % str(time.time()),
+                'schedule_at': schedule_at,
+                'amount': 2001,
+                'name': '주문명1',
+                'buyer_name': '주문자명',
+                'buyer_email': '주문자 Email주소',
+                'buyer_tel': '주문자 전화번호',
+                'buyer_addr': '주문자 주소',
+                'buyer_postcode': '주문자 우편번호'
+            },
+        ],
+    }
+
+    try:
+        iamport.pay_schedule(**payload_without_customer_uid)
+    except KeyError as e:
+        assert 'customer_uid is required' in str(e)
+
+    payload_without_merchant_uid = {
+        'customer_uid': '00000000',
+        'schedules': [
+            {
+                # without 'merchant_uid'
+                'schedule_at': schedule_at,
+                'amount': 10000,
+                'name': '주문명2',
+                'buyer_name': '주문자명',
+                'buyer_email': '주문자 Email주소',
+                'buyer_tel': '주문자 전화번호',
+                'buyer_addr': '주문자 주소',
+                'buyer_postcode': '주문자 우편번호'
+            },
+        ],
+    }
+
+    try:
+        iamport.pay_schedule(**payload_without_merchant_uid)
+    except KeyError as e:
+        assert 'Essential parameter is missing!: merchant_uid' in str(e)
+
+    payload_full = {
+        'customer_uid': '00000000',
+        'schedules': [
+            {
+                'merchant_uid': 'pay_schedule_%s' % str(time.time()),
+                'schedule_at': schedule_at,
+                'amount': 5000,
+                'name': '주문명',
+                'buyer_name': '주문자명',
+                'buyer_email': '주문자 Email주소',
+                'buyer_tel': '주문자 전화번호',
+                'buyer_addr': '주문자 주소',
+                'buyer_postcode': '주문자 우편번호'
+            },
+        ],
+    }
+
+    try:
+        iamport.pay_schedule(**payload_full)
+    except iamport.ResponseError as e:
+        assert e.code == 1
+        assert u'등록되지 않은 구매자입니다.' in e.message

tests/test_pay_unschedule.py

@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+import time
+
+
+def test_pay_unschedule(iamport):
+    payload_without_customer_uid = {
+        # without 'customer_uid'
+        'merchant_uid': 'pay_unschedule_%s' % str(time.time()),
+    }
+
+    try:
+        iamport.pay_unschedule(**payload_without_customer_uid)
+    except KeyError as e:
+        assert 'customer_uid is required' in str(e)
+
+    payload_full = {
+        'customer_uid': '00000000',
+        'merchant_uid': 'pay_unschedule_%s' % str(time.time()),
+    }
+
+    try:
+        iamport.pay_unschedule(**payload_full)
+    except iamport.ResponseError as e:
+        assert e.code == 1
+        assert u'취소할 예약결제 기록이 존재하지 않습니다.' in e.message