api/app/factories/subscription_factory.py

20 lines
455 B
Python

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,
}