diff --git a/tests/test_emoney.py b/tests/test_emoney.py index a15185d..f1f66e8 100644 --- a/tests/test_emoney.py +++ b/tests/test_emoney.py @@ -1,5 +1,4 @@ # -*- coding: utf-8 -*- -from mangopay.resources import EMoney from tests.test_base import BaseTestLive @@ -8,14 +7,20 @@ class EMoneyTest(BaseTestLive): def test_retrieve_emoney(self): john = self.get_john() emoney = john.get_emoney() - self.assertIsNotNone(emoney) + self.assertResponse(emoney) def test_retrieve_emoney_for_year(self): john = self.get_john() emoney = john.get_emoney(**{'year': 2019}) - self.assertIsNotNone(emoney) + self.assertResponse(emoney) def test_retrieve_emoney_for_month(self): john = self.get_john() emoney = john.get_emoney(**{'year': 2019, 'month': 4}) - self.assertIsNotNone(emoney) + self.assertResponse(emoney) + + def assertResponse(self, emoney_response): + self.assertIsNotNone(emoney_response) + self.assertTrue(len(emoney_response.data) > 0) + self.assertIsNotNone(emoney_response.data[0].credited_emoney) + self.assertIsNotNone(emoney_response.data[0].debited_emoney)