38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
# 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
|
|
|