diff options
Diffstat (limited to 'ctrack/users/tests/factories.py')
-rw-r--r-- | ctrack/users/tests/factories.py | 16 |
1 files changed, 15 insertions, 1 deletions
diff --git a/ctrack/users/tests/factories.py b/ctrack/users/tests/factories.py index d4b9544..b104a23 100644 --- a/ctrack/users/tests/factories.py +++ b/ctrack/users/tests/factories.py @@ -1,7 +1,7 @@ from typing import Any, Sequence from django.contrib.auth import get_user_model -from ctrack.organisations.models import Organisation +from ctrack.organisations.models import Organisation, Address from factory import DjangoModelFactory, Faker, post_generation @@ -13,6 +13,20 @@ class OrganisationFactory(DjangoModelFactory): model = Organisation +class AddressFactory(DjangoModelFactory): + line1 = Faker("line1") + line2 = Faker("line2") + line3 = Faker("line3") + city = Faker("city") + county = Faker("county") + postcode = Faker("postcode") + country = Faker("country") + other_details = Faker("other_details") + + class Meta: + model = Address + + class UserFactory(DjangoModelFactory): username = Faker("user_name") |