mirror of
https://github.com/simple-login/app.git
synced 2024-11-15 13:14:36 +08:00
42 lines
1.1 KiB
Python
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 not 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
|