#29, subscribe 빌링키 생성 추가

Author: james-song

.gitignore

@@ -105,3 +105,8 @@ atlassian-ide-plugin.xml
 com_crashlytics_export_strings.xml
 crashlytics.properties
 crashlytics-build.properties
+
+.pytest_cache/
+.python-version
+venv/
+

iamport/client.py

@@ -95,6 +95,18 @@ def pay_again(self, **kwargs):
 
         return self._post(url, kwargs)
 
+    def customer_create(self, **kwargs):
+        customer_uid = kwargs.get('customer_uid')
+        for key in ['customer_uid', 'card_number', 'expiry', 'birth']:
+            if key not in kwargs:
+                raise KeyError('Essential parameter is missing!: %s' % key)
+        url = '{}subscribe/customers/{}'.format(self.imp_url, customer_uid)
+        return self._post(url, kwargs)
+
+    def customer_get(self, customer_uid):
+        url = '{}subscribe/customers/{}'.format(self.imp_url, customer_uid)
+        return self._get(url)
+
     def pay_foreign(self, **kwargs):
         url = '{}subscribe/payments/foreign'.format(self.imp_url)
         for key in ['merchant_uid', 'amount', 'card_number', 'expiry']:

tests/test_customer_create.py

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+import pytest
+
+
+def test_customer_create(iamport):
+    # Without 'card_number'
+    payload_notEnough = {
+        'customer_uid': 'customer_1234',
+        'expiry': '2019-03',
+        'birth': '500203',
+    }
+
+    with pytest.raises(KeyError) as e:
+        iamport.customer_create(**payload_notEnough)
+        assert "Essential parameter is missing!: card_number" in str(e)
+
+    payload_full = {
+        'customer_uid': 'customer_1234',
+        'expiry': '2019-03',
+        'birth': '500203',
+        'card_number': '4092-0230-1234-1234',
+    }
+
+    with pytest.raises(iamport.ResponseError) as e:
+        iamport.customer_create(**payload_full)
+        assert e.code == -1
+        assert u'카드정보 인증 및 빌키 발급에 실패하였습니다.' in e.message

tests/test_customer_get.py

@@ -0,0 +1,10 @@
+# -*- coding: utf-8 -*-
+import pytest
+
+
+def test_customer_get(iamport):
+    customer_uid = '000000'
+    with pytest.raises(iamport.ResponseError) as e:
+        iamport.customer_get(customer_uid)
+        assert u'요청하신 customer_uid(000000)로 등록된 정보를 찾을 수 없습니다.' == e.message
+