Corrige serialização de Configurations (#11)

* corrige serialização do configurations * adiciona teste * fmt
imobanco · Sep 1, 2020 · f85338d · f85338d
1 parent 4f2e5e1
commit f85338d
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 5 deletions.
diff --git a/imopay_wrapper/models/transaction.py b/imopay_wrapper/models/transaction.py
@@ -44,11 +44,15 @@ def to_dict(self):
         Ao invés de solucionar isso, mais fácil sobreescrever o método
         no momento.
         """
-        data = {
-            "fine": self.fine.to_dict(),
-            "interest": self.interest.to_dict(),
-            "discounts": [discount.to_dict() for discount in self.discounts],
-        }
+        data = {}
+        if self.fine:
+            data["fine"] = self.fine.to_dict()
+
+        if self.interest:
+            data["interest"] = self.interest.to_dict()
+
+        if self.discounts:
+            data["discounts"] = [discount.to_dict() for discount in self.discounts]
         return data
 
 

diff --git a/tests/models/transaction/test_invoice_configurations.py b/tests/models/transaction/test_invoice_configurations.py
@@ -19,3 +19,13 @@ def test_2(self):
             {"fine": {"value": 1, "type": "foo", "charge_type": "foo", "days": 0}}
         )
         self.assertEqual(t.fine.value, 1)
+
+    def test_3(self):
+        t = InvoiceConfigurations.from_dict(
+            {
+                "fine": Configuration.from_dict(
+                    {"value": 1, "type": "foo", "charge_type": "foo", "days": 0}
+                )
+            }
+        )
+        t.to_dict()