From d5f85896bd7d9976df300f710062bc0c5f8544d3 Mon Sep 17 00:00:00 2001 From: Daniel McKnight Date: Fri, 20 Dec 2024 09:56:07 -0800 Subject: [PATCH] Update unit tests to account for refactored `api_services` config --- tests/test_diana_utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_diana_utils.py b/tests/test_diana_utils.py index 3508f90..87efc1f 100644 --- a/tests/test_diana_utils.py +++ b/tests/test_diana_utils.py @@ -292,9 +292,9 @@ def test_get_chatbots_mq_config(self, confirm): def test_get_unconfigured_backend_services(self): from neon_diana_utils.configuration import _get_unconfigured_mq_backend_services - all_configured = {'keys': {'api_services': {'configured': True}, - 'emails': {'configured': True}, + all_configured = {'keys': {'emails': {'configured': True}, 'track_my_brands': True}, + 'api_services': {'configured': True}, 'LLM_VLLM': {'config': False}, 'LLM_CHAT_GPT': {'config': False}, 'LLM_CLAUDE': {'': ''}, @@ -304,8 +304,8 @@ def test_get_unconfigured_backend_services(self): disabled = _get_unconfigured_mq_backend_services(all_configured) self.assertEqual(disabled, set()) - none_configured = {'keys': {'api_services': {}, - 'emails': None}, + none_configured = {'keys': {'emails': None}, + 'api_services': {}, 'LLM_CHAT_GPT': {}} disabled = _get_unconfigured_mq_backend_services(none_configured) self.assertEqual(disabled, {'neon-api-proxy', 'neon-brands-service',