Alexander Hess
fdcc93a1ea
- use SQLAlchemy (and PostgreSQL) to model the ORM layer
- add the following models:
+ Address => modelling all kinds of addresses
+ City => model the three target cities
+ Courier => model the UDP's couriers
+ Customer => model the UDP's customers
+ Order => model the orders received by the UDP
+ Restaurant => model the restaurants active on the UDP
- so far, the emphasis lies on expression the Foreign Key
and Check Constraints that are used to validate the assumptions
inherent to the cleanded data
- provide database-independent unit tests with 100% coverage
- provide additional integration tests ("e2e") that commit data to
a PostgreSQL instance to validate that the constraints work
- adapt linting rules a bit
141 lines
4.3 KiB
Python
141 lines
4.3 KiB
Python
"""Test the ORM's Address model."""
|
|
|
|
import pytest
|
|
from sqlalchemy import exc as sa_exc
|
|
from sqlalchemy.orm import exc as orm_exc
|
|
|
|
from urban_meal_delivery import db
|
|
|
|
|
|
class TestSpecialMethods:
|
|
"""Test special methods in Address."""
|
|
|
|
# pylint:disable=no-self-use
|
|
|
|
def test_create_address(self, address_data):
|
|
"""Test instantiation of a new Address object."""
|
|
result = db.Address(**address_data)
|
|
|
|
assert result is not None
|
|
|
|
def test_text_representation(self, address_data):
|
|
"""Address has a non-literal text representation."""
|
|
address = db.Address(**address_data)
|
|
street = address_data['street']
|
|
city_name = address_data['city_name']
|
|
|
|
result = repr(address)
|
|
|
|
assert result == f'<Address({street} in {city_name})>'
|
|
|
|
|
|
@pytest.mark.e2e
|
|
@pytest.mark.no_cover
|
|
class TestConstraints:
|
|
"""Test the database constraints defined in Address."""
|
|
|
|
# pylint:disable=no-self-use
|
|
|
|
def test_insert_into_database(self, address, db_session):
|
|
"""Insert an instance into the database."""
|
|
db_session.add(address)
|
|
db_session.commit()
|
|
|
|
def test_dublicate_primary_key(self, address, address_data, city, db_session):
|
|
"""Can only add a record once."""
|
|
db_session.add(address)
|
|
db_session.commit()
|
|
|
|
another_address = db.Address(**address_data)
|
|
another_address.city = city
|
|
db_session.add(another_address)
|
|
|
|
with pytest.raises(orm_exc.FlushError):
|
|
db_session.commit()
|
|
|
|
def test_delete_a_referenced_address(self, address, address_data, db_session):
|
|
"""Remove a record that is referenced with a FK."""
|
|
db_session.add(address)
|
|
db_session.commit()
|
|
|
|
# Fake a second address that belongs to the same primary address.
|
|
address_data['id'] += 1
|
|
another_address = db.Address(**address_data)
|
|
db_session.add(another_address)
|
|
db_session.commit()
|
|
|
|
with pytest.raises(sa_exc.IntegrityError):
|
|
db_session.execute(
|
|
db.Address.__table__.delete().where( # noqa:WPS609
|
|
db.Address.id == address.id,
|
|
),
|
|
)
|
|
|
|
def test_delete_a_referenced_city(self, address, city, db_session):
|
|
"""Remove a record that is referenced with a FK."""
|
|
db_session.add(address)
|
|
db_session.commit()
|
|
|
|
with pytest.raises(sa_exc.IntegrityError):
|
|
db_session.execute(
|
|
db.City.__table__.delete().where(db.City.id == city.id), # noqa:WPS609
|
|
)
|
|
|
|
@pytest.mark.parametrize('latitude', [-91, 91])
|
|
def test_invalid_latitude(self, address, db_session, latitude):
|
|
"""Insert an instance with invalid data."""
|
|
address.latitude = latitude
|
|
db_session.add(address)
|
|
|
|
with pytest.raises(sa_exc.IntegrityError):
|
|
db_session.commit()
|
|
|
|
@pytest.mark.parametrize('longitude', [-181, 181])
|
|
def test_invalid_longitude(self, address, db_session, longitude):
|
|
"""Insert an instance with invalid data."""
|
|
address.longitude = longitude
|
|
db_session.add(address)
|
|
|
|
with pytest.raises(sa_exc.IntegrityError):
|
|
db_session.commit()
|
|
|
|
@pytest.mark.parametrize('zip_code', [-1, 0, 9999, 100000])
|
|
def test_invalid_zip_code(self, address, db_session, zip_code):
|
|
"""Insert an instance with invalid data."""
|
|
address.zip_code = zip_code
|
|
db_session.add(address)
|
|
|
|
with pytest.raises(sa_exc.IntegrityError):
|
|
db_session.commit()
|
|
|
|
@pytest.mark.parametrize('floor', [-1, 41])
|
|
def test_invalid_floor(self, address, db_session, floor):
|
|
"""Insert an instance with invalid data."""
|
|
address.floor = floor
|
|
db_session.add(address)
|
|
|
|
with pytest.raises(sa_exc.IntegrityError):
|
|
db_session.commit()
|
|
|
|
|
|
class TestProperties:
|
|
"""Test properties in Address."""
|
|
|
|
# pylint:disable=no-self-use
|
|
|
|
def test_is_primary(self, address_data):
|
|
"""Test Address.is_primary property."""
|
|
address = db.Address(**address_data)
|
|
|
|
result = address.is_primary
|
|
|
|
assert result is True
|
|
|
|
def test_is_not_primary(self, address_data):
|
|
"""Test Address.is_primary property."""
|
|
address_data['_primary_id'] = 999
|
|
address = db.Address(**address_data)
|
|
|
|
result = address.is_primary
|
|
|
|
assert result is False
|