mirror of
https://github.com/simple-login/app.git
synced 2024-11-10 17:35:27 +08:00
372466ab06
- add __tablename__ for all models - use sa and orm instead of db - rollback all changes in tests - remove session in @app.teardown_appcontext
28 lines
832 B
Python
28 lines
832 B
Python
from app.db import Session
|
|
from app.jose_utils import make_id_token, verify_id_token
|
|
from app.models import ClientUser, User, Client
|
|
|
|
|
|
def test_encode_decode(flask_client):
|
|
user = User.create(
|
|
email="a@b.c", password="password", name="Test User", activated=True
|
|
)
|
|
Session.commit()
|
|
|
|
client1 = Client.create_new(name="Demo", user_id=user.id)
|
|
client1.oauth_client_id = "client-id"
|
|
client1.oauth_client_secret = "client-secret"
|
|
Session.commit()
|
|
|
|
client_user = ClientUser.create(client_id=client1.id, user_id=user.id)
|
|
Session.commit()
|
|
|
|
jwt_token = make_id_token(client_user)
|
|
|
|
assert type(jwt_token) is str
|
|
assert verify_id_token(jwt_token)
|
|
|
|
|
|
def test_db_tear_down(flask_client):
|
|
"""make sure the db is reset after each test"""
|
|
assert len(ClientUser.filter_by().all()) == 0
|