blob: b104a23021cc5c813d519e2f5b58058984f4675a (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
from typing import Any, Sequence
from django.contrib.auth import get_user_model
from ctrack.organisations.models import Organisation, Address
from factory import DjangoModelFactory, Faker, post_generation
class OrganisationFactory(DjangoModelFactory):
name = Faker("name")
class Meta:
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")
email = Faker("email")
name = Faker("name")
@post_generation
def password(self, create: bool, extracted: Sequence[Any], **kwargs):
password = Faker(
"password",
length=42,
special_chars=True,
digits=True,
upper_case=True,
lower_case=True,
).generate(extra_kwargs={})
self.set_password(password)
class Meta:
model = get_user_model()
django_get_or_create = ["username"]
|
