andela-sjames /
paystack-python
| @@ 11-70 (lines=60) @@ | ||
| 8 | from paystackapi.plan import Plan |
|
| 9 | ||
| 10 | ||
| 11 | class TestPlan(unittest.TestCase): |
|
| 12 | """Class to test Plans.""" |
|
| 13 | ||
| 14 | @httpretty.activate |
|
| 15 | def test_create(self): |
|
| 16 | """Method defined to test plan creation.""" |
|
| 17 | httpretty.register_uri( |
|
| 18 | httpretty.POST, |
|
| 19 | "https://api.paystack.co/plan", |
|
| 20 | content_type='text/json', |
|
| 21 | body='{"status": true, "contributors": true}', |
|
| 22 | status=201, |
|
| 23 | ) |
|
| 24 | ||
| 25 | response = Plan.create( |
|
| 26 | name="John Doe", description="Payment plan for awesome people contributions", |
|
| 27 | amount=50000, interval="daily", send_invoices=True, ) |
|
| 28 | self.assertTrue(response['status']) |
|
| 29 | ||
| 30 | @httpretty.activate |
|
| 31 | def test_get(self): |
|
| 32 | """Function defined to test Plan get method.""" |
|
| 33 | httpretty.register_uri( |
|
| 34 | httpretty.GET, |
|
| 35 | "https://api.paystack.co/plan/78", |
|
| 36 | content_type='text/json', |
|
| 37 | body='{"status": true, "contributors": true}', |
|
| 38 | status=201, |
|
| 39 | ) |
|
| 40 | ||
| 41 | response = Plan.get(plan_id=78) |
|
| 42 | self.assertEqual(response['status'], True) |
|
| 43 | ||
| 44 | @httpretty.activate |
|
| 45 | def test_list(self): |
|
| 46 | """Function defined to test paystackapi plan list method.""" |
|
| 47 | httpretty.register_uri( |
|
| 48 | httpretty.GET, |
|
| 49 | "https://api.paystack.co/plan", |
|
| 50 | content_type='text/json', |
|
| 51 | body='{"status": true, "contributors": true}', |
|
| 52 | status=201, |
|
| 53 | ) |
|
| 54 | ||
| 55 | response = Plan.list() |
|
| 56 | self.assertEqual(response['status'], True) |
|
| 57 | ||
| 58 | @httpretty.activate |
|
| 59 | def test_update(self): |
|
| 60 | """Function defined to test paystackapi plan update.""" |
|
| 61 | httpretty.register_uri( |
|
| 62 | httpretty.PUT, |
|
| 63 | "https://api.paystack.co/plan/78", |
|
| 64 | content_type='text/json', |
|
| 65 | body='{"status": true, "contributors": true}', |
|
| 66 | status=201, |
|
| 67 | ) |
|
| 68 | ||
| 69 | response = Plan.update(plan_id=78, interval="monthly", amount=70000) |
|
| 70 | self.assertEqual(response['status'], True) |
|
| 71 | ||
| 72 | if __name__ == '__main__': |
|
| 73 | unittest.main(verbosity=2) |
|
| @@ 10-69 (lines=60) @@ | ||
| 7 | from paystackapi.customer import Customer |
|
| 8 | ||
| 9 | ||
| 10 | class TestCustomer(unittest.TestCase): |
|
| 11 | """Class to test customer actions.""" |
|
| 12 | ||
| 13 | @httpretty.activate |
|
| 14 | def test_create(self): |
|
| 15 | """Method defined to test customer creation.""" |
|
| 16 | httpretty.register_uri( |
|
| 17 | httpretty.POST, |
|
| 18 | "https://api.paystack.co/customer", |
|
| 19 | content_type='text/json', |
|
| 20 | body='{"status": true, "contributors": true}', |
|
| 21 | status=201, |
|
| 22 | ) |
|
| 23 | ||
| 24 | response = Customer.create( |
|
| 25 | first_name='samuel', last_name='james', |
|
| 26 | email='[email protected]', phone='00000000000') |
|
| 27 | self.assertTrue(response['status']) |
|
| 28 | ||
| 29 | @httpretty.activate |
|
| 30 | def test_get(self): |
|
| 31 | """Function defined to test Customer get method.""" |
|
| 32 | httpretty.register_uri( |
|
| 33 | httpretty.GET, |
|
| 34 | "https://api.paystack.co/customer/4013", |
|
| 35 | content_type='text/json', |
|
| 36 | body='{"status": true, "contributors": true}', |
|
| 37 | status=201, |
|
| 38 | ) |
|
| 39 | ||
| 40 | response = Customer.get(customer_id=4013) |
|
| 41 | self.assertEqual(response['status'], True) |
|
| 42 | ||
| 43 | @httpretty.activate |
|
| 44 | def test_list(self): |
|
| 45 | """Function defined to test paystackapi customer list method.""" |
|
| 46 | httpretty.register_uri( |
|
| 47 | httpretty.GET, |
|
| 48 | "https://api.paystack.co/customer", |
|
| 49 | content_type='text/json', |
|
| 50 | body='{"status": true, "contributors": true}', |
|
| 51 | status=201, |
|
| 52 | ) |
|
| 53 | ||
| 54 | response = Customer.list() |
|
| 55 | self.assertEqual(response['status'], True) |
|
| 56 | ||
| 57 | @httpretty.activate |
|
| 58 | def test_update(self): |
|
| 59 | """Function defined to test paystackapi customer update.""" |
|
| 60 | httpretty.register_uri( |
|
| 61 | httpretty.PUT, |
|
| 62 | "https://api.paystack.co/customer/4013", |
|
| 63 | content_type='text/json', |
|
| 64 | body='{"status": true, "contributors": true}', |
|
| 65 | status=201, |
|
| 66 | ) |
|
| 67 | ||
| 68 | response = Customer.update(customer_id=4013, first_name='andela') |
|
| 69 | self.assertEqual(response['status'], True) |
|
| 70 | ||
| 71 | if __name__ == '__main__': |
|
| 72 | unittest.main(verbosity=2) |
|
