This commit is contained in:
Paul Gauthier 2023-07-16 09:02:35 -03:00
parent 9785ede960
commit cdf35cd122
2 changed files with 16 additions and 14 deletions

View file

@ -1,24 +1,11 @@
import os
import tempfile
import unittest
from unittest.mock import patch
from aider.io import InputOutput
from aider.repomap import RepoMap
class IgnorantTemporaryDirectory:
def __init__(self):
self.temp_dir = tempfile.TemporaryDirectory()
def __enter__(self):
return self.temp_dir.__enter__()
def __exit__(self, exc_type, exc_val, exc_tb):
try:
self.temp_dir.__exit__(exc_type, exc_val, exc_tb)
except OSError:
pass # Ignore errors (Windows)
from .utils import IgnorantTemporaryDirectory
class TestRepoMap(unittest.TestCase):

15
tests/utils.py Normal file
View file

@ -0,0 +1,15 @@
import tempfile
class IgnorantTemporaryDirectory:
def __init__(self):
self.temp_dir = tempfile.TemporaryDirectory()
def __enter__(self):
return self.temp_dir.__enter__()
def __exit__(self, exc_type, exc_val, exc_tb):
try:
self.temp_dir.__exit__(exc_type, exc_val, exc_tb)
except OSError:
pass # Ignore errors (Windows)