from datetime import datetime, timedelta

from app.factories import Factory
from app.models.subscription import Subscription


class SubscriptionFactory(Factory):
    model = Subscription

    def definition(self) -> dict:
        return {
            "id": None,
            "tenant_id": None,
            "plan_id": None,
            "limits": None,
            "status": "TRIAL",
            "started_at": None,
            "ended_at": None,
        }