44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
# Import libraries
|
|
import pytest
|
|
|
|
# Import db
|
|
from accounts.models import db
|
|
|
|
# Import db models
|
|
from accounts.models.dog import Dog
|
|
from accounts.models.dog.breed_size import BreedSize
|
|
from accounts.models.user import User
|
|
from accounts.models.visit.visit_type import VisitType
|
|
|
|
# Import flask app factory function
|
|
from tests.testing_app import create_app
|
|
from tests.testing_app import create_app
|
|
|
|
def test_db():
|
|
app = create_app()
|
|
|
|
with app.app_context():
|
|
man_man = db.session.execute(db.select(User).where(User.name == 'Man Man')).scalar_one()
|
|
man_man_bookings = man_man.get_bookings()
|
|
|
|
real_person = db.session.execute(db.select(User).where(User.name == 'Real Person')).scalar_one()
|
|
real_person_bookings = real_person.get_bookings()
|
|
|
|
# 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_bookings) == 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_bookings) == 1
|
|
|