mirror of
https://github.com/Balshgit/gpt_chat_bot.git
synced 2025-09-10 17:20:41 +03:00
132 lines
4.2 KiB
Python
132 lines
4.2 KiB
Python
import string
|
|
import time
|
|
from typing import Any, NamedTuple
|
|
|
|
import factory.fuzzy
|
|
from faker import Faker
|
|
|
|
from constants import BotStagesEnum
|
|
from core.bot.models.chat_gpt import ChatGpt
|
|
from tests.integration.factories.utils import BaseModelFactory
|
|
|
|
faker = Faker("ru_RU")
|
|
|
|
|
|
class User(NamedTuple):
|
|
id: int
|
|
is_bot: bool
|
|
first_name: str | None
|
|
last_name: str | None
|
|
username: str | None
|
|
language_code: str
|
|
|
|
|
|
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 Chat(NamedTuple):
|
|
id: int
|
|
first_name: str | None
|
|
last_name: str | None
|
|
username: str
|
|
type: str
|
|
|
|
|
|
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 ChatGptModelFactory(BaseModelFactory):
|
|
id = factory.Sequence(lambda n: n + 1)
|
|
model = factory.Faker("word")
|
|
priority = factory.Faker("random_int", min=0, max=42)
|
|
|
|
class Meta:
|
|
model = ChatGpt
|
|
|
|
|
|
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 = 42
|
|
|
|
|
|
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()])
|
|
voice = None
|
|
|
|
@classmethod
|
|
def create_instance(cls, **kwargs: Any) -> dict[str, Any]:
|
|
return {**cls.build(**kwargs), "from": BotUserFactory()._asdict()}
|
|
|
|
|
|
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]:
|
|
return {**cls.build(**kwargs), "from": BotUserFactory()._asdict()}
|
|
|
|
|
|
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())
|
|
|
|
|
|
class BotVoiceFactory(factory.DictFactory):
|
|
duration = factory.Faker("random_int", min=1, max=700)
|
|
file_id = factory.Faker(
|
|
"lexify", text="????????????????????????????????????????????????????????????????????????", locale="en_US"
|
|
)
|
|
file_size = factory.Faker("random_int")
|
|
file_unique_id = factory.Faker("lexify", text="???????????????", locale="en_US")
|
|
mime_type = "audio/ogg"
|