31 lines
877 B
Python
31 lines
877 B
Python
# Import libraries
|
|
import pytest
|
|
|
|
# Import db
|
|
from accounts.models import db
|
|
|
|
# Import db models
|
|
from accounts.models.dog import Dog
|
|
|
|
def test_user_model(app, db):
|
|
lassey = db.session.execute(db.select(Dog).where(Dog.name == 'Lassey')).scalar_one()
|
|
rufus = db.session.execute(db.select(Dog).where(Dog.name == 'Rufus')).scalar_one()
|
|
air_bud = db.session.execute(db.select(Dog).where(Dog.name == 'Air Bud')).scalar_one()
|
|
|
|
# Assert something
|
|
assert lassey.id == 1
|
|
assert lassey.name == 'Lassey'
|
|
assert lassey.breed_size.value == 'Large'
|
|
assert lassey.owner_id == 1
|
|
|
|
assert rufus.id == 2
|
|
assert rufus.name == 'Rufus'
|
|
assert rufus.breed_size.value == 'Medium'
|
|
assert rufus.owner_id == 1
|
|
|
|
assert air_bud.id == 3
|
|
assert air_bud.name == 'Air Bud'
|
|
assert air_bud.breed_size.value == 'Large'
|
|
assert air_bud.owner_id == 2
|
|
|