diff options
Diffstat (limited to 'ctrack/users/tests/factories.py')
-rw-r--r-- | ctrack/users/tests/factories.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/ctrack/users/tests/factories.py b/ctrack/users/tests/factories.py index b537136..d4b9544 100644 --- a/ctrack/users/tests/factories.py +++ b/ctrack/users/tests/factories.py @@ -1,9 +1,18 @@ from typing import Any, Sequence from django.contrib.auth import get_user_model +from ctrack.organisations.models import Organisation from factory import DjangoModelFactory, Faker, post_generation +class OrganisationFactory(DjangoModelFactory): + + name = Faker("name") + + class Meta: + model = Organisation + + class UserFactory(DjangoModelFactory): username = Faker("user_name") |