100 lines
2.8 KiB
Python
100 lines
2.8 KiB
Python
|
"""Test the ORM's City model."""
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from sqlalchemy.orm import exc as orm_exc
|
||
|
|
|
||
|
|
from urban_meal_delivery import db
|
||
|
|
|
||
|
|
|
||
|
|
class TestSpecialMethods:
|
||
|
|
"""Test special methods in City."""
|
||
|
|
|
||
|
|
# pylint:disable=no-self-use
|
||
|
|
|
||
|
|
def test_create_city(self, city_data):
|
||
|
|
"""Test instantiation of a new City object."""
|
||
|
|
result = db.City(**city_data)
|
||
|
|
|
||
|
|
assert result is not None
|
||
|
|
|
||
|
|
def test_text_representation(self, city_data):
|
||
|
|
"""City has a non-literal text representation."""
|
||
|
|
city = db.City(**city_data)
|
||
|
|
name = city_data['name']
|
||
|
|
|
||
|
|
result = repr(city)
|
||
|
|
|
||
|
|
assert result == f'<City({name})>'
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.e2e
|
||
|
|
@pytest.mark.no_cover
|
||
|
|
class TestConstraints:
|
||
|
|
"""Test the database constraints defined in City."""
|
||
|
|
|
||
|
|
# pylint:disable=no-self-use
|
||
|
|
|
||
|
|
def test_insert_into_database(self, city, db_session):
|
||
|
|
"""Insert an instance into the database."""
|
||
|
|
db_session.add(city)
|
||
|
|
db_session.commit()
|
||
|
|
|
||
|
|
def test_dublicate_primary_key(self, city, city_data, db_session):
|
||
|
|
"""Can only add a record once."""
|
||
|
|
db_session.add(city)
|
||
|
|
db_session.commit()
|
||
|
|
|
||
|
|
another_city = db.City(**city_data)
|
||
|
|
db_session.add(another_city)
|
||
|
|
|
||
|
|
with pytest.raises(orm_exc.FlushError):
|
||
|
|
db_session.commit()
|
||
|
|
|
||
|
|
|
||
|
|
class TestProperties:
|
||
|
|
"""Test properties in City."""
|
||
|
|
|
||
|
|
# pylint:disable=no-self-use
|
||
|
|
|
||
|
|
def test_location_data(self, city_data):
|
||
|
|
"""Test City.location property."""
|
||
|
|
city = db.City(**city_data)
|
||
|
|
|
||
|
|
result = city.location
|
||
|
|
|
||
|
|
assert isinstance(result, dict)
|
||
|
|
assert len(result) == 2
|
||
|
|
assert result['latitude'] == pytest.approx(city_data['_center_latitude'])
|
||
|
|
assert result['longitude'] == pytest.approx(city_data['_center_longitude'])
|
||
|
|
|
||
|
|
def test_viewport_data_overall(self, city_data):
|
||
|
|
"""Test City.viewport property."""
|
||
|
|
city = db.City(**city_data)
|
||
|
|
|
||
|
|
result = city.viewport
|
||
|
|
|
||
|
|
assert isinstance(result, dict)
|
||
|
|
assert len(result) == 2
|
||
|
|
|
||
|
|
def test_viewport_data_northeast(self, city_data):
|
||
|
|
"""Test City.viewport property."""
|
||
|
|
city = db.City(**city_data)
|
||
|
|
|
||
|
|
result = city.viewport['northeast']
|
||
|
|
|
||
|
|
assert isinstance(result, dict)
|
||
|
|
assert len(result) == 2
|
||
|
|
assert result['latitude'] == pytest.approx(city_data['_northeast_latitude'])
|
||
|
|
assert result['longitude'] == pytest.approx(city_data['_northeast_longitude'])
|
||
|
|
|
||
|
|
def test_viewport_data_southwest(self, city_data):
|
||
|
|
"""Test City.viewport property."""
|
||
|
|
city = db.City(**city_data)
|
||
|
|
|
||
|
|
result = city.viewport['southwest']
|
||
|
|
|
||
|
|
assert isinstance(result, dict)
|
||
|
|
assert len(result) == 2
|
||
|
|
assert result['latitude'] == pytest.approx(city_data['_southwest_latitude'])
|
||
|
|
assert result['longitude'] == pytest.approx(city_data['_southwest_longitude'])
|