mirror of
https://github.com/facefusion/facefusion.git
synced 2026-04-23 01:46:09 +02:00
Henry Ruhs
46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
import secrets
|
|
from datetime import timedelta
|
|
|
|
from facefusion.session_manager import clear_session, create_session, get_session, set_session, validate_session
|
|
|
|
|
|
def test_get_and_set_session() -> None:
|
|
session = create_session()
|
|
session_id = secrets.token_urlsafe(16)
|
|
|
|
set_session(session_id, session)
|
|
|
|
assert get_session(session_id) == session
|
|
|
|
|
|
def test_validate_session() -> None:
|
|
session = create_session()
|
|
session_id = secrets.token_urlsafe(16)
|
|
|
|
set_session(session_id, session)
|
|
|
|
assert validate_session(session_id) is True
|
|
|
|
set_session(session_id,
|
|
{
|
|
'access_token': session.get('access_token'),
|
|
'refresh_token': session.get('refresh_token'),
|
|
'created_at': session.get('created_at'),
|
|
'expires_at': session.get('expires_at') - timedelta(hours = 1)
|
|
})
|
|
|
|
assert validate_session(session_id) is False
|
|
|
|
|
|
def test_clear_session() -> None:
|
|
session = create_session()
|
|
session_id = secrets.token_urlsafe(16)
|
|
|
|
set_session(session_id, session)
|
|
|
|
assert validate_session(session_id) is True
|
|
|
|
clear_session(session_id)
|
|
|
|
assert validate_session(session_id) is None
|