klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-02 21:19:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix failing unit tests by hard-coding model presence of expected search types

Browse Source
This commit is contained in:
Saba
2023-06-13 16:32:47 -07:00
parent 019d3732de
commit 3a61919344
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/conftest.py
View File

@@ -159,6 +159,10 @@ def client(content_config: ContentConfig, search_config: SearchConfig, processor
state.config.search_type = search_config state.config.search_type = search_config
state.SearchType = configure_search_types(state.config) state.SearchType = configure_search_types(state.config)
# These lines help us Mock the Search models for these search types
state.model.org_search = {}
state.model.image_search = {}
configure_routes(app) configure_routes(app)
return TestClient(app) return TestClient(app)

2
tests/test_conversation_utils.py
View File

@@ -32,7 +32,7 @@ class TestTruncateMessage:
def test_truncate_message_first_large(self): def test_truncate_message_first_large(self):
chat_messages = ChatMessageFactory.build_batch(25) chat_messages = ChatMessageFactory.build_batch(25)
big_chat_message = ChatMessageFactory.build(content=factory.Faker("paragraph", nb_sentences=1000)) big_chat_message = ChatMessageFactory.build(content=factory.Faker("paragraph", nb_sentences=2000))
big_chat_message.content = big_chat_message.content + "\n" + "Question?" big_chat_message.content = big_chat_message.content + "\n" + "Question?"
copy_big_chat_message = big_chat_message.copy() copy_big_chat_message = big_chat_message.copy()
chat_messages.insert(0, big_chat_message) chat_messages.insert(0, big_chat_message)