cecli/tests/basic/test_scripting.py at main · cecli-dev/cecli · GitHub

Repository navigation

from pathlib import Path

from unittest.mock import patch

from cecli.coders import Coder

from cecli.models import Model

from cecli.utils import GitTemporaryDirectory

class TestScriptingAPI:

@patch("cecli.coders.base_coder.Coder.send")

async def test_basic_scripting(self, mock_send):

with GitTemporaryDirectory():

# Setup - create an async generator mock

async def mock_send_side_effect(messages, functions=None, tools=None):

# Simulate the async generator behavior

coder.partial_response_content = "Changes applied successfully."

coder.partial_response_function_call = None

yield "Changes applied successfully."

mock_send.side_effect = mock_send_side_effect

# Test script

fname = Path("greeting.py")

fname.touch()

fnames = [str(fname)]

model = Model("gpt-4-turbo")

coder = await Coder.create(main_model=model, fnames=fnames)

result1 = await coder.run("make a script that prints hello world")

result2 = await coder.run("make it say goodbye")

# Assertions

assert mock_send.call_count == 2

assert result1 == "Changes applied successfully."

assert result2 == "Changes applied successfully."