Commit f6c040ab authored by Waleed Akbar's avatar Waleed Akbar
Browse files

New test cases are added

parent 1c177bb7

src/telemetry_frontend/tests/test_unitary.py

+17 −3
Original line number Diff line number Diff line
@@ -16,6 +16,7 @@ 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, CollectorList
@@ -28,7 +29,8 @@ from common.Settings import ( 


from telemetry_frontend.client.TelemetryFrontendClient import TelemetryFrontendClient

from telemetry_frontend.service.TelemetryFrontendService import TelemetryFrontendService

from telemetry_frontend.tests.Messages import create_collector_request

from telemetry_frontend.tests.Messages import ( create_collector_id, create_collector_request, 

    create_collector_filter, create_collector_request_a, create_collector_request_b)



from device.client.DeviceClient import DeviceClient

from device.service.DeviceService import DeviceService
@@ -168,14 +170,26 @@ def test_start_collector(telemetryFrontend_client): 
    LOGGER.debug(str(response))

    assert isinstance(response, CollectorId)



def test_start_collector_a(telemetryFrontend_client):

    LOGGER.warning('test_start_collector requesting')

    response = telemetryFrontend_client.StartCollector(create_collector_request_a())

    LOGGER.debug(str(response))

    assert isinstance(response, CollectorId)



def test_start_collector_b(telemetryFrontend_client):

    LOGGER.warning('test_start_collector requesting')

    response = telemetryFrontend_client.StartCollector(create_collector_request_b('1',10,2))

    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')

    response = telemetryFrontend_client.StopCollector(create_collector_id("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()

    response = telemetryFrontend_client.SelectCollectors(create_collector_filter())

    LOGGER.debug(str(response))

    assert isinstance(response, CollectorList)
 
 No newline at end of file