mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-05 02:41:56 +00:00
fix(ci): update websocket fixtures and vulnerable development parser
This commit is contained in:
parent
ef0a67c754
commit
09f52d9aa4
2 changed files with 6 additions and 3 deletions
|
|
@ -849,6 +849,7 @@ async def test_user_api_key_auth_websocket():
|
|||
# Prepare a mock WebSocket object
|
||||
mock_websocket = MagicMock(spec=WebSocket)
|
||||
mock_websocket.query_params = {"model": "some_model"}
|
||||
mock_websocket.path_params = {}
|
||||
mock_websocket.headers = {"authorization": "Bearer some_api_key"}
|
||||
# Mock the scope attribute that user_api_key_auth_websocket accesses
|
||||
mock_websocket.scope = {"headers": [(b"authorization", b"Bearer some_api_key")]}
|
||||
|
|
@ -872,6 +873,7 @@ async def test_user_api_key_auth_websocket():
|
|||
assert request_arg.headers["authorization"] == "Bearer some_api_key"
|
||||
|
||||
assert mock_user_api_key_auth.call_args.kwargs["api_key"] == "Bearer some_api_key"
|
||||
assert await request_arg.json() == {"model": "some_model"}
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
|
|
@ -885,6 +887,7 @@ async def test_user_api_key_auth_websocket_carries_asgi_path():
|
|||
|
||||
mock_websocket = MagicMock(spec=WebSocket)
|
||||
mock_websocket.query_params = {"model": "some_model"}
|
||||
mock_websocket.path_params = {}
|
||||
mock_websocket.headers = {"authorization": "Bearer some_api_key"}
|
||||
mock_websocket.scope = {
|
||||
"type": "websocket",
|
||||
|
|
|
|||
6
ui/litellm-dashboard/package-lock.json
generated
6
ui/litellm-dashboard/package-lock.json
generated
|
|
@ -11402,9 +11402,9 @@
|
|||
}
|
||||
},
|
||||
"node_modules/smol-toml": {
|
||||
"version": "1.6.1",
|
||||
"resolved": "https://registry.npmjs.org/smol-toml/-/smol-toml-1.6.1.tgz",
|
||||
"integrity": "sha512-dWUG8F5sIIARXih1DTaQAX4SsiTXhInKf1buxdY9DIg4ZYPZK5nGM1VRIYmEbDbsHt7USo99xSLFu5Q1IqTmsg==",
|
||||
"version": "1.8.0",
|
||||
"resolved": "https://registry.npmjs.org/smol-toml/-/smol-toml-1.8.0.tgz",
|
||||
"integrity": "sha512-kCZr2V3ch9i00x8zXRhjUNVcjG9ijES5dDudkXvUVCT5QlJNQWElSJdZqyPemffHoLNUYwOcou0Fy+ojN0uHSQ==",
|
||||
"dev": true,
|
||||
"license": "BSD-3-Clause",
|
||||
"engines": {
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue