diff --git a/tests/factories/sequences.py b/tests/factories/sequences.py new file mode 100644 index 00000000..0d4de041 --- /dev/null +++ b/tests/factories/sequences.py @@ -0,0 +1,6 @@ +import factory + +id_ = factory.Sequence(lambda n: n) +first_name = factory.Sequence(lambda n: f"First name #{n}") +last_name = factory.Sequence(lambda n: f"Last name #{n}") +username = factory.Sequence(lambda n: f"Username #{n}") diff --git a/tests/factories/user.py b/tests/factories/user.py index e080f8bf..51227dd2 100644 --- a/tests/factories/user.py +++ b/tests/factories/user.py @@ -1,13 +1,13 @@ import factory from aiogram.api.types import User +from tests.factories import sequences class UserFactory(factory.Factory): class Meta: model = User - id = factory.Sequence(lambda n: n) + id = sequences.id_ first_name = factory.Sequence(lambda n: f"First name #{n}") - last_name = factory.Sequence(lambda n: f"Last name #{n}") is_bot = False