mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2026-02-02 12:54:28 +01:00
add decorator to tests
Signed-off-by: Merwane Hamadi <merwanehamadi@gmail.com>
This commit is contained in:
committed by
Reinier van der Leer
parent
996a3b331a
commit
eb5a8a87d8
4
.github/workflows/ci.yml
vendored
4
.github/workflows/ci.yml
vendored
@@ -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
|
||||
|
||||
37
.github/workflows/goal_oriented_tasks.yml
vendored
37
.github/workflows/goal_oriented_tasks.yml
vendored
@@ -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
5
tests/conftest.py
Normal file
@@ -0,0 +1,5 @@
|
||||
from dotenv import load_dotenv
|
||||
|
||||
load_dotenv()
|
||||
|
||||
# Your other pytest configurations and fixtures
|
||||
14
tests/integration/goal_oriented/decorators.py
Normal file
14
tests/integration/goal_oriented/decorators.py
Normal 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
|
||||
@@ -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"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user