def test_definition(self):
"""Tests all ways of defining a FuzzyInteger."""
fuzz = fuzzy.FuzzyInteger(2, 3)
for _i in range(20):
res = utils.evaluate_declaration(fuzz)
self.assertIn(res, [2, 3])
fuzz = fuzzy.FuzzyInteger(4)
for _i in range(20):
res = utils.evaluate_declaration(fuzz)
self.assertIn(res, [0, 1, 2, 3, 4])
class UserProfileFactory(factory.django.DjangoModelFactory):
"""Factory for UserProfile model."""
date_joined_program = fuzzy.FuzzyDateTime(START_DT)
local_name = 'Local Name'
birth_date = datetime.date(1992, 1, 1)
city = 'User city'
region = 'User region'
country = fuzzy.FuzzyChoice(COUNTRIES)
lat = fuzzy.FuzzyInteger(-90, 90)
lon = fuzzy.FuzzyInteger(-180, 180)
display_name = factory.Sequence(lambda n: 'UserProfile%s' % n)
private_email = factory.Sequence(lambda n: '*****@*****.**' % n)
mozillians_profile_url = 'https://mozillians.org/'
irc_name = factory.Sequence(lambda n: 'user%s' % n)
twitter_account = factory.Sequence(lambda n: 'user%s' % n)
facebook_url = factory.Sequence(lambda n: 'http://facebook.com/user%s' % n)
linkedin_url = 'http://linkedin.com/profile/'
wiki_profile_url = 'https://wiki.mozilla.org/User:'******'userprofile_email_mentor_notification'
pre_save.disconnect(email_mentor_notification, UserProfile, dispatch_uid=dispatch_uid)
profile = super(UserProfileFactory, cls)._create(target_class, *args, **kwargs)
pre_save.connect(email_mentor_notification, UserProfile, dispatch_uid=dispatch_uid)
return profile
@factory.post_generation
def functional_areas(self, create, extracted, **kwargs):
"""Add functional areas list after object generation."""
if not create:
return
if extracted:
for area in extracted:
self.functional_areas.add(area)
else:
# create random
for i in range(randint(1, 6)):
area = FunctionalAreaFactory.create()
self.functional_areas.add(area)
@factory.post_generation
def initial_council(self, create, extracted, **kwargs):
"""Create userprofile with self as mentor."""
if not create:
return
if extracted:
self.mentor = self.user
class AppliedIndicatorFactory(factory.django.DjangoModelFactory):
class Meta:
model = models.AppliedIndicator
indicator = factory.SubFactory(IndicatorBlueprintFactory)
lower_result = factory.SubFactory(LowerResultFactory)
context_code = fuzzy.FuzzyText(length=5)
target = factory.Dict({
'd': fuzzy.FuzzyInteger(0, 5),
'v': fuzzy.FuzzyInteger(10, 20)
})
class SchoolWeeklyStatusFactory(django_factory.DjangoModelFactory):
school = SubFactory(SchoolFactory)
year = fuzzy.FuzzyInteger(1900, 2200)
week = fuzzy.FuzzyInteger(1, 53)
connectivity_type = fuzzy.FuzzyText(length=64)
connectivity_speed = fuzzy.FuzzyInteger(1, 1000000)
connectivity_latency = fuzzy.FuzzyInteger(1, 100)
class Meta:
model = SchoolWeeklyStatus
def test_definition(self):
"""Tests all ways of defining a FuzzyInteger."""
fuzz = fuzzy.FuzzyInteger(2, 3)
for _i in range(20):
res = fuzz.evaluate(2, None, False)
self.assertIn(res, [2, 3])
fuzz = fuzzy.FuzzyInteger(4)
for _i in range(20):
res = fuzz.evaluate(2, None, False)
self.assertIn(res, [0, 1, 2, 3, 4])
class CertifiedFilesHistoryFactory(factory.Factory):
class Meta:
model = jobModels.CertifiedFilesHistory
certified_files_history_id = None
certify_history_id = fuzzy.FuzzyInteger(9999)
submission_id = fuzzy.FuzzyInteger(9999)
filename = fuzzy.FuzzyText()
file_type_id = fuzzy.FuzzyInteger(9999)
warning_filename = fuzzy.FuzzyText()
comment = fuzzy.FuzzyText()
class ObjectClassProgramActivityFactory(factory.Factory):
class Meta:
model = stagingModels.ObjectClassProgramActivity
object_class_program_activity_id = None
submission_id = fuzzy.FuzzyInteger(9999)
job_id = fuzzy.FuzzyInteger(9999)
row_number = fuzzy.FuzzyInteger(9999)
agency_identifier = fuzzy.FuzzyText()
allocation_transfer_agency = fuzzy.FuzzyText()
availability_type_code = fuzzy.FuzzyText()
beginning_period_of_availa = fuzzy.FuzzyText()
by_direct_reimbursable_fun = fuzzy.FuzzyText()
deobligations_recov_by_pro_cpe = fuzzy.FuzzyDecimal(9999)
ending_period_of_availabil = fuzzy.FuzzyText()
gross_outlay_amount_by_pro_cpe = fuzzy.FuzzyDecimal(9999)
gross_outlay_amount_by_pro_fyb = fuzzy.FuzzyDecimal(9999)
gross_outlays_delivered_or_cpe = fuzzy.FuzzyDecimal(9999)
gross_outlays_delivered_or_fyb = fuzzy.FuzzyDecimal(9999)
gross_outlays_undelivered_cpe = fuzzy.FuzzyDecimal(9999)
gross_outlays_undelivered_fyb = fuzzy.FuzzyDecimal(9999)
main_account_code = fuzzy.FuzzyText()
object_class = fuzzy.FuzzyText()
obligations_delivered_orde_cpe = fuzzy.FuzzyDecimal(9999)
obligations_delivered_orde_fyb = fuzzy.FuzzyDecimal(9999)
obligations_incurred_by_pr_cpe = fuzzy.FuzzyDecimal(9999)
obligations_undelivered_or_cpe = fuzzy.FuzzyDecimal(9999)
obligations_undelivered_or_fyb = fuzzy.FuzzyDecimal(9999)
program_activity_code = fuzzy.FuzzyText()
program_activity_name = fuzzy.FuzzyText()
sub_account_code = fuzzy.FuzzyText()
ussgl480100_undelivered_or_cpe = fuzzy.FuzzyDecimal(9999)
ussgl480100_undelivered_or_fyb = fuzzy.FuzzyDecimal(9999)
ussgl480200_undelivered_or_cpe = fuzzy.FuzzyDecimal(9999)
ussgl480200_undelivered_or_fyb = fuzzy.FuzzyDecimal(9999)
ussgl483100_undelivered_or_cpe = fuzzy.FuzzyDecimal(9999)
ussgl483200_undelivered_or_cpe = fuzzy.FuzzyDecimal(9999)
ussgl487100_downward_adjus_cpe = fuzzy.FuzzyDecimal(9999)
ussgl487200_downward_adjus_cpe = fuzzy.FuzzyDecimal(9999)
ussgl488100_upward_adjustm_cpe = fuzzy.FuzzyDecimal(9999)
ussgl488200_upward_adjustm_cpe = fuzzy.FuzzyDecimal(9999)
ussgl490100_delivered_orde_cpe = fuzzy.FuzzyDecimal(9999)
ussgl490100_delivered_orde_fyb = fuzzy.FuzzyDecimal(9999)
ussgl490200_delivered_orde_cpe = fuzzy.FuzzyDecimal(9999)
ussgl490800_authority_outl_cpe = fuzzy.FuzzyDecimal(9999)
ussgl490800_authority_outl_fyb = fuzzy.FuzzyDecimal(9999)
ussgl493100_delivered_orde_cpe = fuzzy.FuzzyDecimal(9999)
ussgl497100_downward_adjus_cpe = fuzzy.FuzzyDecimal(9999)
ussgl497200_downward_adjus_cpe = fuzzy.FuzzyDecimal(9999)
ussgl498100_upward_adjustm_cpe = fuzzy.FuzzyDecimal(9999)
ussgl498200_upward_adjustm_cpe = fuzzy.FuzzyDecimal(9999)
tas = fuzzy.FuzzyText()
disaster_emergncy_fund_code = fuzzy.FuzzyText()
class LineScoreFactory(DjangoModelFactory):
class Meta:
model = LineScore
game = factory.SubFactory(GameFactory)
game_date_est = factory.LazyAttribute(lambda obj: obj.game.game_date_est)
game_sequence = factory.LazyAttribute(lambda obj: obj.game.game_sequence)
team = None
pts_qtr1 = fuzzy.FuzzyInteger(low=0, high=50)
pts_qtr2 = fuzzy.FuzzyInteger(low=0, high=50)
pts_qtr3 = fuzzy.FuzzyInteger(low=0, high=50)
pts_qtr4 = fuzzy.FuzzyInteger(low=0, high=50)
pts_ot1 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger,
prob_null=0.8,
low=0,
high=50)
pts_ot2 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger,
prob_null=0.8,
low=0,
high=50)
pts_ot3 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger,
prob_null=0.8,
low=0,
high=50)
pts_ot4 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger,
prob_null=0.8,
low=0,
high=50)
pts_ot5 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger,
prob_null=0.8,
low=0,
high=50)
pts_ot6 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger,
prob_null=0.8,
low=0,
high=50)
pts_ot7 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger,
prob_null=0.8,
low=0,
high=50)
pts_ot8 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger,
prob_null=0.8,
low=0,
high=50)
pts_ot9 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger,
prob_null=0.8,
low=0,
high=50)
pts_ot10 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger,
prob_null=0.8,
low=0,
high=50)
pts = fuzzy.FuzzyInteger(low=0, high=50)
class CodingResultFactory(factory.django.DjangoModelFactory):
class Meta:
model = CodingResult
coder_facebook_id = "123456"
question_solved_id = fuzzy.FuzzyInteger(0, 1000)
last_testcase_passed_index = fuzzy.FuzzyInteger(0, 1000)
coder_source_code = coding_result_data['coder_source_code']
error = coding_result_data['error']
scores = coding_result_data['scores']
possible_total = fuzzy.FuzzyInteger(0, 1000)
class AssetValueFactory(factory.DjangoModelFactory):
class Meta:
model = AssetValue
# class Params:
asset_id = factory.SubFactory(AssetFactory)
reporting_date = faker.date()
opening_price = fuzzy.FuzzyInteger(0)
settlement_price = fuzzy.FuzzyInteger(0)
trading_volume = fuzzy.FuzzyInteger(0, high=100)
currency = fuzzy.FuzzyText(length=5)
class FABSFactory(factory.Factory):
class Meta:
model = stagingModels.FABS
fabs_id = None
afa_generated_unique = fuzzy.FuzzyText()
submission_id = fuzzy.FuzzyInteger(9999)
job_id = fuzzy.FuzzyInteger(9999)
row_number = fuzzy.FuzzyInteger(9999)
action_date = fuzzy.FuzzyText()
action_type = fuzzy.FuzzyText()
assistance_type = fuzzy.FuzzyText()
award_description = fuzzy.FuzzyText()
awardee_or_recipient_legal = fuzzy.FuzzyText()
awardee_or_recipient_uniqu = fuzzy.FuzzyText()
awarding_agency_code = fuzzy.FuzzyText()
awarding_office_code = fuzzy.FuzzyText()
awarding_sub_tier_agency_c = fuzzy.FuzzyText()
award_modification_amendme = fuzzy.FuzzyText()
business_funds_indicator = fuzzy.FuzzyText()
business_types = fuzzy.FuzzyText()
correction_delete_indicatr = fuzzy.FuzzyText()
face_value_loan_guarantee = fuzzy.FuzzyDecimal(9999)
fain = fuzzy.FuzzyText()
federal_action_obligation = fuzzy.FuzzyDecimal(9999)
funding_agency_code = fuzzy.FuzzyText()
funding_office_code = fuzzy.FuzzyText()
funding_opportunity_goals = fuzzy.FuzzyText()
funding_opportunity_number = fuzzy.FuzzyText()
funding_sub_tier_agency_co = fuzzy.FuzzyText()
indirect_federal_sharing = fuzzy.FuzzyDecimal(9999)
legal_entity_address_line1 = fuzzy.FuzzyText()
legal_entity_address_line2 = fuzzy.FuzzyText()
legal_entity_country_code = fuzzy.FuzzyText()
legal_entity_foreign_city = fuzzy.FuzzyText()
legal_entity_foreign_posta = fuzzy.FuzzyText()
legal_entity_foreign_provi = fuzzy.FuzzyText()
legal_entity_zip5 = fuzzy.FuzzyText()
legal_entity_zip_last4 = fuzzy.FuzzyText()
non_federal_funding_amount = fuzzy.FuzzyDecimal(9999)
original_loan_subsidy_cost = fuzzy.FuzzyDecimal(9999)
period_of_performance_curr = fuzzy.FuzzyText()
period_of_performance_star = fuzzy.FuzzyText()
place_of_performance_code = fuzzy.FuzzyText()
place_of_performance_congr = fuzzy.FuzzyText()
place_of_perform_country_c = fuzzy.FuzzyText()
place_of_performance_forei = fuzzy.FuzzyText()
place_of_performance_zip4a = fuzzy.FuzzyText()
record_type = fuzzy.FuzzyInteger(1, 2)
sai_number = fuzzy.FuzzyText()
uri = fuzzy.FuzzyText()
is_valid = True
class PlayerFactory(django.DjangoModelFactory):
"""
Factory-boy for Player Model.
"""
class Meta:
model = PlayerModel
pk = fuzzy.FuzzyInteger(min_value, max_value)
team = SubFactory(TeamFactory)
first_name = fuzzy.FuzzyText(length=10)
last_name = fuzzy.FuzzyText(length=10)
country = fuzzy.FuzzyText(length=10)
jersey_number = fuzzy.FuzzyInteger(min_value, max_value)
class FillLevelFactory(factory.DjangoModelFactory):
class Meta:
model = EnevoFillLevel
time = fuzzy.FuzzyDateTime(start_dt=timezone.now())
fill_level = fuzzy.FuzzyInteger(0, 100)
e_site = fuzzy.FuzzyInteger(0, 2000)
e_site_name = fuzzy.FuzzyText(length=100)
e_site_content_type = fuzzy.FuzzyInteger(0, 5)
content_type = fuzzy.FuzzyInteger(0, 5)
content_type_name = fuzzy.FuzzyText(length=4)
confidence = fuzzy.FuzzyInteger(0, 100)
class InventoryFactory(factory.Factory):
class Meta:
model = Inventory
product_id = factory.Sequence(lambda n: n)
condition = fuzzy.FuzzyChoice(choices=keys.CONDITIONS)
quantity = fuzzy.FuzzyInteger(keys.QTY_LOW, keys.QTY_HIGH, keys.QTY_STEP)
restock_level = fuzzy.FuzzyInteger(keys.QTY_LOW, keys.RESTOCK_LVL,
keys.QTY_STEP)
available = fuzzy.FuzzyChoice(
choices=[keys.AVAILABLE_TRUE, keys.AVAILABLE_FALSE])
######################################################################
## Helper
######################################################################
@staticmethod
def get_data(type=keys.ATTR_DEFAULT):
"""
This extracts command line arguments
"""
if len(sys.argv) < keys.MAX_ATTR + 1:
return None, None, None, None, None
if type == keys.ATTR_DEFAULT or len(sys.argv) > keys.MAX_ATTR:
pid = sys.argv[keys.ATTR_PRODUCT_ID]
cnd = sys.argv[keys.ATTR_CONDITION]
qty = sys.argv[keys.ATTR_QUANTITY]
lvl = sys.argv[keys.ATTR_RESTOCK_LEVEL]
avl = sys.argv[keys.ATTR_AVAILABLE]
return pid, cnd, qty, lvl, avl
sys.argv[type]
@staticmethod
def read_test_data():
"""
Read data for test cases into test_data
"""
data = []
global test_data_file
try:
fr = open("./tests/data_to_test.csv", 'r')
line = fr.readline()
while line:
line = fr.readline()
row = line.strip().split(',')
if row[0]:
data.append(row)
fr.close()
except IOError as err:
print(err)
finally:
return data
class FileRequestFactory(factory.Factory):
class Meta:
model = jobModels.FileRequest
file_request_id = fuzzy.FuzzyInteger(9999)
request_date = fuzzy.FuzzyDate(date(2010, 1, 1))
job_id = fuzzy.FuzzyInteger(9999)
job = factory.SubFactory(JobFactory)
parent_job_id = fuzzy.FuzzyInteger(9999)
start_date = fuzzy.FuzzyDate(date(2010, 1, 1))
end_date = fuzzy.FuzzyDate(date(2010, 1, 1))
agency_code = fuzzy.FuzzyText()
file_type = fuzzy.FuzzyText()
is_cached_file = fuzzy.FuzzyChoice((False, True))
class AlertFactory(factory.DjangoModelFactory):
class Meta:
model = EnevoAlert
id = factory.Sequence(lambda n: n)
type = fuzzy.FuzzyInteger(0, 50)
type_name = fuzzy.FuzzyText(length=30)
reported = fuzzy.FuzzyDateTime(start_dt=timezone.now())
last_observed = fuzzy.FuzzyDateTime(start_dt=timezone.now())
site_name = fuzzy.FuzzyText(length=30)
area = fuzzy.FuzzyInteger(0, 50)
area_name = fuzzy.FuzzyText(length=30)
content_type_name = fuzzy.FuzzyText(length=30)
start = fuzzy.FuzzyDateTime(start_dt=timezone.now())
class ContainerTypeFactory(factory.DjangoModelFactory):
class Meta:
model = EnevoContainerType
id = factory.Sequence(lambda n: n)
name = fuzzy.FuzzyText(length=40)
volume = fuzzy.FuzzyInteger(0, 50)
sensor_height = fuzzy.FuzzyFloat(0.5, 100.0)
full_height = fuzzy.FuzzyFloat(0.5, 100.0)
shape = fuzzy.FuzzyText(length=20)
has_bag = fuzzy.FuzzyInteger(0, 1)
servicing_amount = fuzzy.FuzzyText(length=20)
servicing_method = fuzzy.FuzzyText(length=20)
last_modified = fuzzy.FuzzyDateTime(start_dt=timezone.now())
class ItemFactory(factory.Factory):
class Meta:
model = Item
buildyear = fuzzy.FuzzyInteger(0, 9999999999)
charges_parkingspace = fuzzy.FuzzyDecimal(0, 99999999999)
chargesfinancebasemonth = fuzzy.FuzzyDecimal(0, 99999999999)
chargesmaintbasemonth = fuzzy.FuzzyDecimal(0, 99999999999)
chargeswater_period = fuzzy.FuzzyText()
condition_name = fuzzy.FuzzyChoice([c for c in Condition])
coordinate = factory.List([factory.SubFactory(CoordinateFactory)])
country = fuzzy.FuzzyChoice([c for c in Country])
currency_code = "EUR"
cust_itemcode = fuzzy.FuzzyText()
debtfreeprice = fuzzy.FuzzyDecimal(0, 99999999999)
extralink = factory.List(
[factory.SubFactory(ExtraLinkFactory) for _ in range(2)])
floors = fuzzy.FuzzyInteger(0, 9999999999)
energyclass = fuzzy.FuzzyText(length=10)
holdingtype = fuzzy.FuzzyChoice([c for c in HoldingType])
image = factory.List([factory.SubFactory(ImageFactory) for _ in range(2)])
itemgroup = fuzzy.FuzzyChoice([c for c in ItemGroup])
livingaream2 = fuzzy.FuzzyDecimal(0, 99999999999)
loclvlid = 1
locsourceid = 4
lotarea = fuzzy.FuzzyDecimal(0, 99999999999)
lotareaunitcode = fuzzy.FuzzyText(length=2)
lotholding = fuzzy.FuzzyText()
postcode = fuzzy.FuzzyText(length=5)
price = fuzzy.FuzzyDecimal(0, 99999999999)
price_m2 = fuzzy.FuzzyDecimal(0, 99999999999)
quarteroftown = fuzzy.FuzzyText(length=60)
realtygroup = fuzzy.FuzzyChoice([c for c in RealtyGroup])
realtyidentifier = fuzzy.FuzzyText(length=40)
realty_itemgroup = fuzzy.FuzzyChoice([c for c in ItemGroup])
realtytype = fuzzy.FuzzyChoice([c for c in RealtyType])
realtyoption = factory.List([fuzzy.FuzzyText() for _ in range(2)])
rc_parkingspace_count = fuzzy.FuzzyInteger(0, 9999999999)
roomcount = fuzzy.FuzzyInteger(0, 9999999999)
scontact = factory.List(
[factory.SubFactory(ScontactFactory) for _ in range(2)])
showingdate = fuzzy.FuzzyDateTime(timezone.now())
showing_date2 = fuzzy.FuzzyDateTime(timezone.now())
street = fuzzy.FuzzyText(length=200)
supplier_source_itemcode = fuzzy.FuzzyText()
text = factory.List([factory.SubFactory(TextFactory) for _ in range(2)])
town = fuzzy.FuzzyText(length=60)
tradetype = fuzzy.FuzzyChoice([c for c in TradeType])
zoningname = fuzzy.FuzzyText()
class CardFactory(factory.django.DjangoModelFactory):
class Meta:
model = models.Card
name = factory.Faker('word')
image = factory.django.ImageField()
attributes = models.Attribute.objects.all().first()
rarity = fuzzy.FuzzyInteger(*CARD_RARITY_RANGE)
summon_cost = fuzzy.FuzzyInteger(*GENERAL_RANGE)
soultrap_cost = fuzzy.FuzzyInteger(*GENERAL_RANGE)
magicka_cost = fuzzy.FuzzyInteger(*GENERAL_RANGE)
mechanics_text = factory.Faker('word')
flavor_text = factory.Faker('sentence')
is_collectible = fuzzy.FuzzyChoice([True, False])
expansion = factory.Iterator(models.Expansion.objects.all())
class SubmissionFactory(factory.Factory):
class Meta:
model = jobModels.Submission
submission_id = None
user_id = None
cgac_code = fuzzy.FuzzyText()
reporting_start_date = fuzzy.FuzzyDate(date(2010, 1, 1))
reporting_end_date = fuzzy.FuzzyDate(date(2010, 1, 1))
reporting_fiscal_year = fuzzy.FuzzyInteger(2010, 2040)
reporting_fiscal_period = fuzzy.FuzzyInteger(1, 4)
is_quarter_format = False
publishable = False
number_of_errors = 0
number_of_warnings = 0
class ScanFactory(factory.DjangoModelFactory):
class Meta:
model = models.Scan
scan_id = fuzzy.FuzzyInteger(0)
latitude = lat1
longitude = lon1
scan_moment = fuzzy.FuzzyDateTime(datetime.datetime(2017, 1, 1,
tzinfo=UTC))
parkeervak_id = fuzzy.FuzzyInteger(0, 100)
bgt_wegdeel = fuzzy.FuzzyInteger(222, 234)
geometrie = Point(uniform(lon1, lon2), uniform(lat1, lat2), srid=4326)
class TransponderFactory(factory.django.DjangoModelFactory):
"""Transponder model factory."""
description = fuzzy.FuzzyText()
alive = fuzzy.FuzzyChoice(choices=[True, False])
uplink_low = fuzzy.FuzzyInteger(200, 500)
uplink_high = fuzzy.FuzzyInteger(200, 500)
downlink_low = fuzzy.FuzzyInteger(200, 500)
downlink_high = fuzzy.FuzzyInteger(200, 500)
mode = fuzzy.FuzzyChoice(choices=MODE_CHOICES)
invert = fuzzy.FuzzyChoice(choices=[True, False])
baud = fuzzy.FuzzyInteger(4000, 22000, step=1000)
satellite = factory.SubFactory(SatelliteFactory)
class Meta:
model = Transponder
class SitueringFactory(factory.DjangoModelFactory):
class Meta(object):
model = models.Situering
betreft_nummeraanduiding = fuzzy.FuzzyInteger(low=0)
# situering_nummeraanduiding = models.CharField(length=128, null=True)
eerste_situering = fuzzy.FuzzyAttribute(gen_janee)
huisnummer = fuzzy.FuzzyInteger(low=1)
if random.randint(0, 50) % 3 == 0:
huisletter = fuzzy.FuzzyText(length=1)
if random.randint(0, 50) % 3 == 0:
huisnummertoevoeging = fuzzy.FuzzyText(length=4)
postcode = fuzzy.FuzzyText(length=6)
straat = fuzzy.FuzzyText(length=80)
class EventFactory(factory.django.DjangoModelFactory):
"""Factory for Event model."""
FACTORY_FOR = Event
name = factory.Sequence(lambda n: 'Event #%s' % n)
start = fuzzy.FuzzyDateTime(START_DT, END_DT)
end = fuzzy.FuzzyDateTime(END_DT + datetime.timedelta(days=3))
timezone = fuzzy.FuzzyChoice(common_timezones)
venue = 'VenueName'
city = 'CityName'
region = 'RegionName'
country = fuzzy.FuzzyChoice(COUNTRIES)
lat = fuzzy.FuzzyInteger(-90, 90)
lon = fuzzy.FuzzyInteger(-180, 180)
external_link = 'example.com'
owner = factory.SubFactory(UserFactory)
estimated_attendance = fuzzy.FuzzyChoice(ATTENDANCE_CHOICES)
description = 'This is an event description.'
extra_content = 'Extra content for event page.'
mozilla_event = fuzzy.FuzzyChoice([True, False])
hashtag = 'EventHashtag'
converted_visitors = fuzzy.FuzzyInteger(100)
swag_bug = factory.SubFactory(BugFactory)
budget_bug = factory.SubFactory(BugFactory)
times_edited = fuzzy.FuzzyInteger(10)
@factory.post_generation
def categories(self, create, extracted, **kwargs):
"""Add event categories after event creation."""
if not create:
return
if extracted:
for category in extracted:
self.categories.add(category)
else:
# add random number of categories
for i in range(randint(1, 6)):
area = FunctionalAreaFactory.create()
self.categories.add(area)
@factory.post_generation
def metrics(self, create, extracted, **kwargs):
"""Add event metrics after event creation."""
if not create:
return
# create 2 metrics by default
MetricFactory.create_batch(2, event=self)
class DriveFactory(factory.DjangoModelFactory):
class Meta:
model = Drive
driver = factory.SubFactory(UserFactory)
car = factory.SubFactory(CarFactory)
date = fuzzy.FuzzyDate((now() - timedelta(days=1000)).date())
start_mileage = fuzzy.FuzzyInteger(1000000)
description = factory.Faker('text', max_nb_chars=1000)
@factory.lazy_attribute
def end_mileage(self):
return random.randint(self.start_mileage, 1000000)
@factory.post_generation
def passengers(self, create, extracted, **kwargs):
if not create:
return
if extracted:
for passenger in extracted:
self.passengers.add(passenger)
@factory.post_generation
def project_set(self, create, extracted, **kwargs):
if not create:
return
if extracted:
for project in extracted:
self.project_set.add(project)
class KiloFactory(factory.DjangoModelFactory):
class Meta:
model = KilogramWeighMeasurement
id = factory.Sequence(lambda n: n)
buurt_code = fuzzy.FuzzyText(length=4)
stadsdeel = fuzzy.FuzzyText(length=1)
seq_id = fuzzy.FuzzyInteger(1, 2000)
weigh_at = fuzzy.FuzzyDateTime(
datetime.datetime(2018, 10, 16, tzinfo=UTC)
)
first_weight = fuzzy.FuzzyInteger(1, 2000)
second_weight = fuzzy.FuzzyInteger(1000, 3000)
net_weight = fuzzy.FuzzyInteger(1000, 3000)
geometrie = get_puntje()
class OfferingFactory(factory.DjangoModelFactory):
class Meta:
model = models.Offering
issue = factory.SubFactory(IssueFactory)
unit_price = fuzzy.FuzzyInteger(1, 10)
project = factory.SelfAttribute('issue.project')
template = factory.SubFactory(OfferingTemplateFactory)
@classmethod
def get_url(cls, offering=None, action=None):
if offering is None:
offering = OfferingFactory()
url = 'http://testserver' + reverse('support-offering-detail',
kwargs={'uuid': offering.uuid.hex})
return url if action is None else url + action + '/'
@classmethod
def get_list_url(cls):
return 'http://testserver' + reverse('support-offering-list')
@classmethod
def get_list_action_url(cls, action=None):
url = 'http://testserver' + reverse('support-offering-list')
return url if action is None else url + action + '/'
class DepartmentFactory(factory.DjangoModelFactory):
id = factory.Faker('uuid4')
name = FuzzyUnicode()
number = fuzzy.FuzzyInteger(low=1, high=99)
class Meta:
model = Department
class DriveFactory(MakeFactoryMixin, DjangoModelFactory):
class Meta:
model = Drive
driver = SubFactory(UserFactory)
project = SubFactory(ProjectFactory)
passenger = SubFactory(UserFactory)
car = SubFactory(CarFactory)
date = fuzzy.FuzzyDate((now() - timedelta(days=1000)).date())
start_mileage = fuzzy.FuzzyInteger(1, 1000)
end_mileage = fuzzy.FuzzyInteger(1000, 9999)
description = Faker("text", max_nb_chars=1000)
timestamp = fuzzy.FuzzyInteger(1, 999999999)
start_location = Faker("city", locale="uk_UA")
end_location = Faker("city", locale="uk_UA")
is_verified = False
class RecipeIngredientFactory(django.DjangoModelFactory):
recipe = factory.SubFactory(RecipeFactory)
ingredient = factory.SubFactory(Ingredient)
count = fuzzy.FuzzyInteger(1, 40)
class Meta:
model = RecipeIngredient
