mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-08-16 11:05:53 +08:00
Test: Added test cases for Update Session With Chat Assistant HTTP API (#6968)
### What problem does this PR solve? cover [Update chat assistant's sessions](https://ragflow.io/docs/dev/http_api_reference#update-chat-assistants-session) endpoints ### Type of change - [x] Update test cases
This commit is contained in:
parent
ffb9f01bea
commit
9b789c2ae9
@ -33,3 +33,21 @@ def add_sessions_with_chat_assistant(request, get_http_api_auth, add_chat_assist
|
|||||||
session_ids.append(res["data"]["id"])
|
session_ids.append(res["data"]["id"])
|
||||||
|
|
||||||
return chat_assistant_ids[0], session_ids
|
return chat_assistant_ids[0], session_ids
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(scope="function")
|
||||||
|
def add_sessions_with_chat_assistant_func(request, get_http_api_auth, add_chat_assistants):
|
||||||
|
_, _, chat_assistant_ids = add_chat_assistants
|
||||||
|
|
||||||
|
def cleanup():
|
||||||
|
for chat_assistant_id in chat_assistant_ids:
|
||||||
|
delete_session_with_chat_assistants(get_http_api_auth, chat_assistant_id)
|
||||||
|
|
||||||
|
request.addfinalizer(cleanup)
|
||||||
|
|
||||||
|
session_ids = []
|
||||||
|
for i in range(5):
|
||||||
|
res = create_session_with_chat_assistant(get_http_api_auth, chat_assistant_ids[0], {"name": f"session_with_chat_assistant_{i}"})
|
||||||
|
session_ids.append(res["data"]["id"])
|
||||||
|
|
||||||
|
return chat_assistant_ids[0], session_ids
|
||||||
|
@ -0,0 +1,142 @@
|
|||||||
|
#
|
||||||
|
# Copyright 2025 The InfiniFlow Authors. All Rights Reserved.
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
#
|
||||||
|
from concurrent.futures import ThreadPoolExecutor
|
||||||
|
from random import randint
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
from common import INVALID_API_TOKEN, SESSION_WITH_CHAT_NAME_LIMIT, delete_chat_assistants, list_session_with_chat_assistants, update_session_with_chat_assistant
|
||||||
|
from libs.auth import RAGFlowHttpApiAuth
|
||||||
|
|
||||||
|
|
||||||
|
class TestAuthorization:
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"auth, expected_code, expected_message",
|
||||||
|
[
|
||||||
|
(None, 0, "`Authorization` can't be empty"),
|
||||||
|
(
|
||||||
|
RAGFlowHttpApiAuth(INVALID_API_TOKEN),
|
||||||
|
109,
|
||||||
|
"Authentication error: API key is invalid!",
|
||||||
|
),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
def test_invalid_auth(self, auth, expected_code, expected_message):
|
||||||
|
res = update_session_with_chat_assistant(auth, "chat_assistant_id", "session_id")
|
||||||
|
assert res["code"] == expected_code
|
||||||
|
assert res["message"] == expected_message
|
||||||
|
|
||||||
|
|
||||||
|
class TestSessionWithChatAssistantUpdate:
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"payload, expected_code, expected_message",
|
||||||
|
[
|
||||||
|
({"name": "valid_name"}, 0, ""),
|
||||||
|
pytest.param({"name": "a" * (SESSION_WITH_CHAT_NAME_LIMIT + 1)}, 102, "", marks=pytest.mark.skip(reason="issues/")),
|
||||||
|
pytest.param({"name": 1}, 100, "", marks=pytest.mark.skip(reason="issues/")),
|
||||||
|
({"name": ""}, 102, "`name` can not be empty."),
|
||||||
|
({"name": "duplicated_name"}, 0, ""),
|
||||||
|
({"name": "case insensitive"}, 0, ""),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
def test_name(self, get_http_api_auth, add_sessions_with_chat_assistant_func, payload, expected_code, expected_message):
|
||||||
|
chat_assistant_id, session_ids = add_sessions_with_chat_assistant_func
|
||||||
|
if payload["name"] == "duplicated_name":
|
||||||
|
update_session_with_chat_assistant(get_http_api_auth, chat_assistant_id, session_ids[0], payload)
|
||||||
|
elif payload["name"] == "case insensitive":
|
||||||
|
update_session_with_chat_assistant(get_http_api_auth, chat_assistant_id, session_ids[0], {"name": payload["name"].upper()})
|
||||||
|
|
||||||
|
res = update_session_with_chat_assistant(get_http_api_auth, chat_assistant_id, session_ids[0], payload)
|
||||||
|
assert res["code"] == expected_code, res
|
||||||
|
if expected_code == 0:
|
||||||
|
res = list_session_with_chat_assistants(get_http_api_auth, chat_assistant_id, {"id": session_ids[0]})
|
||||||
|
assert res["data"][0]["name"] == payload["name"]
|
||||||
|
else:
|
||||||
|
assert res["message"] == expected_message
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"chat_assistant_id, expected_code, expected_message",
|
||||||
|
[
|
||||||
|
("", 100, "<NotFound '404: Not Found'>"),
|
||||||
|
pytest.param("invalid_chat_assistant_id", 102, "Session does not exist", marks=pytest.mark.skip(reason="issues/")),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
def test_invalid_chat_assistant_id(self, get_http_api_auth, add_sessions_with_chat_assistant_func, chat_assistant_id, expected_code, expected_message):
|
||||||
|
_, session_ids = add_sessions_with_chat_assistant_func
|
||||||
|
res = update_session_with_chat_assistant(get_http_api_auth, chat_assistant_id, session_ids[0], {"name": "valid_name"})
|
||||||
|
assert res["code"] == expected_code
|
||||||
|
assert res["message"] == expected_message
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"session_id, expected_code, expected_message",
|
||||||
|
[
|
||||||
|
("", 100, "<MethodNotAllowed '405: Method Not Allowed'>"),
|
||||||
|
("invalid_session_id", 102, "Session does not exist"),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
def test_invalid_session_id(self, get_http_api_auth, add_sessions_with_chat_assistant_func, session_id, expected_code, expected_message):
|
||||||
|
chat_assistant_id, _ = add_sessions_with_chat_assistant_func
|
||||||
|
res = update_session_with_chat_assistant(get_http_api_auth, chat_assistant_id, session_id, {"name": "valid_name"})
|
||||||
|
assert res["code"] == expected_code
|
||||||
|
assert res["message"] == expected_message
|
||||||
|
|
||||||
|
def test_repeated_update_session(self, get_http_api_auth, add_sessions_with_chat_assistant_func):
|
||||||
|
chat_assistant_id, session_ids = add_sessions_with_chat_assistant_func
|
||||||
|
res = update_session_with_chat_assistant(get_http_api_auth, chat_assistant_id, session_ids[0], {"name": "valid_name_1"})
|
||||||
|
assert res["code"] == 0
|
||||||
|
|
||||||
|
res = update_session_with_chat_assistant(get_http_api_auth, chat_assistant_id, session_ids[0], {"name": "valid_name_2"})
|
||||||
|
assert res["code"] == 0
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"payload, expected_code, expected_message",
|
||||||
|
[
|
||||||
|
pytest.param({"unknown_key": "unknown_value"}, 100, "ValueError", marks=pytest.mark.skip),
|
||||||
|
({}, 0, ""),
|
||||||
|
pytest.param(None, 100, "TypeError", marks=pytest.mark.skip),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
def test_invalid_params(self, get_http_api_auth, add_sessions_with_chat_assistant_func, payload, expected_code, expected_message):
|
||||||
|
chat_assistant_id, session_ids = add_sessions_with_chat_assistant_func
|
||||||
|
res = update_session_with_chat_assistant(get_http_api_auth, chat_assistant_id, session_ids[0], payload)
|
||||||
|
assert res["code"] == expected_code
|
||||||
|
if expected_code != 0:
|
||||||
|
assert expected_message in res["message"]
|
||||||
|
|
||||||
|
@pytest.mark.slow
|
||||||
|
def test_concurrent_update_session(self, get_http_api_auth, add_sessions_with_chat_assistant_func):
|
||||||
|
chunk_num = 50
|
||||||
|
chat_assistant_id, session_ids = add_sessions_with_chat_assistant_func
|
||||||
|
|
||||||
|
with ThreadPoolExecutor(max_workers=5) as executor:
|
||||||
|
futures = [
|
||||||
|
executor.submit(
|
||||||
|
update_session_with_chat_assistant,
|
||||||
|
get_http_api_auth,
|
||||||
|
chat_assistant_id,
|
||||||
|
session_ids[randint(0, 4)],
|
||||||
|
{"name": f"update session test {i}"},
|
||||||
|
)
|
||||||
|
for i in range(chunk_num)
|
||||||
|
]
|
||||||
|
responses = [f.result() for f in futures]
|
||||||
|
assert all(r["code"] == 0 for r in responses)
|
||||||
|
|
||||||
|
def test_update_session_to_deleted_chat_assistant(self, get_http_api_auth, add_sessions_with_chat_assistant_func):
|
||||||
|
chat_assistant_id, session_ids = add_sessions_with_chat_assistant_func
|
||||||
|
delete_chat_assistants(get_http_api_auth, {"ids": [chat_assistant_id]})
|
||||||
|
res = update_session_with_chat_assistant(get_http_api_auth, chat_assistant_id, session_ids[0], {"name": "valid_name"})
|
||||||
|
assert res["code"] == 102
|
||||||
|
assert res["message"] == "You do not own the session"
|
Loading…
x
Reference in New Issue
Block a user