mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-29 20:04:30 +01:00
dafbd11686
* Rearrange tests into unit/integration/challenge categories * Fix linting + `tests.challenges` imports * Fix obscured duplicate test in test_url_validation.py * Move VCR conftest to tests.vcr * Specify tests to run & their order (unit -> integration -> challenges) in CI * Fail Docker CI when tests fail * Fix import & linting errors in tests * Fix `get_text_summary` * Fix linting errors * Clean up pytest args in CI * Remove bogus tests from GoCodeo
62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
import os
|
|
|
|
import openai.api_requestor
|
|
import pytest
|
|
from pytest_mock import MockerFixture
|
|
|
|
from .vcr_filter import PROXY, before_record_request, before_record_response
|
|
|
|
BASE_VCR_CONFIG = {
|
|
"record_mode": "new_episodes",
|
|
"before_record_request": before_record_request,
|
|
"before_record_response": before_record_response,
|
|
"filter_headers": [
|
|
"Authorization",
|
|
"X-OpenAI-Client-User-Agent",
|
|
"User-Agent",
|
|
],
|
|
"match_on": ["method", "body"],
|
|
}
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def vcr_config():
|
|
# this fixture is called by the pytest-recording vcr decorator.
|
|
return BASE_VCR_CONFIG
|
|
|
|
|
|
@pytest.fixture()
|
|
def vcr_cassette_dir(request):
|
|
test_name = os.path.splitext(request.node.name)[0]
|
|
return os.path.join("tests/Auto-GPT-test-cassettes", test_name)
|
|
|
|
|
|
def patch_api_base(requestor):
|
|
new_api_base = f"{PROXY}/v1"
|
|
requestor.api_base = new_api_base
|
|
return requestor
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_api_requestor(mocker: MockerFixture):
|
|
original_init = openai.api_requestor.APIRequestor.__init__
|
|
original_validate_headers = openai.api_requestor.APIRequestor._validate_headers
|
|
|
|
def patched_init(requestor, *args, **kwargs):
|
|
original_init(requestor, *args, **kwargs)
|
|
patch_api_base(requestor)
|
|
|
|
def patched_validate_headers(self, supplied_headers):
|
|
headers = original_validate_headers(self, supplied_headers)
|
|
headers["AGENT-MODE"] = os.environ.get("AGENT_MODE")
|
|
headers["AGENT-TYPE"] = os.environ.get("AGENT_TYPE")
|
|
return headers
|
|
|
|
if PROXY:
|
|
mocker.patch("openai.api_requestor.APIRequestor.__init__", new=patched_init)
|
|
mocker.patch.object(
|
|
openai.api_requestor.APIRequestor,
|
|
"_validate_headers",
|
|
new=patched_validate_headers,
|
|
)
|