vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix test case CORS Options

Browse source
This commit is contained in:
ngxson 2024-03-08 22:37:49 +01:00
parent 7445bb8ad9
commit 1a7c5fd50a
3 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/server/server.cpp
View file

@ -2750,6 +2750,8 @@ int main(int argc, char ** argv) {
} }
// API key is invalid or not provided // API key is invalid or not provided
// TODO: make another middleware for CORS related logic
res.set_header("Access-Control-Allow-Origin", req.get_header_value("Origin"));
res.set_content("Unauthorized: Invalid API Key", "text/plain; charset=utf-8"); res.set_content("Unauthorized: Invalid API Key", "text/plain; charset=utf-8");
res.status = 401; // Unauthorized res.status = 401; // Unauthorized

5
examples/server/tests/features/security.feature
View file

@ -39,8 +39,9 @@ Feature: Security
Scenario Outline: CORS Options Scenario Outline: CORS Options
When an OPTIONS request is sent from <origin> Given a user api key llama.cpp
Then CORS header <cors_header> is set to <cors_header_value> When an OPTIONS request is sent from <origin>
Then CORS header <cors_header> is set to <cors_header_value>
Examples: Headers Examples: Headers
| origin | cors_header | cors_header_value | | origin | cors_header | cors_header_value |

3
examples/server/tests/features/steps/steps.py
View file

@ -564,8 +564,9 @@ async def step_detokenize(context):
@async_run_until_complete @async_run_until_complete
async def step_options_request(context, origin): async def step_options_request(context, origin):
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
headers = {'Authorization': f'Bearer {context.user_api_key}', 'Origin': origin}
async with session.options(f'{context.base_url}/v1/chat/completions', async with session.options(f'{context.base_url}/v1/chat/completions',
headers={"Origin": origin}) as response: headers=headers) as response:
assert response.status == 200 assert response.status == 200
context.options_response = response context.options_response = response