updated emoney tests

Co-authored-by: Iulian Masar <[email protected]>
Mangopay · Feb 25, 2025 · ac83332 · ac83332
1 parent c294acb
commit ac83332
Showing 1 changed file with 9 additions and 4 deletions.
diff --git 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)