partial way through playing with Faker

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")