From 52e42c5be77477355a9f776a6c2bc26f0b6b208a Mon Sep 17 00:00:00 2001 From: Egor Date: Fri, 3 Jul 2020 15:49:03 +0500 Subject: [PATCH] chore: move sequences to separate file --- tests/factories/sequences.py | 6 ++++++ tests/factories/user.py | 4 ++-- 2 files changed, 8 insertions(+), 2 deletions(-) create mode 100644 tests/factories/sequences.py 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