Alexander Hess
e4e543bd40
- add new ORM models `ReplaySimulation` and `ReplayedOrder`
to store the data generated by the routing simulations
- add migrations script to create the corresponding database tables
+ create "replay_simulations" and "replayed_orders" tables
+ add missing check constraints to "orders" table
+ add unique constraint to "orders" table to enable compound key
+ drop unnecessary check constraints from the "orders" table
- add tests for the new ORM models
+ add `simulation`, `replayed_order`, `make_replay_order()`, and
`pre_order` fixtures
+ add `ReplayedOrderFactor` faker class
- fix some typos
99 lines
2.3 KiB
Python
99 lines
2.3 KiB
Python
"""Fake data for testing the ORM layer."""
|
|
|
|
import pytest
|
|
|
|
from urban_meal_delivery import db
|
|
|
|
|
|
@pytest.fixture
|
|
def city_data():
|
|
"""The data for the one and only `City` object as a `dict`."""
|
|
return {
|
|
'id': 1,
|
|
'name': 'Paris',
|
|
'kml': "<?xml version='1.0' encoding='UTF-8'?> ...",
|
|
'center_latitude': 48.856614,
|
|
'center_longitude': 2.3522219,
|
|
'northeast_latitude': 48.9021449,
|
|
'northeast_longitude': 2.4699208,
|
|
'southwest_latitude': 48.815573,
|
|
'southwest_longitude': 2.225193,
|
|
'initial_zoom': 12,
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def city(city_data):
|
|
"""The one and only `City` object."""
|
|
return db.City(**city_data)
|
|
|
|
|
|
@pytest.fixture
|
|
def address(make_address):
|
|
"""An `Address` object in the `city`."""
|
|
return make_address()
|
|
|
|
|
|
@pytest.fixture
|
|
def courier(make_courier):
|
|
"""A `Courier` object."""
|
|
return make_courier()
|
|
|
|
|
|
@pytest.fixture
|
|
def customer(make_customer):
|
|
"""A `Customer` object."""
|
|
return make_customer()
|
|
|
|
|
|
@pytest.fixture
|
|
def restaurant(address, make_restaurant):
|
|
"""A `Restaurant` object located at the `address`."""
|
|
return make_restaurant(address=address)
|
|
|
|
|
|
@pytest.fixture
|
|
def order(make_order, restaurant):
|
|
"""An ad-hoc `Order` object for the `restaurant`."""
|
|
return make_order(restaurant=restaurant)
|
|
|
|
|
|
@pytest.fixture
|
|
def pre_order(make_order, restaurant):
|
|
"""A scheduled `Order` object for the `restaurant`."""
|
|
return make_order(restaurant=restaurant, scheduled=True)
|
|
|
|
|
|
@pytest.fixture
|
|
def replayed_order(make_replay_order, order):
|
|
"""A `ReplayedOrder` object for the `restaurant`."""
|
|
return make_replay_order(order=order)
|
|
|
|
|
|
@pytest.fixture
|
|
def grid(city):
|
|
"""A `Grid` with a pixel area of 1 square kilometer."""
|
|
return db.Grid(city=city, side_length=1000)
|
|
|
|
|
|
@pytest.fixture
|
|
def pixel(grid):
|
|
"""The `Pixel` in the lower-left corner of the `grid`."""
|
|
return db.Pixel(id=1, grid=grid, n_x=0, n_y=0)
|
|
|
|
|
|
@pytest.fixture
|
|
def simulation_data(city):
|
|
"""The data for the one and only `ReplaySimulation` object as a `dict`."""
|
|
return {
|
|
'id': 1,
|
|
'city': city,
|
|
'strategy': 'best_possible_routing',
|
|
'run': 0,
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def simulation(simulation_data):
|
|
"""The one and only `ReplaySimulation` object."""
|
|
return db.ReplaySimulation(**simulation_data)
|