dcae98b02a
Related to https://github.com/grafana/oncall/issues/96 --------- Co-authored-by: Ravishankar <ravishankar.gnanaprakasam@gmail.com>
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
import factory
|
|
|
|
from apps.mattermost.models import MattermostChannel, MattermostMessage, MattermostUser
|
|
from common.utils import UniqueFaker
|
|
|
|
|
|
class MattermostChannelFactory(factory.DjangoModelFactory):
|
|
mattermost_team_id = factory.LazyAttribute(
|
|
lambda v: str(UniqueFaker("pystr", min_chars=5, max_chars=26).generate())
|
|
)
|
|
channel_id = factory.LazyAttribute(lambda v: str(UniqueFaker("pystr", min_chars=5, max_chars=26).generate()))
|
|
channel_name = factory.Faker("word")
|
|
display_name = factory.Faker("word")
|
|
|
|
class Meta:
|
|
model = MattermostChannel
|
|
|
|
|
|
class MattermostMessageFactory(factory.DjangoModelFactory):
|
|
post_id = factory.LazyAttribute(lambda v: str(UniqueFaker("pystr", min_chars=5, max_chars=26).generate()))
|
|
channel_id = factory.LazyAttribute(lambda v: str(UniqueFaker("pystr", min_chars=5, max_chars=26).generate()))
|
|
|
|
class Meta:
|
|
model = MattermostMessage
|
|
|
|
|
|
class MattermostUserFactory(factory.DjangoModelFactory):
|
|
mattermost_user_id = factory.LazyAttribute(
|
|
lambda v: str(UniqueFaker("pystr", min_chars=5, max_chars=26).generate())
|
|
)
|
|
username = factory.Faker("word")
|
|
nickname = factory.Faker("word")
|
|
|
|
class Meta:
|
|
model = MattermostUser
|