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
|
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 Address, Organisation, Person, Role
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="######")
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
|
