diff --git a/src/telemetry_frontend/tests/test_unitary.py b/src/telemetry_frontend/tests/test_unitary.py index 4f37514dc874531a39499fc509bea56e828a7b17..4b2594839b72c2d7e34c5bc3f04e70e8ad3d0168 100644 --- a/src/telemetry_frontend/tests/test_unitary.py +++ b/src/telemetry_frontend/tests/test_unitary.py @@ -16,8 +16,9 @@ import os import pytest import logging from typing import Union +from common.proto.context_pb2 import Empty from common.Constants import ServiceNameEnum -from common.proto.telemetry_frontend_pb2 import CollectorId +from common.proto.telemetry_frontend_pb2 import CollectorId, CollectorList from common.proto.context_pb2_grpc import add_ContextServiceServicer_to_server from context.client.ContextClient import ContextClient from common.tools.service.GenericGrpcService import GenericGrpcService @@ -167,3 +168,14 @@ def test_start_collector(telemetryFrontend_client): LOGGER.debug(str(response)) assert isinstance(response, CollectorId) +def test_stop_collector(telemetryFrontend_client): + LOGGER.warning('test_stop_collector requesting') + response = telemetryFrontend_client.StopCollector('1') + LOGGER.debug(str(response)) + assert isinstance(response, Empty) + +def test_select_collectors(telemetryFrontend_client): + LOGGER.warning('test_select_collector requesting') + response = telemetryFrontend_client.SelectCollectors() + LOGGER.debug(str(response)) + assert isinstance(response, CollectorList) \ No newline at end of file