wip: Refactor main function to include tests for empty directory with no files and new file creation.

This commit is contained in:
Paul Gauthier 2023-05-11 17:33:59 -07:00
parent 490a6999fb
commit e500f8e37f

View file

@ -5,9 +5,20 @@ from unittest import TestCase
from aider.main import main
class TestMain(TestCase):
def test_main_with_dev_null(self):
def test_main_with_empty_dir_no_files_on_command(self):
with tempfile.TemporaryDirectory() as temp_dir:
os.chdir(temp_dir)
with open(os.devnull, 'r') as dev_null:
sys.stdin.close()
save_stdin = sys.stdin
sys.stdin = dev_null
main()
sys.stdin = save_stdin
def test_main_with_empty_dir_new_file(self):
with tempfile.TemporaryDirectory() as temp_dir:
os.chdir(temp_dir)
with open(os.devnull, 'r') as dev_null:
save_stdin = sys.stdin
sys.stdin = dev_null
main(['foo.txt'])
sys.stdin = save_stdin