mirror of
https://github.com/Balshgit/gpt_chat_bot.git
synced 2025-09-11 22:30:41 +03:00
97 lines
3.3 KiB
Python
97 lines
3.3 KiB
Python
import string
|
|
import time
|
|
from typing import Any
|
|
|
|
import factory
|
|
import factory.fuzzy
|
|
from faker import Faker
|
|
|
|
from constants import BotStagesEnum
|
|
from tests.integration.factories.models import Chat, User
|
|
|
|
faker = Faker("ru_RU")
|
|
|
|
|
|
class BotUserFactory(factory.Factory):
|
|
id = factory.Sequence(lambda n: 1000 + n)
|
|
is_bot = False
|
|
first_name = factory.Faker("first_name")
|
|
last_name = factory.Faker("last_name")
|
|
username = faker.profile(fields=["username"])["username"]
|
|
language_code = "ru"
|
|
|
|
class Meta:
|
|
model = User
|
|
|
|
|
|
class BotChatFactory(factory.Factory):
|
|
id = factory.Sequence(lambda n: 1 + n)
|
|
first_name = factory.Faker("first_name")
|
|
last_name = factory.Faker("last_name")
|
|
username = faker.profile(fields=["username"])["username"]
|
|
type = "private"
|
|
|
|
class Meta:
|
|
model = Chat
|
|
|
|
|
|
class BotInfoFactory(factory.DictFactory):
|
|
token = factory.Faker(
|
|
"bothify", text="#########:??????????????????????????-#????????#?", letters=string.ascii_letters
|
|
) # example: 579694714:AAFpK8w6zkkUrD4xSeYwF3MO8e-4Grmcy7c
|
|
payment_provider_token = factory.Faker(
|
|
"bothify", text="#########:TEST:????????????????", letters=string.ascii_letters
|
|
) # example: 579694714:TEST:K8w6zkkUrD4xSeYw
|
|
chat_id = factory.Faker("random_int", min=10**8, max=10**9 - 1)
|
|
super_group_id = factory.Faker("random_int", min=-(10**12) - 10**9, max=-(10**12)) # -1001838004577
|
|
forum_group_id = factory.Faker("random_int", min=-(10**12) - 10**9, max=-(10**12))
|
|
channel_name = factory.Faker("name")
|
|
channel_id = factory.LazyAttribute(lambda obj: f"@{obj.channel_name}")
|
|
name = factory.Faker("name")
|
|
fake_username = factory.Faker("name")
|
|
username = factory.LazyAttribute(lambda obj: "_".join(f"@{obj.fake_username}".split(" "))) # @Peter_Parker
|
|
|
|
class Meta:
|
|
exclude = ("channel_name", "fake_username")
|
|
|
|
|
|
class BotEntitleFactory(factory.DictFactory):
|
|
type = "bot_command"
|
|
offset = 0
|
|
length = 7
|
|
|
|
|
|
class BotMessageFactory(factory.DictFactory):
|
|
message_id = factory.Faker("random_int", min=10**8, max=10**9 - 1)
|
|
chat = factory.LazyFunction(lambda: BotChatFactory()._asdict())
|
|
date = time.time()
|
|
text = factory.Faker("text")
|
|
entities = factory.LazyFunction(lambda: [BotEntitleFactory()])
|
|
|
|
@classmethod
|
|
def create_instance(cls, **kwargs: Any) -> dict[str, Any]:
|
|
data = {**cls.build(**kwargs), "from": BotUserFactory()._asdict()}
|
|
return data
|
|
|
|
|
|
class BotUpdateFactory(factory.DictFactory):
|
|
update_id = factory.Faker("random_int", min=10**8, max=10**9 - 1)
|
|
message = factory.LazyFunction(lambda: BotMessageFactory.create_instance())
|
|
|
|
|
|
class CallBackFactory(factory.DictFactory):
|
|
id = factory.Faker("bothify", text="###################")
|
|
chat_instance = factory.Faker("bothify", text="###################")
|
|
message = factory.LazyFunction(lambda: BotMessageFactory.create_instance())
|
|
data = factory.fuzzy.FuzzyChoice(BotStagesEnum)
|
|
|
|
@classmethod
|
|
def create_instance(cls, **kwargs: Any) -> dict[str, Any]:
|
|
data = {**cls.build(**kwargs), "from": BotUserFactory()._asdict()}
|
|
return data
|
|
|
|
|
|
class BotCallBackQueryFactory(factory.DictFactory):
|
|
update_id = factory.Faker("random_int", min=10**8, max=10**9 - 1)
|
|
callback_query = factory.LazyFunction(lambda: BotMessageFactory.create_instance())
|