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
|
import random
from datetime import datetime
import factory
from django.contrib.auth import get_user_model
from factory import Faker, SubFactory, post_generation
from factory.django import DjangoModelFactory
from ctrack.organisations.models import Address, Organisation, Person, Role
from ctrack.register.models import SingleDateTimeEvent
from ctrack.users.tests.factories import UserFactory
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 OrganisationFactory(DjangoModelFactory):
class Meta:
model = Organisation
name = Faker("company")
slug = Faker("lexify", text="????", letters="abcdsg")
designation_type = 1
oes = factory.LazyFunction(lambda: random.choice([True, False]))
registered_company_name = Faker("company")
registered_company_number = Faker("numerify", text="######")
lead_inspector = factory.SubFactory(UserFactory)
deputy_lead_inspector = factory.SubFactory(UserFactory)
date_updated = Faker("date_this_year", before_today=True)
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 organisation(self, create, extracted, **kwargs):
# if not create:
# return
# if extracted:
# for org in extracted:
# self.organisation.add(org)
@post_generation
def role(self, create, extracted, **kwargs):
if not create:
return
if extracted:
self.role.add(extracted)
primary_nis_contact = factory.LazyFunction(lambda: random.choice([True, False]))
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 = factory.LazyAttribute(
lambda o: "%s@%s.com" % (o.first_name.lower(), o.organisation.slug)
)
secondary_email = Faker("ascii_company_email")
mobile = Faker("cellphone_number", locale="en_GB")
landline = Faker("phone_number", locale="en_GB")
date_updated = factory.LazyFunction(datetime.now)
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 = Faker("text", max_nb_chars=500, ext_word_list=None)
class AddressFactory(DjangoModelFactory):
# type = # manually entered
organisation = SubFactory(OrganisationFactory)
line1 = Faker("building_number", 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 = "UK"
other_details = Faker("text", max_nb_chars=200, ext_word_list=None)
class Meta:
model = Address
class SingleDateTimeEventFactory(DjangoModelFactory):
type_descriptor = None # must pass this in
short_description = "Nothing short description"
datetime = "2020-02-10 10:00"
comments = "Nothing comments"
location = "Gark's basin"
user = SubFactory(UserFactory)
class Meta:
model = SingleDateTimeEvent
|
