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

@@ -72,7 +72,9 @@ jobs:
- name: Run unittest tests with coverage
run: |
pytest --cov=autogpt --without-integration --without-slow-integration --cov-report term-missing --cov-branch --cov-report xml --cov-report term
pytest --cov=autogpt tests/integration/goal_oriented --cov-report term-missing --cov-branch --cov-report xml --cov-report term
env:
OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
- name: Generate coverage report
run: |
coverage report

View File

@@ -1,37 +0,0 @@
name: Goal Oriented Tasks
on:
push:
branches:
- master
pull_request:
branches:
- master
jobs:
build:
runs-on: ubuntu-latest
strategy:
matrix:
python-version: ['3.10']
steps:
- name: Check out repository
uses: actions/checkout@v3
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v3
with:
python-version: ${{ matrix.python-version }}
- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install -r requirements.txt
- name: Generate coverage report
run: |
coverage run --source=autogpt -m pytest -s -k tests/integration/goal_oriented
env:
OPENAI_API_KEY: 'dummy_api_key'

5
tests/conftest.py Normal file
View File

@@ -0,0 +1,5 @@
from dotenv import load_dotenv
load_dotenv()
# Your other pytest configurations and fixtures

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"