Fix tests

InsaLan · Feb 18, 2024 · 81ff7b3 · 81ff7b3
1 parent f3517cb
commit 81ff7b3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/insalan/pizza/serializers.py b/insalan/pizza/serializers.py
@@ -43,8 +43,12 @@ def create(self, validated_data):
         """Create an order"""
         price_type = validated_data.pop('type')
         pizza = validated_data.pop("pizza")
+        if "payment_method" not in validated_data:
+            payment_method = PaymentMethod.CB
+        else:
+            payment_method = validated_data["payment_method"]
 
-        if validated_data["payment_method"] == PaymentMethod.FR:
+        if payment_method == PaymentMethod.FR:
             price = 0
         elif price_type == "staff":
             price = TimeSlot.objects.get(id=validated_data["time_slot"].id).staff_price * len(pizza)