From bf9601377d3578c33bfbbbcdc1008572af324989 Mon Sep 17 00:00:00 2001
From: Tom Roussel <21120212+TomRoussel@users.noreply.github.com>
Date: Sat, 2 Mar 2024 12:52:55 +0100
Subject: [PATCH] Added test checking correct bill_type validation

---
 ihatemoney/tests/api_test.py | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/ihatemoney/tests/api_test.py b/ihatemoney/tests/api_test.py
index d47b1650..f3f1d61a 100644
--- a/ihatemoney/tests/api_test.py
+++ b/ihatemoney/tests/api_test.py
@@ -1013,3 +1013,38 @@ class TestAPI(IhatemoneyTestCase):
             headers=self.get_auth("raclette"),
         )
         self.assertStatus(400, req)
+
+    def test_validate_bill_type(self):
+        self.api_create("raclette")
+        self.api_add_member("raclette", "zorglub")
+        
+
+        req = self.client.post(
+            "/api/projects/raclette/bills",
+            data={
+                "date": "2011-08-10",
+                "what": "fromage",
+                "payer": "1",
+                "payed_for": ["1"],
+                "bill_type": "wrong_bill_type",
+                "amount": "50",
+            },
+            headers=self.get_auth("raclette")
+        )
+
+        self.assertStatus(400, req)
+
+        req = self.client.post(
+            "/api/projects/raclette/bills",
+            data={
+                "date": "2011-08-10",
+                "what": "fromage",
+                "payer": "1",
+                "payed_for": ["1"],
+                "bill_type": "Expense",
+                "amount": "50",
+            },
+            headers=self.get_auth("raclette")
+        )
+
+        self.assertStatus(201, req)
\ No newline at end of file