24 lines
611 B
Python
24 lines
611 B
Python
"""Factory to create `Courier` instances."""
|
|
|
|
|
|
import factory
|
|
from factory import alchemy
|
|
|
|
from tests.db.fake_data.factories import utils
|
|
from urban_meal_delivery import db
|
|
|
|
|
|
class CourierFactory(alchemy.SQLAlchemyModelFactory):
|
|
"""Create instances of the `db.Courier` model."""
|
|
|
|
class Meta:
|
|
model = db.Courier
|
|
sqlalchemy_get_or_create = ('id',)
|
|
|
|
id = factory.Sequence(lambda num: num) # noqa:WPS125
|
|
created_at = factory.LazyFunction(utils.early_in_the_morning)
|
|
vehicle = 'bicycle'
|
|
historic_speed = 7.89
|
|
capacity = 100
|
|
pay_per_hour = 750
|
|
pay_per_order = 200
|