36 lines
1.2 KiB
Python
36 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
|