81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
|
"""Test the ORM's Restaurant 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 Restaurant."""
|
||
|
|
|
||
|
|
# pylint:disable=no-self-use
|
||
|
|
|
||
|
|
def test_create_restaurant(self, restaurant_data):
|
||
|
|
"""Test instantiation of a new Restaurant object."""
|
||
|
|
result = db.Restaurant(**restaurant_data)
|
||
|
|
|
||
|
|
assert result is not None
|
||
|
|
|
||
|
|
def test_text_representation(self, restaurant_data):
|
||
|
|
"""Restaurant has a non-literal text representation."""
|
||
|
|
restaurant = db.Restaurant(**restaurant_data)
|
||
|
|
name = restaurant_data['name']
|
||
|
|
|
||
|
|
result = repr(restaurant)
|
||
|
|
|
||
|
|
assert result == f'<Restaurant({name})>'
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.e2e
|
||
|
|
@pytest.mark.no_cover
|
||
|
|
class TestConstraints:
|
||
|
|
"""Test the database constraints defined in Restaurant."""
|
||
|
|
|
||
|
|
# pylint:disable=no-self-use
|
||
|
|
|
||
|
|
def test_insert_into_database(self, restaurant, db_session):
|
||
|
|
"""Insert an instance into the database."""
|
||
|
|
db_session.add(restaurant)
|
||
|
|
db_session.commit()
|
||
|
|
|
||
|
|
def test_dublicate_primary_key(self, restaurant, restaurant_data, db_session):
|
||
|
|
"""Can only add a record once."""
|
||
|
|
db_session.add(restaurant)
|
||
|
|
db_session.commit()
|
||
|
|
|
||
|
|
another_restaurant = db.Restaurant(**restaurant_data)
|
||
|
|
db_session.add(another_restaurant)
|
||
|
|
|
||
|
|
with pytest.raises(orm_exc.FlushError):
|
||
|
|
db_session.commit()
|
||
|
|
|
||
|
|
def test_delete_a_referenced_address(self, restaurant, address, db_session):
|
||
|
|
"""Remove a record that is referenced with a FK."""
|
||
|
|
db_session.add(restaurant)
|
||
|
|
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_negative_prep_duration(self, restaurant, db_session):
|
||
|
|
"""Insert an instance with invalid data."""
|
||
|
|
restaurant.estimated_prep_duration = -1
|
||
|
|
db_session.add(restaurant)
|
||
|
|
|
||
|
|
with pytest.raises(sa_exc.IntegrityError):
|
||
|
|
db_session.commit()
|
||
|
|
|
||
|
|
def test_too_high_prep_duration(self, restaurant, db_session):
|
||
|
|
"""Insert an instance with invalid data."""
|
||
|
|
restaurant.estimated_prep_duration = 2500
|
||
|
|
db_session.add(restaurant)
|
||
|
|
|
||
|
|
with pytest.raises(sa_exc.IntegrityError):
|
||
|
|
db_session.commit()
|