fixed failing tests

This commit is contained in:
Nicholas Albion
2023-09-27 11:49:51 +10:00
parent 0ba4c1976c
commit 5f110322ed
2 changed files with 4 additions and 4 deletions

View File

@@ -10,7 +10,7 @@ from database.models.files import File
from database.models.development_steps import DevelopmentSteps
from helpers.Project import Project, update_file, clear_directory
from helpers.AgentConvo import AgentConvo
from test.mock_terminal_size import mock_terminal_size
from test.test_utils import mock_terminal_size
SEND_TO_LLM = False
WRITE_TO_FILE = False

View File

@@ -1,6 +1,7 @@
import builtins
import os
from unittest.mock import patch, Mock
import pytest
from unittest.mock import patch
from helpers.AgentConvo import AgentConvo
from dotenv import load_dotenv
@@ -9,7 +10,6 @@ load_dotenv()
from main import get_custom_print
from .Developer import Developer, ENVIRONMENT_SETUP_STEP
from helpers.Project import Project
from test.mock_terminal_size import mock_terminal_size
class TestDeveloper:
@@ -33,7 +33,7 @@ class TestDeveloper:
self.project.current_step = ENVIRONMENT_SETUP_STEP
self.developer = Developer(self.project)
# @pytest.mark.uses_tokens
@pytest.mark.uses_tokens
@patch('helpers.AgentConvo.get_saved_development_step')
@patch('helpers.AgentConvo.save_development_step')
@patch('helpers.AgentConvo.create_gpt_chat_completion',