Commit 231ab1e6 authored by Waleed Akbar's avatar Waleed Akbar
Browse files

"test_stop_collector" and "test_select_collectors" are added

parent e2c690c1

src/telemetry_frontend/tests/test_unitary.py

+13 −1
Original line number Diff line number Diff line
@@ -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