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}"