From 3a61919344e7ab1955a0b539c8fe98fc95ee4aea Mon Sep 17 00:00:00 2001 From: Saba Date: Tue, 13 Jun 2023 16:32:47 -0700 Subject: [PATCH] Fix failing unit tests by hard-coding model presence of expected search types --- tests/conftest.py | 4 ++++ tests/test_conversation_utils.py | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/conftest.py b/tests/conftest.py index 84ec658d..6ef2394f 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -159,6 +159,10 @@ def client(content_config: ContentConfig, search_config: SearchConfig, processor state.config.search_type = search_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) return TestClient(app) diff --git a/tests/test_conversation_utils.py b/tests/test_conversation_utils.py index 06a507c5..ac8a7665 100644 --- a/tests/test_conversation_utils.py +++ b/tests/test_conversation_utils.py @@ -32,7 +32,7 @@ class TestTruncateMessage: def test_truncate_message_first_large(self): 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?" copy_big_chat_message = big_chat_message.copy() chat_messages.insert(0, big_chat_message)