mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-20 12:24:59 +00:00
4bb043f5c0
# Context:
USER: try using create_input(StringIO('')) in the tests
ASSISTANT: I understand that you want to use `create_input(StringIO(''))` in the tests. However, I need more information about which test cases you want to apply this change to and how you want to use it. Please provide more details or examples.
USER: the ones that currently use create_pipe_input()
ASSISTANT: I see that you want to replace `create_pipe_input()` with `create_input(StringIO(''))` in the test cases. Here are the changes for the `tests/test_main.py` file:
```python
tests/test_main.py
<<<<<<< ORIGINAL
from prompt_toolkit.input import create_pipe_input
=======
from prompt_toolkit.input import create_input
from io import StringIO
>>>>>>> UPDATED
```
```python
tests/test_main.py
<<<<<<< ORIGINAL
pipe_input = create_pipe_input()
=======
pipe_input = create_input(StringIO(''))
>>>>>>> UPDATED
```
```python
tests/test_main.py
<<<<<<< ORIGINAL
pipe_input = create_pipe_input()
=======
pipe_input = create_input(StringIO(''))
>>>>>>> UPDATED
```
```python
tests/test_main.py
<<<<<<< ORIGINAL
pipe_input = create_pipe_input()
=======
pipe_input = create_input(StringIO(''))
>>>>>>> UPDATED
```
These changes will replace the usage of `create_pipe_input()` with `create_input(StringIO(''))` in the test cases.
44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
import os
|
|
import sys
|
|
import tempfile
|
|
from unittest import TestCase
|
|
from aider.main import main
|
|
import subprocess
|
|
from prompt_toolkit.input import create_input
|
|
from io import StringIO
|
|
from prompt_toolkit.output import DummyOutput
|
|
|
|
|
|
class TestMain(TestCase):
|
|
def test_main_with_empty_dir_no_files_on_command(self):
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
os.chdir(temp_dir)
|
|
pipe_input = create_input(StringIO(''))
|
|
save_stdin = sys.stdin
|
|
sys.stdin = pipe_input
|
|
main([], input=pipe_input, output=DummyOutput())
|
|
sys.stdin = save_stdin
|
|
pipe_input.close()
|
|
|
|
def test_main_with_empty_dir_new_file(self):
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
os.chdir(temp_dir)
|
|
pipe_input = create_input(StringIO(''))
|
|
save_stdin = sys.stdin
|
|
sys.stdin = pipe_input
|
|
main(["foo.txt"], input=pipe_input, output=DummyOutput())
|
|
sys.stdin = save_stdin
|
|
pipe_input.close()
|
|
self.assertTrue(os.path.exists("foo.txt"))
|
|
|
|
def test_main_with_empty_git_dir_new_file(self):
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
os.chdir(temp_dir)
|
|
subprocess.run(["git", "init"], cwd=temp_dir)
|
|
pipe_input = create_input(StringIO(''))
|
|
save_stdin = sys.stdin
|
|
sys.stdin = pipe_input
|
|
main(["--yes", "foo.txt"], input=pipe_input, output=DummyOutput())
|
|
sys.stdin = save_stdin
|
|
pipe_input.close()
|
|
self.assertTrue(os.path.exists("foo.txt"))
|