mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-29 03:44:28 +01:00
28 lines
893 B
Python
28 lines
893 B
Python
import os
|
|
|
|
import pytest
|
|
|
|
from autogpt.commands.file_operations import read_file
|
|
from autogpt.config import Config
|
|
from tests.integration.agent_factory import create_writer_agent
|
|
from tests.integration.agent_utils import run_interaction_loop
|
|
from tests.utils import requires_api_key
|
|
|
|
CFG = Config()
|
|
|
|
|
|
@requires_api_key("OPENAI_API_KEY")
|
|
@pytest.mark.vcr
|
|
def test_write_file(workspace) -> None:
|
|
CFG.workspace_path = workspace.root
|
|
CFG.file_logger_path = os.path.join(workspace.root, "file_logger.txt")
|
|
|
|
file_name = str(workspace.get_path("hello_world.txt"))
|
|
agent = create_writer_agent(workspace)
|
|
try:
|
|
run_interaction_loop(agent, 40)
|
|
# catch system exit exceptions
|
|
except SystemExit: # the agent returns an exception when it shuts down
|
|
content = read_file(file_name)
|
|
assert content == "Hello World", f"Expected 'Hello World', got {content}"
|