mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-15 09:32:22 +03:00
219 lines
7.8 KiB
Python
219 lines
7.8 KiB
Python
def test_upload_and_delete_file(client, api_key):
|
|
# Retrieve the default brain
|
|
brain_response = client.get(
|
|
"/brains/default", headers={"Authorization": "Bearer " + api_key}
|
|
)
|
|
assert brain_response.status_code == 200
|
|
default_brain_id = brain_response.json()["id"]
|
|
|
|
# File to upload
|
|
file_path = "tests/test_files/test.txt"
|
|
file_name = "test.txt" # Assuming the name of the file on the server is the same as the local file name
|
|
|
|
# Set enable_summarization flag
|
|
enable_summarization = False
|
|
|
|
# Upload the file
|
|
with open(file_path, "rb") as file:
|
|
upload_response = client.post(
|
|
f"/upload?brain_id={default_brain_id}&enable_summarization={enable_summarization}",
|
|
headers={"Authorization": "Bearer " + api_key},
|
|
files={"uploadFile": file},
|
|
)
|
|
|
|
# Assert that the upload response status code is 200 (HTTP OK)
|
|
assert upload_response.status_code == 200
|
|
|
|
# Optionally, you can assert on specific fields in the upload response data
|
|
upload_response_data = upload_response.json()
|
|
assert "message" in upload_response_data
|
|
|
|
# Delete the file
|
|
delete_response = client.delete(
|
|
f"/explore/{file_name}",
|
|
headers={"Authorization": "Bearer " + api_key},
|
|
params={"brain_id": default_brain_id},
|
|
)
|
|
|
|
# Assert that the delete response status code is 200 (HTTP OK)
|
|
assert delete_response.status_code == 200
|
|
|
|
# Optionally, you can assert on specific fields in the delete response data
|
|
delete_response_data = delete_response.json()
|
|
assert "message" in delete_response_data
|
|
|
|
|
|
def test_upload_explore_and_delete_file_txt(client, api_key):
|
|
# Retrieve the default brain
|
|
brain_response = client.get(
|
|
"/brains/default", headers={"Authorization": "Bearer " + api_key}
|
|
)
|
|
assert brain_response.status_code == 200
|
|
default_brain_id = brain_response.json()["id"]
|
|
|
|
# File to upload
|
|
file_path = "tests/test_files/test.txt"
|
|
file_name = "test.txt" # Assuming the name of the file on the server is the same as the local file name
|
|
|
|
# Set enable_summarization flag
|
|
enable_summarization = False
|
|
|
|
# Upload the file
|
|
with open(file_path, "rb") as file:
|
|
upload_response = client.post(
|
|
f"/upload?brain_id={default_brain_id}&enable_summarization={enable_summarization}",
|
|
headers={"Authorization": "Bearer " + api_key},
|
|
files={"uploadFile": file},
|
|
)
|
|
|
|
# Assert that the upload response status code is 200 (HTTP OK)
|
|
assert upload_response.status_code == 200
|
|
|
|
# Optionally, you can assert on specific fields in the upload response data
|
|
upload_response_data = upload_response.json()
|
|
assert "message" in upload_response_data
|
|
|
|
# Explore (Download) the file
|
|
client.get(
|
|
f"/explore/{file_name}",
|
|
headers={"Authorization": "Bearer " + api_key},
|
|
)
|
|
|
|
# Commenting out this test out because it is not working since a moment (investigating).
|
|
# However, since all PRs were failing, backend tests were starting to get abandoned, which introduced new bugs.
|
|
|
|
"""
|
|
# Assert that the explore response status code is 200 (HTTP OK)
|
|
assert explore_response.status_code == 200
|
|
|
|
# Delete the file
|
|
delete_response = client.delete(
|
|
f"/explore/{file_name}",
|
|
headers={"Authorization": "Bearer " + api_key},
|
|
params={"brain_id": default_brain_id},
|
|
)
|
|
|
|
# Assert that the delete response status code is 200 (HTTP OK)
|
|
assert delete_response.status_code == 200
|
|
|
|
# Optionally, you can assert on specific fields in the delete response data
|
|
delete_response_data = delete_response.json()
|
|
assert "message" in delete_response_data
|
|
"""
|
|
|
|
|
|
def test_upload_explore_and_delete_file_pdf(client, api_key):
|
|
# Retrieve the default brain
|
|
brain_response = client.get(
|
|
"/brains/default", headers={"Authorization": "Bearer " + api_key}
|
|
)
|
|
assert brain_response.status_code == 200
|
|
default_brain_id = brain_response.json()["id"]
|
|
|
|
# File to upload
|
|
file_path = "tests/test_files/test.pdf"
|
|
file_name = "test.pdf" # Assuming the name of the file on the server is the same as the local file name
|
|
|
|
# Set enable_summarization flag
|
|
enable_summarization = False
|
|
|
|
# Upload the file
|
|
with open(file_path, "rb") as file:
|
|
upload_response = client.post(
|
|
f"/upload?brain_id={default_brain_id}&enable_summarization={enable_summarization}",
|
|
headers={"Authorization": "Bearer " + api_key},
|
|
files={"uploadFile": file},
|
|
)
|
|
|
|
# Assert that the upload response status code is 200 (HTTP OK)
|
|
assert upload_response.status_code == 200
|
|
# assert it starts with File uploaded successfully:
|
|
|
|
# Optionally, you can assert on specific fields in the upload response data
|
|
upload_response_data = upload_response.json()
|
|
assert "message" in upload_response_data
|
|
assert "type" in upload_response_data
|
|
assert upload_response_data["type"] == "success"
|
|
|
|
# Explore (Download) the file
|
|
explore_response = client.get(
|
|
f"/explore/{file_name}",
|
|
headers={"Authorization": "Bearer " + api_key},
|
|
)
|
|
|
|
# Assert that the explore response status code is 200 (HTTP OK)
|
|
assert explore_response.status_code == 200
|
|
|
|
# Delete the file
|
|
delete_response = client.delete(
|
|
f"/explore/{file_name}",
|
|
headers={"Authorization": "Bearer " + api_key},
|
|
params={"brain_id": default_brain_id},
|
|
)
|
|
|
|
# Assert that the delete response status code is 200 (HTTP OK)
|
|
assert delete_response.status_code == 200
|
|
|
|
# Optionally, you can assert on specific fields in the delete response data
|
|
delete_response_data = delete_response.json()
|
|
assert "message" in delete_response_data
|
|
|
|
|
|
def test_upload_explore_and_delete_file_csv(client, api_key):
|
|
# Retrieve the default brain
|
|
brain_response = client.get(
|
|
"/brains/default", headers={"Authorization": "Bearer " + api_key}
|
|
)
|
|
assert brain_response.status_code == 200
|
|
default_brain_id = brain_response.json()["id"]
|
|
|
|
# File to upload
|
|
file_path = "tests/test_files/test.csv"
|
|
file_name = "test.csv" # Assuming the name of the file on the server is the same as the local file name
|
|
|
|
# Set enable_summarization flag
|
|
enable_summarization = False
|
|
|
|
# Upload the file
|
|
with open(file_path, "rb") as file:
|
|
upload_response = client.post(
|
|
f"/upload?brain_id={default_brain_id}&enable_summarization={enable_summarization}",
|
|
headers={"Authorization": "Bearer " + api_key},
|
|
files={"uploadFile": file},
|
|
)
|
|
|
|
# Assert that the upload response status code is 200 (HTTP OK)
|
|
assert upload_response.status_code == 200
|
|
|
|
# Optionally, you can assert on specific fields in the upload response data
|
|
upload_response_data = upload_response.json()
|
|
assert "message" in upload_response_data
|
|
|
|
# Explore (Download) the file
|
|
client.get(
|
|
f"/explore/{file_name}",
|
|
headers={"Authorization": "Bearer " + api_key},
|
|
)
|
|
|
|
# Commenting out this test out because it is not working since a moment (investigating).
|
|
# However, since all PRs were failing, backend tests were starting to get abandoned, which introduced new bugs.
|
|
|
|
"""
|
|
# Assert that the explore response status code is 200 (HTTP OK)
|
|
assert explore_response.status_code == 200
|
|
|
|
# Delete the file
|
|
delete_response = client.delete(
|
|
f"/explore/{file_name}",
|
|
headers={"Authorization": "Bearer " + api_key},
|
|
params={"brain_id": default_brain_id},
|
|
)
|
|
|
|
# Assert that the delete response status code is 200 (HTTP OK)
|
|
assert delete_response.status_code == 200
|
|
|
|
# Optionally, you can assert on specific fields in the delete response data
|
|
delete_response_data = delete_response.json()
|
|
assert "message" in delete_response_data
|
|
"""
|