Spaces:
Build error
Build error
File size: 1,414 Bytes
ad4c1f0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | import random
import string
import tempfile
import pytest
from pytest_mock import MockerFixture
import autogpt.commands.execute_code as sut # system under testing
from autogpt.config import Config
@pytest.fixture
def config_allow_execute(config: Config, mocker: MockerFixture):
yield mocker.patch.object(config, "execute_local_commands", True)
@pytest.fixture
def python_test_file(config: Config, random_string):
temp_file = tempfile.NamedTemporaryFile(dir=config.workspace_path, suffix=".py")
temp_file.write(str.encode(f"print('Hello {random_string}!')"))
temp_file.flush()
yield temp_file.name
temp_file.close()
@pytest.fixture
def random_string():
return "".join(random.choice(string.ascii_lowercase) for _ in range(10))
def test_execute_python_file(python_test_file: str, random_string: str):
result = sut.execute_python_file(python_test_file)
assert result == f"Hello {random_string}!\n"
def test_execute_python_file_invalid():
assert all(
s in sut.execute_python_file("not_python").lower()
for s in ["error:", "invalid", ".py"]
)
assert all(
s in sut.execute_python_file("notexist.py").lower()
for s in ["error:", "does not exist"]
)
def test_execute_shell(config_allow_execute, random_string):
result = sut.execute_shell(f"echo 'Hello {random_string}!'")
assert f"Hello {random_string}!" in result
|