mirror of
https://github.com/facefusion/facefusion.git
synced 2026-06-01 02:11:36 +02:00
Add session_id, make token size more reasonable (#983)
* Add session_id, make token size more reasonable * Use more direct approach
This commit is contained in:
@@ -28,7 +28,8 @@ def test_create_session(test_client : TestClient) -> None:
|
||||
})
|
||||
create_session_body = create_session_response.json()
|
||||
|
||||
assert session_manager.get_session(create_session_body.get('access_token'))
|
||||
assert create_session_body.get('access_token')
|
||||
assert create_session_body.get('refresh_token')
|
||||
assert create_session_response.status_code == 201
|
||||
|
||||
create_session_response = test_client.post('/session', json =
|
||||
@@ -78,9 +79,9 @@ def test_get_session(test_client : TestClient) -> None:
|
||||
|
||||
assert get_session_response.status_code == 200
|
||||
|
||||
access_token = create_session_body.get('access_token')
|
||||
session : Session = session_manager.get_session(access_token)
|
||||
session_manager.set_session(access_token,
|
||||
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'),
|
||||
@@ -90,7 +91,7 @@ def test_get_session(test_client : TestClient) -> None:
|
||||
|
||||
get_session_response = test_client.get('/session', headers =
|
||||
{
|
||||
'Authorization': 'Bearer ' + access_token
|
||||
'Authorization': 'Bearer ' + create_session_body.get('access_token')
|
||||
})
|
||||
|
||||
assert get_session_response.status_code == 426
|
||||
@@ -116,9 +117,9 @@ def test_refresh_session(test_client : TestClient) -> None:
|
||||
})
|
||||
refresh_session_body = refresh_session_response.json()
|
||||
|
||||
assert session_manager.get_session(create_session_body.get('access_token')) is None
|
||||
|
||||
assert session_manager.get_session(refresh_session_body.get('access_token'))
|
||||
assert refresh_session_body.get('access_token')
|
||||
assert refresh_session_body.get('refresh_token')
|
||||
assert not refresh_session_body.get('access_token') == create_session_body.get('access_token')
|
||||
|
||||
assert refresh_session_response.status_code == 200
|
||||
|
||||
@@ -149,6 +150,6 @@ def test_destroy_session(test_client : TestClient) -> None:
|
||||
'Authorization': 'Bearer ' + create_session_body.get('access_token')
|
||||
})
|
||||
|
||||
assert session_manager.get_session(create_session_body.get('access_token')) is None
|
||||
assert session_manager.find_session_id(create_session_body.get('access_token')) is None
|
||||
|
||||
assert delete_session_response.status_code == 200
|
||||
|
||||
Reference in New Issue
Block a user