mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2026-01-21 23:14:24 +01:00
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
import json
|
|
import os
|
|
import pytest
|
|
import shutil
|
|
from agbenchmark.mocks.tests.retrieval_manual import mock_retrieval
|
|
import requests
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def config():
|
|
config_file = os.path.abspath("agbenchmark/config.json")
|
|
print(f"Config file: {config_file}")
|
|
with open(config_file, "r") as f:
|
|
config = json.load(f)
|
|
return config
|
|
|
|
|
|
@pytest.fixture
|
|
def workspace(config):
|
|
yield config["workspace"]
|
|
# teardown after test function completes
|
|
for filename in os.listdir(config["workspace"]):
|
|
file_path = os.path.join(config["workspace"], filename)
|
|
try:
|
|
if os.path.isfile(file_path) or os.path.islink(file_path):
|
|
os.unlink(file_path)
|
|
elif os.path.isdir(file_path):
|
|
shutil.rmtree(file_path)
|
|
except Exception as e:
|
|
print(f"Failed to delete {file_path}. Reason: {e}")
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def server_response(request, config):
|
|
task = request.param # The task is passed in indirectly
|
|
print(f"Server starting at {request.module}")
|
|
# response = requests.post(
|
|
# f"{config['hostname']}:{config['port']}", data={"task": task}
|
|
# )
|
|
# assert (
|
|
# response.status_code == 200
|
|
# ), f"Request failed with status code {response.status_code}"
|
|
mock_retrieval(task, config["workspace"])
|