# 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