From e3bb57396be5cb9765f660219831fd13f9d0fd54 Mon Sep 17 00:00:00 2001
From: Gabriel Augendre <gabriel@augendre.info>
Date: Mon, 27 Mar 2023 18:28:14 +0200
Subject: [PATCH] Generate dummy baskets with no fixed price items

---
 src/purchase/fixtures/products.json           | 11 ++++++++
 .../commands/generate_dummy_baskets.py        | 26 +++++++++++++------
 2 files changed, 29 insertions(+), 8 deletions(-)

diff --git a/src/purchase/fixtures/products.json b/src/purchase/fixtures/products.json
index d04f895..f766d2f 100644
--- a/src/purchase/fixtures/products.json
+++ b/src/purchase/fixtures/products.json
@@ -75,5 +75,16 @@
       "unit_price_cents": 290,
       "display_order": 7
     }
+  },
+  {
+    "model": "purchase.product",
+    "fields": {
+      "created_at": "2022-04-28T16:08:20.471Z",
+      "updated_at": "2022-04-28T16:09:34.003Z",
+      "name": "Tomme de ch\u00e8vre",
+      "image": "",
+      "unit_price_cents": 0,
+      "display_order": 8
+    }
   }
 ]
diff --git a/src/purchase/management/commands/generate_dummy_baskets.py b/src/purchase/management/commands/generate_dummy_baskets.py
index 0cb7d4a..7c9c75f 100644
--- a/src/purchase/management/commands/generate_dummy_baskets.py
+++ b/src/purchase/management/commands/generate_dummy_baskets.py
@@ -52,13 +52,23 @@ class Command(BaseCommand):
             )
             items = []
             for product in selected_products:
-                items.append(
-                    BasketItem(
-                        product=product,
-                        basket=basket,
-                        quantity=random.randint(1, 3),
-                        unit_price_cents=product.unit_price_cents,
-                    ),
-                )
+                if not product.has_fixed_price:
+                    items.append(
+                        BasketItem(
+                            product=product,
+                            basket=basket,
+                            quantity=1,
+                            unit_price_cents=random.randint(317, 514),
+                        ),
+                    )
+                else:
+                    items.append(
+                        BasketItem(
+                            product=product,
+                            basket=basket,
+                            quantity=random.randint(1, 3),
+                            unit_price_cents=product.unit_price_cents,
+                        ),
+                    )
             BasketItem.objects.bulk_create(items)
         return count