oncall-engine/engine/apps/webhooks/tests/factories.py

20 lines
469 B
Python

import factory
import pytz
from apps.webhooks.models import Webhook, WebhookResponse
from common.utils import UniqueFaker
class CustomWebhookFactory(factory.DjangoModelFactory):
url = factory.Faker("url")
name = UniqueFaker("sentence", nb_words=3)
class Meta:
model = Webhook
class WebhookResponseFactory(factory.DjangoModelFactory):
timestamp = factory.Faker("date_time", tzinfo=pytz.UTC)
class Meta:
model = WebhookResponse