Skip to content

Commit 4b4d727

Browse files
committed
Fix mock setups to match how generate_title accesses title
1 parent e6754df commit 4b4d727

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

server/api/views/uploadFile/test_title.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def test_falls_back_to_first_page_text_if_metadata_title_does_not_match_regex(se
5757
def test_falls_back_to_chatgpt_if_no_title_found(self, mock_openAI):
5858
doc = MagicMock()
5959
doc.metadata = {"title": None}
60-
doc.get_text.return_value = []
60+
doc[0].get_text.return_value = []
6161

6262
mock_response = MagicMock()
6363
mock_response.choices = [MagicMock()]
@@ -72,7 +72,7 @@ def test_falls_back_to_chatgpt_if_no_title_found(self, mock_openAI):
7272
def test_strips_quotes_from_openai_title(self, mock_openAI):
7373
doc = MagicMock()
7474
doc.metadata = {"title": None}
75-
doc.get_text.return_value = []
75+
doc[0].get_text.return_value = []
7676

7777
mock_response = MagicMock()
7878
mock_response.choices = [MagicMock()]
@@ -87,7 +87,7 @@ def test_strips_quotes_from_openai_title(self, mock_openAI):
8787
def test_truncates_long_openai_title(self, mock_openAI):
8888
doc = MagicMock()
8989
doc.metadata = {"title": None}
90-
doc.get_text.return_value = []
90+
doc[0].get_text.return_value = []
9191

9292
mock_response = MagicMock()
9393
mock_response.choices = [MagicMock()]

0 commit comments

Comments
 (0)