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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
|
import random
from datetime import datetime
import factory
from django.contrib.auth import get_user_model
from factory import DjangoModelFactory, Faker, SubFactory, post_generation
from ctrack.organisations.models import Mode, Organisation, Person, Role, Submode, Address, AddressType
User = get_user_model()
def _random_mode():
sms = [
"Aviation",
"Rail",
"Maritime",
"Animal",
"Stellar",
]
return sms[random.randint(0, len(sms) - 1)]
def _random_submode():
sms = [
"Light Rail",
"Heavy Rail",
"Passenger Rail",
"Animal Transport",
"Infrastructure",
]
return sms[random.randint(0, len(sms) - 1)]
class UserFactory(DjangoModelFactory):
class Meta:
model = User
username = Faker("lexify", text="???????", letters="abcdsgTGQA")
password = Faker("lexify", text="????????", letters="AdOIqkcvBnMP")
class ModeFactory(DjangoModelFactory):
class Meta:
model = Mode
descriptor = factory.LazyFunction(_random_mode)
class SubModeFactory(DjangoModelFactory):
class Meta:
model = Submode
descriptor = factory.LazyFunction(_random_submode)
mode = SubFactory(ModeFactory)
class OrganisationFactory(DjangoModelFactory):
class Meta:
model = Organisation
name = Faker("company")
slug = Faker("lexify", text="????", letters="abcdsg")
submode = SubFactory(SubModeFactory)
designation_type = 1
registered_company_name = Faker("company")
registered_company_number = Faker("numerify", text="######")
date_updated = Faker("date_this_year", before_today=True)
updated_by = SubFactory(UserFactory)
comments = Faker("paragraph", nb_sentences=3)
active = True
class RoleFactory(DjangoModelFactory):
class Meta:
model = Role
name = Faker("job")
class PersonFactory(DjangoModelFactory):
class Meta:
model = Person
@post_generation
def role(self, create, extracted, **kwargs):
if not create:
return
if extracted:
self.role.add(extracted)
primary_nis_contact = True
voluntary_point_of_contact = True
has_egress = False
title = factory.LazyFunction(lambda: random.randint(1, 8))
job_title = Faker("job")
first_name = Faker("first_name")
last_name = Faker("last_name")
organisation = SubFactory("ctrack.organisations.tests.factories.OrganisationFactory")
email = Faker("ascii_company_email")
secondary_email = "ascii_company_email"
mobile = Faker("msisdn", locale="en_GB")
landline = Faker("phone_number", locale="en_GB")
date_updated = factory.LazyFunction(datetime.now)
updated_by = SubFactory(UserFactory)
clearance = factory.LazyFunction(lambda: random.randint(1,6))
clearance_sponsor = Faker("name", locale="en_GB")
clearance_start_date = factory.LazyFunction(datetime.now)
clearance_last_checked = factory.LazyFunction(datetime.now)
clearance_expiry = factory.LazyFunction(datetime.now)
active = True
date_ended = Faker("date_this_year")
predecessor = SubFactory("ctrack.organisations.tests.factories.PersonFactory")
comments = "Yaa!"
class AddressFactory(DjangoModelFactory):
type = SubFactory("ctrack.organisations.tests.factories.AddressTypeFactory")
organisation = SubFactory(OrganisationFactory)
line1 = Faker("secondary_address", locale="en_GB")
line2 = Faker("street_name", locale="en_GB")
line3 = Faker("secondary_address", locale="en_GB")
city = Faker("city", locale="en_GB")
county = Faker("lexify", locale="en_GB", text="??????", letters="aeioutzyj")
postcode = Faker("postcode", locale="en_GB")
country = Faker("country")
other_details = Faker("lexify", locale="en_GB", text="??????", letters="aeioutzyj")
class Meta:
model = Address
class AddressTypeFactory(DjangoModelFactory):
descriptor = "Primary Address"
class Meta:
model = AddressType
|
