From 16aaf12fc8e520d78b1e912370c9c38dac22ec26 Mon Sep 17 00:00:00 2001 From: Mahesh Date: Wed, 13 Dec 2023 17:43:26 +0530 Subject: [PATCH] Fixed test cases. --- tests/integration_test/chat_service_test.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/integration_test/chat_service_test.py b/tests/integration_test/chat_service_test.py index 288048f4e..3ac1eaa79 100644 --- a/tests/integration_test/chat_service_test.py +++ b/tests/integration_test/chat_service_test.py @@ -395,14 +395,17 @@ def test_business_messages_without_message(mock_process_message, mock_check_mess @patch("kairon.chat.handlers.channels.business_messages.BusinessMessagesHandler.check_message_create_time") +@patch("kairon.chat.agent_processor.AgentProcessor.reload") +@patch("kairon.chat.agent_processor.AgentProcessor.cache_provider.get") @patch('oauth2client.service_account.ServiceAccountCredentials.from_json_keyfile_dict') @patch('businessmessages.businessmessages_v1_client.BusinessmessagesV1') -def test_business_messages_with_valid_data(mock_business_messages, mock_credentials, +def test_business_messages_with_valid_data(mock_business_messages, mock_credentials, mock_get_agent, mock_reload, mock_check_message_create_time): mock_check_message_create_time.return_value = True + mock_get_agent.return_value = KaironAgent mock_credentials.return_value = {} mock_business_messages.return_value = {} - with patch.object(Agent, "handle_message") as mock_agent: + with patch.object(KaironAgent, "handle_message") as mock_agent: mock_agent.side_effect = mock_agent_response response = client.post( f"/api/bot/business_messages/{bot}/{token}",