from app.factories import Factory, random_string
from app.models.plan import Plan


class PlanFactory(Factory):
    model = Plan

    def definition(self) -> dict:
        return {
            "id": None,
            "name": f"Plan {random_string(6).title()}",
            "description": None,
            "price": 0,
            "limits": None,
            "is_active": True,
        }