From b5b6538762096846928e5b0bca35f859b003b9e2 Mon Sep 17 00:00:00 2001
From: Alex <a@tushynski.me>
Date: Sun, 12 Apr 2026 12:35:23 +0100
Subject: [PATCH] fix: tests

---
 tests/parser/remote/test_github_loader.py | 3 ++-
 tests/test_worker.py                      | 2 +-
 tests/vectorstore/test_pgvector.py        | 1 +
 3 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/parser/remote/test_github_loader.py b/tests/parser/remote/test_github_loader.py
index bd0a5fc0..32662279 100644
--- a/tests/parser/remote/test_github_loader.py
+++ b/tests/parser/remote/test_github_loader.py
@@ -31,6 +31,7 @@ class TestGitHubLoaderFetchFileContent:
         mock_get.assert_called_once_with(
             "https://api.github.com/repos/owner/repo/contents/README.md",
             headers=loader.headers,
+            timeout=100,
         )
 
     @patch("application.parser.remote.github_loader.requests.get")
@@ -66,7 +67,7 @@ class TestGitHubLoaderFetchRepoFiles:
     def test_recurses_directories(self, mock_get):
         loader = GitHubLoader()
 
-        def side_effect(url, headers=None):
+        def side_effect(url, headers=None, timeout=None):
             if url.endswith("/contents/"):
                 return make_response([
                     {"type": "file", "path": "README.md"},
diff --git a/tests/test_worker.py b/tests/test_worker.py
index 417ee160..29bb4f15 100644
--- a/tests/test_worker.py
+++ b/tests/test_worker.py
@@ -342,7 +342,7 @@ class TestDownloadFile:
         dest = str(tmp_path / "downloaded.txt")
         download_file("http://example.com/file", {"key": "val"}, dest)
 
-        mock_get.assert_called_once_with("http://example.com/file", params={"key": "val"})
+        mock_get.assert_called_once_with("http://example.com/file", params={"key": "val"}, timeout=100)
         with open(dest, "rb") as f:
             assert f.read() == b"file content"
 
diff --git a/tests/vectorstore/test_pgvector.py b/tests/vectorstore/test_pgvector.py
index 08cfc11e..fa01279a 100644
--- a/tests/vectorstore/test_pgvector.py
+++ b/tests/vectorstore/test_pgvector.py
@@ -70,6 +70,7 @@ class TestPGVectorStoreInit:
             mock_get_emb.return_value = Mock(dimension=768)
             mock_settings.EMBEDDINGS_NAME = "test_model"
             mock_settings.PGVECTOR_CONNECTION_STRING = None
+            mock_settings.POSTGRES_URI = None
 
             from application.vectorstore.pgvector import PGVectorStore