chore: remove git directory after tests (#484)

Author: mhordynski

packages/ragbits-document-search/tests/integration/test_git_source.py

@@ -1,4 +1,5 @@
 import os
+import shutil
 from pathlib import Path
 
 import pytest
@@ -14,6 +15,18 @@
 TEST_REPO_SSH_URL = "git@github.com:psf/requests.git"
 
 
+@pytest.fixture(autouse=True)
+def cleanup_test_dirs():
+    """Clean up test directories after each test."""
+    yield
+    # Get the test directory path
+    test_dir = Path(__file__).parent
+    # Remove the git directory if it exists
+    git_dir = test_dir / "git"
+    if git_dir.exists():
+        shutil.rmtree(git_dir)
+
+
 async def test_git_source_fetch_file():
     """Test fetching a specific file from a git repository."""
     # Fetch the README.md file

packages/ragbits-document-search/tests/unit/test_git_source.py

@@ -1,4 +1,5 @@
 import os
+import shutil
 from pathlib import Path
 from unittest.mock import MagicMock, patch
 
@@ -12,6 +13,18 @@
 os.environ[LOCAL_STORAGE_DIR_ENV] = Path(__file__).parent.as_posix()
 
 
+@pytest.fixture(autouse=True)
+def cleanup_test_dirs():
+    """Clean up test directories after each test."""
+    yield
+    # Get the test directory path
+    test_dir = Path(__file__).parent
+    # Remove the git directory if it exists
+    git_dir = test_dir / "git"
+    if git_dir.exists():
+        shutil.rmtree(git_dir)
+
+
 def test_id():
     """Test the ID property for GitSource"""
     # Test basic case