# Import libraries
import pytest

# Import db
from accounts.models import db

# Import user db model
from accounts.models.user import User

def test_user_model(app, db):
    man_man = db.session.execute(db.select(User).where(User.name == 'Man Man')).scalar_one()
    man_man_visits = man_man.get_visits()
    man_man_dogs = man_man.get_dogs()

    real_person = db.session.execute(db.select(User).where(User.name == 'Real Person')).scalar_one()
    real_person_visits = real_person.get_visits()
    real_person_dogs = real_person.get_dogs()

    # Assert something
    assert man_man.id == 1
    assert man_man.name == 'Man Man'
    assert man_man.username == 'mamanman'
    assert man_man.email == 'man_man@email.com'
    assert man_man.address == '123 Home Ln. City, AA 11223'
    assert man_man.password == 'manword'
    assert len(man_man_visits) == 2
    assert len(man_man_dogs) == 2

    assert real_person.id == 2
    assert real_person.name == 'Real Person'
    assert real_person.username == 'imarealperson'
    assert real_person.email == 'real_person@email.com'
    assert real_person.address == '113 Park St. City, AA 13433'
    assert real_person.password == 'realpassword'
    assert len(real_person_visits) == 1
    assert len(real_person_dogs) == 1