From 231ab1e65b6f4b69c8c65e01b5e3be050273ec84 Mon Sep 17 00:00:00 2001
From: Waleed Akbar <wakbar@cttc.es>
Date: Fri, 26 Apr 2024 16:19:31 +0000
Subject: [PATCH] "test_stop_collector" and "test_select_collectors" are added
---
src/telemetry_frontend/tests/test_unitary.py | 14 +++++++++++++-
1 file changed, 13 insertions(+), 1 deletion(-)
diff --git a/src/telemetry_frontend/tests/test_unitary.py b/src/telemetry_frontend/tests/test_unitary.py
index 4f37514dc..4b2594839 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
--
GitLab