From 9e5c73f205c71b913aed3e2e44196b0d3228b7b6 Mon Sep 17 00:00:00 2001
From: Diego Calvo <diegocc6@gmail.com>
Date: Tue, 16 Feb 2021 12:53:56 +0100
Subject: [PATCH] multiple category in ProductFactory

---
 products/factories.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/products/factories.py b/products/factories.py
index 3644cd1..c656f00 100644
--- a/products/factories.py
+++ b/products/factories.py
@@ -9,9 +9,18 @@ from companies.factories import CompanyFactory
 from products.models import Product
 
 
+FAKE_CATEGORIES = (
+    ('categoria-1'),
+    ('categoria-2'),
+    ('categoria-3'),
+    ('categoria-4'),
+    ('categoria-5'),
+    ('categoria-6'),
+)
 
 class ProductFactory(DjangoModelFactory):
 
+
     company = SubFactory(CompanyFactory)
     sku = FuzzyText(prefix='SKU_', length=10)
     name = FuzzyText(prefix='NAME_', length=10)
@@ -26,7 +35,7 @@ class ProductFactory(DjangoModelFactory):
     discount = FuzzyDecimal(low=0.00, high=100.00)
     stock = FuzzyInteger(low=0)
     tags = ['test']
-    category = 'top-category' # main tag category
+    category = FuzzyChoice(choices=FAKE_CATEGORIES) # main tag category
     attributes = ['programming/python', 'testing']
     identifiers = FuzzyText(prefix='IDENTIFIERS_', length=100)