mirror of
https://github.com/facefusion/facefusion.git
synced 2026-04-22 09:26:02 +02:00
fix refresh_session does not validate expiry before refreshing (#1071)
This commit is contained in:
@@ -48,7 +48,7 @@ async def refresh_session(request : Request) -> JSONResponse:
|
||||
body = await request.json()
|
||||
|
||||
for session_id, session in session_manager.SESSIONS.items():
|
||||
if session.get('refresh_token') == body.get('refresh_token'):
|
||||
if session.get('refresh_token') == body.get('refresh_token') and session_manager.validate_session(session_id):
|
||||
__session__ = session_manager.create_session()
|
||||
session_manager.set_session(session_id, __session__)
|
||||
|
||||
|
||||
@@ -130,6 +130,29 @@ def test_refresh_session(test_client : TestClient) -> None:
|
||||
|
||||
assert refresh_session_response.status_code == 401
|
||||
|
||||
create_session_response = test_client.post('/session', json =
|
||||
{
|
||||
'client_version': metadata.get('version')
|
||||
})
|
||||
create_session_body = create_session_response.json()
|
||||
|
||||
session_id = session_manager.find_session_id(create_session_body.get('access_token'))
|
||||
session : Session = session_manager.get_session(session_id)
|
||||
session_manager.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)
|
||||
})
|
||||
|
||||
refresh_session_response = test_client.put('/session', json =
|
||||
{
|
||||
'refresh_token': create_session_body.get('refresh_token')
|
||||
})
|
||||
|
||||
assert refresh_session_response.status_code == 401
|
||||
|
||||
|
||||
def test_destroy_session(test_client : TestClient) -> None:
|
||||
create_session_response = test_client.post('/session', json =
|
||||
|
||||
Reference in New Issue
Block a user