52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
|
"""Test the ORM's Customer model."""
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from sqlalchemy.orm import exc as orm_exc
|
||
|
|
|
||
|
|
from urban_meal_delivery import db
|
||
|
|
|
||
|
|
|
||
|
|
class TestSpecialMethods:
|
||
|
|
"""Test special methods in Customer."""
|
||
|
|
|
||
|
|
# pylint:disable=no-self-use
|
||
|
|
|
||
|
|
def test_create_customer(self, customer_data):
|
||
|
|
"""Test instantiation of a new Customer object."""
|
||
|
|
result = db.Customer(**customer_data)
|
||
|
|
|
||
|
|
assert result is not None
|
||
|
|
|
||
|
|
def test_text_representation(self, customer_data):
|
||
|
|
"""Customer has a non-literal text representation."""
|
||
|
|
customer = db.Customer(**customer_data)
|
||
|
|
id_ = customer_data['id']
|
||
|
|
|
||
|
|
result = repr(customer)
|
||
|
|
|
||
|
|
assert result == f'<Customer(#{id_})>'
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.e2e
|
||
|
|
@pytest.mark.no_cover
|
||
|
|
class TestConstraints:
|
||
|
|
"""Test the database constraints defined in Customer."""
|
||
|
|
|
||
|
|
# pylint:disable=no-self-use
|
||
|
|
|
||
|
|
def test_insert_into_database(self, customer, db_session):
|
||
|
|
"""Insert an instance into the database."""
|
||
|
|
db_session.add(customer)
|
||
|
|
db_session.commit()
|
||
|
|
|
||
|
|
def test_dublicate_primary_key(self, customer, customer_data, db_session):
|
||
|
|
"""Can only add a record once."""
|
||
|
|
db_session.add(customer)
|
||
|
|
db_session.commit()
|
||
|
|
|
||
|
|
another_customer = db.Customer(**customer_data)
|
||
|
|
db_session.add(another_customer)
|
||
|
|
|
||
|
|
with pytest.raises(orm_exc.FlushError):
|
||
|
|
db_session.commit()
|