add decorator to tests

Signed-off-by: Merwane Hamadi <merwanehamadi@gmail.com>
This commit is contained in:
Merwane Hamadi
2023-04-22 12:39:56 -07:00
committed by Reinier van der Leer
parent 996a3b331a
commit eb5a8a87d8
5 changed files with 26 additions and 38 deletions

View File

@@ -0,0 +1,14 @@
import os
import pytest
def requires_openai_api_key(func):
def wrapper(*args, **kwargs):
if not os.environ.get('OPENAI_API_KEY'):
pytest.skip(
"Environment variable 'OPENAI_API_KEY' is not set, skipping the test."
)
else:
return func(*args, **kwargs)
return wrapper

View File

@@ -13,6 +13,7 @@ from autogpt.memory import get_memory
# from autogpt.prompt import Prompt
from autogpt.workspace import WORKSPACE_PATH
from tests.integration.goal_oriented.decorators import requires_openai_api_key
from tests.integration.goal_oriented.vcr_helper import before_record_request
current_file_dir = os.path.dirname(os.path.abspath(__file__))
@@ -27,8 +28,11 @@ my_vcr = vcr.VCR(
CFG = Config()
@requires_openai_api_key
@pytest.mark.integration_test
def test_write_file() -> None:
# Your test code here
# if file exist
file_name = "hello_world.txt"