From d503908a652fdbd1453f59d5a106b4da8c46055b Mon Sep 17 00:00:00 2001 From: rubenwardy Date: Sun, 19 Jan 2020 15:46:29 +0000 Subject: Add populated homepage test --- app/tests/test_homepage.py | 21 ++++++++++++++++++--- 1 file changed, 18 insertions(+), 3 deletions(-) (limited to 'app/tests') diff --git a/app/tests/test_homepage.py b/app/tests/test_homepage.py index f0f6793..2bfdbca 100644 --- a/app/tests/test_homepage.py +++ b/app/tests/test_homepage.py @@ -1,11 +1,26 @@ import pytest from app import app +from app.default_data import populate_test_data +from app.models import db, License, Tag, User, UserRank from utils import client, recreate_db -def test_homepage_ok(client): +def test_homepage_empty(client): """Start with a blank database.""" - assert app.config["TESTING"] + rv = client.get("/") + assert b"No packages available" in rv.data and b"packagetile" not in rv.data + + +def test_homepage_with_contents(client): + """Start with a test database.""" + + licenses = { x.name : x for x in License.query.all() } + tags = { x.name : x for x in Tag.query.all() } + admin_user = User.query.filter_by(rank=UserRank.ADMIN).first() + + populate_test_data(db.session, licenses, tags, admin_user) + db.session.commit() rv = client.get("/") - assert b"No packages available" in rv.data + + assert b"No packages available" not in rv.data and b"packagetile" in rv.data -- cgit v1.2.3