Started using Bootstrap for styling, moved app factory, added tests, started work on login and registration templates, and more
This commit is contained in:
30
tests/test_dog_model.py
Normal file
30
tests/test_dog_model.py
Normal file
@@ -0,0 +1,30 @@
|
||||
# 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
|
||||
|
||||
@@ -4,18 +4,17 @@ 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
|
||||
# Import user db model
|
||||
from accounts.models.user import User
|
||||
from accounts.models.visit.visit_type import VisitType
|
||||
|
||||
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
|
||||
@@ -25,6 +24,7 @@ def test_user_model(app, db):
|
||||
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'
|
||||
@@ -33,4 +33,5 @@ def test_user_model(app, db):
|
||||
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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user