Skip to content

Commit

Permalink
Update unit tests to account for refactored api_services config
Browse files Browse the repository at this point in the history
  • Loading branch information
NeonDaniel committed Dec 20, 2024
1 parent 79172da commit d5f8589
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tests/test_diana_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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': {'': ''},
Expand All @@ -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',
Expand Down

0 comments on commit d5f8589

Please sign in to comment.