test_models.py

import random
import string
import pytest
from product.models import Category, Product, Retail

@pytest.fixture
def category(db) -> Category:
    return Category.objects.create(name="Books")

@pytest.fixture
def retailer_abc(db):
    return Retail.objects.create(name="ABC")

@pytest.fixture
def product_one(db, category, retailer_abc):
    sku = "".join(random.choices(string.ascii_uppercase + string.digits, k=6))
    product = Product.objects.create(
        sku=sku,
        name="Book 1",
        description="A book for educational purpose.",
        mrp="100.00",
        is_available=True,
        category=category,
    )
    product.retails.set([retailer_abc])
    return product

@pytest.fixture
def product_two(db, category, retailer):
    sku = "".join(random.choices(string.ascii_uppercase + string.digits, k=6))
    product = Product.objects.create(
        sku=sku,
        name="Book 2",
        description="A book with thriller story.",
        mrp="50.00",
        is_available=True,
        category=category,
    )
    product.retails.add([retailer])
    return product