Add the prestation total

YtoTech · Feb 16, 2021 · 2e9fa5c · 2e9fa5c
1 parent 16e21e5
commit 2e9fa5c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/tests/test_definitions_parsing.py b/tests/test_definitions_parsing.py
@@ -317,10 +317,12 @@ def test_parse_quote_with_quantities():
     assert len(quote["optional_prestations"]) == 0
     assert quote["prestations"][0]["title"] == "Création des configurations sur le CPQ"
     assert quote["prestations"][0]["price"] == 5000
+    assert quote["prestations"][0]["total"] == 5000
     assert quote["prestations"][0]["optional"] is False
     assert quote["prestations"][0]["quantity"] == 1
     assert quote["prestations"][1]["title"] == "Intégration de l'UI"
     assert quote["prestations"][1]["price"] == 10000
+    assert quote["prestations"][1]["total"] == 30000
     assert quote["prestations"][1]["optional"] is False
     assert quote["prestations"][1]["quantity"] == 3
 

diff --git a/tqwgp_parser/parser.hy b/tqwgp_parser/parser.hy
@@ -102,6 +102,7 @@
       ["title"]
       ["price" "quantity" "description" "batch" "optional"])
     {
+      "total" (compute-price [prestation] :count-optional True)
       "quantity" (get-default prestation "quantity" 1)
       "section" (get-default (if (none? section) {} section) "title" None)
       "batch" (parse-batch (get-default prestation "batch" (get-default (if (none? section) {} section) "batch" None)))