[OND211-2329]: Updated conftest files, fixed typings.

This commit is contained in:
Hetavi Shah 2025-11-18 14:05:38 +05:30
parent f2eeb2dbbf
commit 26a17e6250
2 changed files with 29 additions and 29 deletions

View file

@ -82,37 +82,37 @@ def ragflow_tmp_dir(request, tmp_path_factory):
@pytest.fixture(scope="session") @pytest.fixture(scope="session")
def HttpApiAuth(token): def http_api_auth(token):
return RAGFlowHttpApiAuth(token) return RAGFlowHttpApiAuth(token)
@pytest.fixture(scope="session") @pytest.fixture(scope="session")
def WebApiAuth(auth): def web_api_auth(auth):
"""Fixture for endpoints that use @login_required (JWT token, not API token).""" """Fixture for endpoints that use @login_required (JWT token, not API token)."""
return RAGFlowWebApiAuth(auth) return RAGFlowWebApiAuth(auth)
@pytest.fixture(scope="function") @pytest.fixture(scope="function")
def clear_datasets(request, HttpApiAuth): def clear_datasets(request, http_api_auth):
def cleanup(): def cleanup():
delete_datasets(HttpApiAuth, {"ids": None}) delete_datasets(http_api_auth, {"ids": None})
request.addfinalizer(cleanup) request.addfinalizer(cleanup)
@pytest.fixture(scope="function") @pytest.fixture(scope="function")
def clear_chat_assistants(request, HttpApiAuth): def clear_chat_assistants(request, http_api_auth):
def cleanup(): def cleanup():
delete_chat_assistants(HttpApiAuth) delete_chat_assistants(http_api_auth)
request.addfinalizer(cleanup) request.addfinalizer(cleanup)
@pytest.fixture(scope="function") @pytest.fixture(scope="function")
def clear_session_with_chat_assistants(request, HttpApiAuth, add_chat_assistants): def clear_session_with_chat_assistants(request, http_api_auth, add_chat_assistants):
def cleanup(): def cleanup():
for chat_assistant_id in chat_assistant_ids: for chat_assistant_id in chat_assistant_ids:
delete_session_with_chat_assistants(HttpApiAuth, chat_assistant_id) delete_session_with_chat_assistants(http_api_auth, chat_assistant_id)
request.addfinalizer(cleanup) request.addfinalizer(cleanup)
@ -120,51 +120,51 @@ def clear_session_with_chat_assistants(request, HttpApiAuth, add_chat_assistants
@pytest.fixture(scope="class") @pytest.fixture(scope="class")
def add_dataset(request, HttpApiAuth): def add_dataset(request, http_api_auth):
def cleanup(): def cleanup():
delete_datasets(HttpApiAuth, {"ids": None}) delete_datasets(http_api_auth, {"ids": None})
request.addfinalizer(cleanup) request.addfinalizer(cleanup)
dataset_ids = batch_create_datasets(HttpApiAuth, 1) dataset_ids = batch_create_datasets(http_api_auth, 1)
return dataset_ids[0] return dataset_ids[0]
@pytest.fixture(scope="function") @pytest.fixture(scope="function")
def add_dataset_func(request, HttpApiAuth): def add_dataset_func(request, http_api_auth):
def cleanup(): def cleanup():
delete_datasets(HttpApiAuth, {"ids": None}) delete_datasets(http_api_auth, {"ids": None})
request.addfinalizer(cleanup) request.addfinalizer(cleanup)
return batch_create_datasets(HttpApiAuth, 1)[0] return batch_create_datasets(http_api_auth, 1)[0]
@pytest.fixture(scope="class") @pytest.fixture(scope="class")
def add_document(HttpApiAuth, add_dataset, ragflow_tmp_dir): def add_document(http_api_auth, add_dataset, ragflow_tmp_dir):
dataset_id = add_dataset dataset_id = add_dataset
document_ids = bulk_upload_documents(HttpApiAuth, dataset_id, 1, ragflow_tmp_dir) document_ids = bulk_upload_documents(http_api_auth, dataset_id, 1, ragflow_tmp_dir)
return dataset_id, document_ids[0] return dataset_id, document_ids[0]
@pytest.fixture(scope="class") @pytest.fixture(scope="class")
def add_chunks(HttpApiAuth, add_document): def add_chunks(http_api_auth, add_document):
dataset_id, document_id = add_document dataset_id, document_id = add_document
parse_documents(HttpApiAuth, dataset_id, {"document_ids": [document_id]}) parse_documents(http_api_auth, dataset_id, {"document_ids": [document_id]})
condition(HttpApiAuth, dataset_id) condition(http_api_auth, dataset_id)
chunk_ids = batch_add_chunks(HttpApiAuth, dataset_id, document_id, 4) chunk_ids = batch_add_chunks(http_api_auth, dataset_id, document_id, 4)
sleep(1) # issues/6487 sleep(1) # issues/6487
return dataset_id, document_id, chunk_ids return dataset_id, document_id, chunk_ids
@pytest.fixture(scope="class") @pytest.fixture(scope="class")
def add_chat_assistants(request, HttpApiAuth, add_document): def add_chat_assistants(request, http_api_auth, add_document):
def cleanup(): def cleanup():
delete_chat_assistants(HttpApiAuth) delete_chat_assistants(http_api_auth)
request.addfinalizer(cleanup) request.addfinalizer(cleanup)
dataset_id, document_id = add_document dataset_id, document_id = add_document
parse_documents(HttpApiAuth, dataset_id, {"document_ids": [document_id]}) parse_documents(http_api_auth, dataset_id, {"document_ids": [document_id]})
condition(HttpApiAuth, dataset_id) condition(http_api_auth, dataset_id)
return dataset_id, document_id, batch_create_chat_assistants(HttpApiAuth, 5) return dataset_id, document_id, batch_create_chat_assistants(http_api_auth, 5)

View file

@ -81,7 +81,7 @@ def generate_unique_email(prefix: str = "test") -> str:
@pytest.fixture(scope="function") @pytest.fixture(scope="function")
def clear_users(request, HttpApiAuth): def clear_users(request, http_api_auth):
"""Fixture to clean up users created during tests.""" """Fixture to clean up users created during tests."""
created_user_ids: list[str] = [] created_user_ids: list[str] = []
@ -95,19 +95,19 @@ def clear_users(request, HttpApiAuth):
@pytest.fixture(name="test_user") @pytest.fixture(name="test_user")
def fixture_test_user(WebApiAuth: RAGFlowWebApiAuth) -> dict[str, Any]: def fixture_test_user(web_api_auth: RAGFlowWebApiAuth) -> dict[str, Any]:
"""Get the current authenticated user's information for update tests. """Get the current authenticated user's information for update tests.
Since users can only update their own account, we use the authenticated Since users can only update their own account, we use the authenticated
user's information instead of creating a new user. user's information instead of creating a new user.
Args: Args:
WebApiAuth: Authenticated user's auth object. web_api_auth: Authenticated user's auth object.
Returns: Returns:
Dictionary containing user_id, email, and original_nickname. Dictionary containing user_id, email, and original_nickname.
""" """
res: dict[str, Any] = get_user_info(WebApiAuth) res: dict[str, Any] = get_user_info(web_api_auth)
assert res["code"] == 0, f"Failed to get user info: {res}" assert res["code"] == 0, f"Failed to get user info: {res}"
user_data: dict[str, Any] = res["data"] user_data: dict[str, Any] = res["data"]