class Params: unpublished = Trait(publish_after=Faker( 'future_datetime', end_date='+30d', tzinfo=timezone.utc)) has_expiry = Trait(expires=Faker( 'future_datetime', end_date='+30d', tzinfo=timezone.utc)) expired = Trait(expires=Faker( 'past_datetime', start_date='-30d', tzinfo=timezone.utc))
class Params: paid = True free = Trait( box_count=1, bandwidth=100, forwards=2, reserved_config=0, cost=0, name="free", stripe_id=None, ) waiting = Trait( box_count=0, bandwidth=0, forwards=0, reserved_config=0, cost=0, name="waiting", stripe_id=None, ) beta = Trait( box_count=2, bandwidth=1000, forwards=10, reserved_config=1, cost=0, name="beta", stripe_id=None, )
class Params: active = True old_levels = Trait( _using_results_framework=ProgramM.NOT_MIGRATED ) has_rf = Trait( generate_levels=PostGeneration(generate_mc_levels) )
class Params: group = Trait(profile=SubFactory(BasicUserProfileFactory, group=True), name=Faker('color_name')) active = Trait( profile=SubFactory(BasicUserProfileFactory, is_active=True)) extended_profile = Trait( profile=SubFactory(ExtendedUserProfileFactory)) use_custom_name = Trait( profile=SubFactory(BasicUserProfileFactory, use_custom_name=True))
class Params: unexpired = Trait( enrollment=now_in_utc() + timedelta(days=1), certificate=now_in_utc() + timedelta(days=1), current_grade=now_in_utc() + timedelta(days=1), ) expired = Trait( enrollment=now_in_utc() - timedelta(days=1), certificate=now_in_utc() - timedelta(days=1), current_grade=now_in_utc() - timedelta(days=1), )
class Params: # pylint: disable=missing-docstring percent = Trait( amount_type='percent-discount', amount=FuzzyDecimal(0, 1), ) fixed = Trait( amount_type='fixed-discount', amount=FuzzyDecimal(50, 1000), ) program = Trait(content_object=SubFactory(ProgramFactory)) course = Trait(content_object=SubFactory(CourseFactory)) course_run = Trait(content_object=SubFactory(CourseRunFactory))
class Params: unpublished = Trait( publish_after=Faker('date_time', tzinfo=timezone.utc ) if RANDOM_SEED and not TESTING else Faker('future_datetime', end_date='+30d', tzinfo=timezone.utc)) has_expiry = Trait( expires=Faker('date_time', tzinfo=timezone.utc ) if RANDOM_SEED and not TESTING else Faker('future_datetime', end_date='+30d', tzinfo=timezone.utc)) expired = Trait( expires=Faker('date_time', tzinfo=timezone.utc ) if RANDOM_SEED and not TESTING else Faker('past_datetime', start_date='-30d', tzinfo=timezone.utc)) video = Trait(is_video=True)
class Params: # Adds a patched PatchPort with a subnet patched_with_subnet = Trait( patch_ports=RelatedFactoryList('tests.factories.facilities.PatchPortFactory', 'room', size=1, patched=True, switch_port__default_vlans__create_subnet=True) )
class Params: """Optional params which change factory output when True. :param rider_address: Generate an address for the OnDemand Web application. :example usage: AddressFactory.create(rider_address=True) """ rider_address = Trait(url=API.build_api_url('/me/rider/addresses'))
class Params: validated = Trait( filled_out=True, phone_number='+1-800-888-8888', country='US', state_or_territory='US-MA', postal_code='02142', agreed_to_terms_of_service=True, )
class Params: """Optional params which change factory output when True. :param future_recurring_ride: Create a recurring ride for a future date. :param ride_with_note: Create a recurring ride with a driver note. :param same_day_future_recurring_ride: Create a same-day recurring ride for a future time. :example usage: RecurringRideFactory.create(future_recurring_ride=True) """ future_recurring_ride: bool = Trait(ride=RideFactory.build( future_ride=True)) ride_with_note: bool = Trait(ride=LazyAttribute( lambda _: RideFactory.build(ride_with_note=True)), ) same_day_future_recurring_ride: bool = Trait( ride=RideFactory.build(same_day_future_ride=True), )
class Params: """Optional boolean params which change factory output when True. :param exception_service_added: Create a service with an add service exception. :param exception_service_removed: Create a service with a remove service exception. :param expired_service: Create a service that has expired. :param fare_service: Create a service with a fare payment. :param future_ride_service: Create a service with future rides enabled. :param hub_address: Create a service with a hub. :param recurring_ride_service: Create a service with recurring rides enabled. :example usage: ServiceFactory.create(expired_service=True) """ exception_service_added: bool = Trait(exceptions=[ { 'end_time': f'{date.today() + timedelta(days=22)}T18:00:00.000Z', 'message': 'Testing Service Addition.', 'start_time': f'{date.today() + timedelta(days=15)}T09:30:00.000Z', 'type': 'service_added', }, ], ) exception_service_removed: bool = Trait(exceptions=[ { 'end_time': f'{date.today() + timedelta(days=7)}T18:00:00.000Z', 'message': 'Testing Service Removal.', 'start_time': f'{date.today() + timedelta(days=3)}T09:30:00.000Z', 'type': 'service_removed', }, ], ) expired_service: bool = Trait( end_date=date.isoformat(date.today() - timedelta(days=5)), start_date=date.isoformat(date.today() - timedelta(days=10)), ) fare_service: bool = Trait( fare_required=True, fare_price=2.0, ) future_ride_service: bool = Trait(in_advance_enabled=True) hub_address: bool = Trait(addresses=[{ 'hub': True, 'pickup': True, 'dropoff': True, 'address_id': '7539' }], ) recurring_ride_service: bool = Trait( in_advance_enabled=True, recurring_rides_enabled=True, )
class Params: """Optional params which change factory output when True. :param wheelchair_vehicle: Generate a vehicle with wheelchair accessibility. :example usage: VehicleFactory.create(wheelchair_vehicle=True) """ wheelchair_vehicle: bool = Trait(wheelchair_capacity=2, wheelchair_impact=2)
class Params: paid = True free = Trait( tunnel_count=1, bandwidth=100, forwards=2, reserved_subdomains=0, cost=0, name="free", stripe_id=None, ) waiting = Trait( tunnel_count=0, bandwidth=0, forwards=0, reserved_subdomains=0, cost=0, name="waiting", stripe_id=None, ) beta = Trait( tunnel_count=2, bandwidth=1000, forwards=10, reserved_subdomains=1, cost=0, name="beta", stripe_id=None, ) admin = Trait( tunnel_count=20, bandwidth=10000, forwards=100, reserved_subdomains=5, cost=0, name="admin", stripe_id=None, )
class Params: verified = False unverified = Trait( data=LazyAttribute(lambda x: { "is_active": True, "mode": "audit", "user": x.user.username, "course_details": { "course_id": x.course_run.edx_course_key, } }) )
class Params: lop_indicator = Trait( lop_target=1000, target_frequency=IndicatorM.LOP, periodic_target=RelatedFactory( 'factories.indicators_models.PeriodicTargetFactory', 'indicator', target=1000, start_date=SelfAttribute('indicator.program.reporting_period_start'), end_date=SelfAttribute('indicator.program.reporting_period_end'), ) )
class Params: """Optional params which change factory output when True. :param account_rider: Generate a rider using an existing account. :example usage: RiderFactory.create(account_rider=True) """ account_rider: bool = Trait( first_name='QA', last_name='Rider', email='*****@*****.**', username='******', ) superuser_account_rider: bool = Trait( first_name='QA', last_name='User', email='*****@*****.**', username='******', )
class Params: """Optional params which change factory output when True. :param account_ride: Generate a ride using a rider with an existing account. :param future_ride: Generate a ride for a future date. :param same_day_future_ride: Generate a same day ride for a future time. :param hub_ride: Generate a ride which uses a hub address as pick up or drop off. :example usage: RideFactory.create(hub_ride=True) """ account_ride: bool = Trait(rider=RiderFactory.build( account_rider=True)) future_ride: bool = Trait(pickup={ 'address': '4506 Emperor Boulevard Durham, NC, USA', 'position': { 'latitude': 35.8724046, 'longitude': -78.8426551 }, 'priority': 0, 'timestamp': date_objects.build_date_string(days=1), }, ) hub_ride: bool = Trait(pickup={ 'address': 'Stop #300 - TransLoc Office', 'position': { 'latitude': 35.8750625, 'longitude': -78.8406699 }, 'priority': 0, }, ) ride_with_note: bool = Trait(note=fake.sentence(nb_words=3)) same_day_future_ride: bool = Trait(pickup={ 'address': '4506 Emperor Boulevard Durham, NC, USA', 'position': { 'latitude': 35.8724046, 'longitude': -78.8426551 }, 'priority': 0, 'timestamp': date_objects.build_date_string(hours=1), }, ) superuser_account_ride: bool = Trait(rider=RiderFactory.build( superuser_account_rider=True))
class Params: active = Trait(is_active=True) use_custom_name = Trait(custom_name=Faker('user_name')) group = Trait(is_group=True)
class Params: with_traffic_group = Trait(default_traffic_group=SubFactory( 'tests.factories.traffic.TrafficGroupFactory'))
class Params: with_program_run = Trait(program_run=SubFactory(ProgramRunFactory))
def trait(name, attrs, parent=None): if hasattr(local, 'working_factory'): _get_active_factory().trait(name, attrs, parent) else: traits[name] = Trait(attrs, parent)
class Params: no_cta = Trait(cta=None)
class Params: has_expired = Trait( expiry_date=past_datetime_faker )
class Params: no_profile = Trait(profile=None)
class Params: has_company_affiliation = Trait(company=SubFactory(CompanyFactory))
class Params: mozillauser = Trait(published_by=Iterator( EmailUser.objects.filter(email__icontains='mozilla'))) is_featured = Trait(featured=True) is_published_by_creator = Trait(published_by_creator=True)
class Params: staff = Trait(is_staff=True) admin = Trait(is_superuser=True) extended_profile = Trait( profile=SubFactory(ExtendedUserProfileFactory))
class Params: mc_template = Trait( name=Sequence( lambda n: LevelTierM.get_templates()['mc_standard']['tiers'][n] ) )
class Params: patched = Trait( switch_port=SubFactory('tests.factories.host.SwitchPortFactory') )