From dd20324cb6426ea6f4574d3a58992b822361f452 Mon Sep 17 00:00:00 2001
From: himanshu_gupta <himanshu.gupta@nimblework.com>
Date: Thu, 12 Dec 2024 16:07:13 +0530
Subject: [PATCH] Updated test cases

---
 tests/integration_test/services_test.py | 35 +++++++------------------
 1 file changed, 9 insertions(+), 26 deletions(-)

diff --git a/tests/integration_test/services_test.py b/tests/integration_test/services_test.py
index 404648470..b84d84416 100644
--- a/tests/integration_test/services_test.py
+++ b/tests/integration_test/services_test.py
@@ -29685,10 +29685,9 @@ def test_leave_non_existent_bot_1():
     assert not actual["success"]
 
 def test_payload_upload_api_with_float_field(monkeypatch):
-    bot_settings = BotSettings.objects(bot=pytest.bot).get()
-    bot_settings.cognition_collections_limit = 20
-    bot_settings.llm_settings['enable_faq'] = True
-    bot_settings.save()
+
+    def _mock_get_bot_settings(*args, **kwargs):
+        return BotSettings(bot=pytest.bot, user="integration@demo.ai", llm_settings=LLMSettings(enable_faq=True), cognition_collections_limit = 20)
     metadata = {
         "metadata": [{"column_name": "item", "data_type": "str", "enable_search": True, "create_embeddings": True},
                      {"column_name": "price", "data_type": "float", "enable_search": True, "create_embeddings": True}],
@@ -29721,16 +29720,11 @@ def test_payload_upload_api_with_float_field(monkeypatch):
     assert data_dict['data']['item'] == 'Box'
     assert data_dict['data']['price'] == 54.02
     CognitionData.objects(bot=pytest.bot, collection="with_float_field").delete()
-    bot_settings = BotSettings.objects(bot=pytest.bot).get()
-    bot_settings.llm_settings['enable_faq'] = False
-    bot_settings.save()
 
 def test_payload_upload_api_with_float_field_value_integer(monkeypatch):
-    bot_settings = BotSettings.objects(bot=pytest.bot).get()
-    bot_settings.content_importer_limit_per_day = 10
-    bot_settings.cognition_collections_limit = 20
-    bot_settings.llm_settings['enable_faq'] = True
-    bot_settings.save()
+
+    def _mock_get_bot_settings(*args, **kwargs):
+        return BotSettings(bot=pytest.bot, user="integration@demo.ai", llm_settings=LLMSettings(enable_faq=True), cognition_collections_limit = 20)
     metadata = {
         "metadata": [{"column_name": "item", "data_type": "str", "enable_search": True, "create_embeddings": True},
                      {"column_name": "price", "data_type": "float", "enable_search": True, "create_embeddings": True}],
@@ -29765,16 +29759,10 @@ def test_payload_upload_api_with_float_field_value_integer(monkeypatch):
     assert isinstance(data_dict['data']['price'], float)
     assert data_dict['data']['price'] == 54.0
     CognitionData.objects(bot=pytest.bot, collection="with_float_field_value_integer").delete()
-    bot_settings = BotSettings.objects(bot=pytest.bot).get()
-    bot_settings.llm_settings['enable_faq'] = False
-    bot_settings.save()
 
 def test_update_payload_upload_api_with_float_field_value_integer(monkeypatch):
-    bot_settings = BotSettings.objects(bot=pytest.bot).get()
-    bot_settings.content_importer_limit_per_day = 10
-    bot_settings.cognition_collections_limit = 20
-    bot_settings.llm_settings['enable_faq'] = True
-    bot_settings.save()
+    def _mock_get_bot_settings(*args, **kwargs):
+        return BotSettings(bot=pytest.bot, user="integration@demo.ai", llm_settings=LLMSettings(enable_faq=True), cognition_collections_limit = 20)
     metadata = {
         "metadata": [{"column_name": "item", "data_type": "str", "enable_search": True, "create_embeddings": True},
                      {"column_name": "price", "data_type": "float", "enable_search": True, "create_embeddings": True}],
@@ -29810,8 +29798,6 @@ def test_update_payload_upload_api_with_float_field_value_integer(monkeypatch):
     assert isinstance(data_dict['data']['price'], float)
     assert data_dict['data']['price'] == 54.08
 
-
-
     update_payload = {
         "data": {"item": "Box", "price": 27},
         "content_type": "json",
@@ -29833,7 +29819,4 @@ def test_update_payload_upload_api_with_float_field_value_integer(monkeypatch):
     assert data_dict['data']['item'] == 'Box'
     assert isinstance(data_dict['data']['price'], float)
     assert data_dict['data']['price'] == 27.0
-    CognitionData.objects(bot=pytest.bot, collection="update_with_float_field_value_integer").delete()
-    bot_settings = BotSettings.objects(bot=pytest.bot).get()
-    bot_settings.llm_settings['enable_faq'] = False
-    bot_settings.save()
\ No newline at end of file
+    CognitionData.objects(bot=pytest.bot, collection="update_with_float_field_value_integer").delete()
\ No newline at end of file