Source code for shuup.testing.factories.product_factory

import factory
import factory.fuzzy as fuzzy
from factory.django import DjangoModelFactory

from shuup.core.models import Product
from shuup.testing.text_data import random_title

from .shared import get_default_product_type, get_default_sales_unit
from .tax_factory import get_default_tax_class


class FuzzyName(fuzzy.FuzzyText):
    def fuzz(self):
        return random_title(prefix=self.prefix, suffix=self.suffix)




class ProductFactory(DjangoModelFactory):
    class Meta:
        model = Product

    type = factory.LazyAttribute(lambda obj: get_default_product_type())
    sku = fuzzy.FuzzyText(length=10)
    sales_unit = factory.LazyAttribute(lambda obj: get_default_sales_unit())
    tax_class = factory.LazyAttribute(lambda obj: get_default_tax_class())
    profit_center = fuzzy.FuzzyInteger(10000, 99999)
    cost_center = fuzzy.FuzzyInteger(10000, 99999)
    name = FuzzyName()

    @factory.post_generation
    def post(self, create, extracted, **kwargs):
        if create:
            # Create ShopProduct association with default shop
            from shuup.core.models import ShopProduct

            from .shared import get_default_shop

            shop = get_default_shop()
            shop_product, created = ShopProduct.objects.get_or_create(
                shop=shop,
                product=self,
                defaults={
                    "purchasable": True,
                    "visibility": 3,  # ALWAYS_VISIBLE
                    "default_price_value": 50.0,
                },
            )