aboutsummaryrefslogtreecommitdiffstats
path: root/ctrack/users/tests/factories.py
diff options
context:
space:
mode:
Diffstat (limited to 'ctrack/users/tests/factories.py')
-rw-r--r--ctrack/users/tests/factories.py16
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")