From 97d3d1f18f07101a860952f40a96b7cfd633fb3c Mon Sep 17 00:00:00 2001 From: Ivan Usov Date: Thu, 9 Nov 2023 15:02:59 +0100 Subject: [PATCH] fix: bec_service test --- bec_lib/tests/test_bec_service.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/bec_lib/tests/test_bec_service.py b/bec_lib/tests/test_bec_service.py index f8013352..66a814ce 100644 --- a/bec_lib/tests/test_bec_service.py +++ b/bec_lib/tests/test_bec_service.py @@ -4,10 +4,10 @@ from unittest import mock import pytest import bec_lib -from bec_lib import BECMessage +from bec_lib import messages from bec_lib.bec_service import BECService -from bec_lib.BECMessage import BECStatus from bec_lib.endpoints import MessageEndpoints +from bec_lib.messages import BECStatus from bec_lib.service_config import ServiceConfig # pylint: disable=no-member @@ -121,8 +121,8 @@ def test_bec_service_update_existing_services(): f'{MessageEndpoints.service_status("service2")}:val'.encode(), ] service_msgs = [ - BECMessage.StatusMessage(name="service1", status=BECStatus.RUNNING, info={}, metadata={}), - BECMessage.StatusMessage(name="service2", status=BECStatus.IDLE, info={}, metadata={}), + messages.StatusMessage(name="service1", status=BECStatus.RUNNING, info={}, metadata={}), + messages.StatusMessage(name="service2", status=BECStatus.IDLE, info={}, metadata={}), ] connector_cls = mock.MagicMock() connector_cls().producer().keys.return_value = service_keys @@ -144,7 +144,7 @@ def test_bec_service_update_existing_services_ignores_wrong_msgs(): f'{MessageEndpoints.service_status("service2")}:val'.encode(), ] service_msgs = [ - BECMessage.StatusMessage(name="service1", status=BECStatus.RUNNING, info={}, metadata={}), + messages.StatusMessage(name="service1", status=BECStatus.RUNNING, info={}, metadata={}), None, ] connector_cls = mock.MagicMock()