app-MAIL-temp/tests/auth/test_login.py

42 lines
1.1 KiB
Python

from flask import url_for
from app.extensions import db
from app.models import User
def test_unactivated_user_login(flask_client):
"""Start with a blank database."""
# create user, user is not activated
User.create(email="a@b.c", password="password", name="Test User")
db.session.commit()
r = flask_client.post(
url_for("auth.login"),
data={"email": "a@b.c", "password": "password"},
follow_redirects=True,
)
assert r.status_code == 200
assert (
b"Please check your inbox for the activation email. You can also have this email re-sent"
in r.data
)
def test_activated_user_login(flask_client):
"""Start with a blank database."""
# create user, user is activated
User.create(email="a@b.c", password="password", name="Test User", activated=True)
db.session.commit()
r = flask_client.post(
url_for("auth.login"),
data={"email": "a@b.c", "password": "password"},
follow_redirects=True,
)
assert r.status_code == 200
assert b"/auth/logout" in r.data