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

Test "test_get_kafka_configs_a" is removed and "test_write_to_kafka" is added

parent a7e1ddfb
Loading
Loading
Loading
Loading
+13 −14
Original line number Diff line number Diff line
@@ -15,8 +15,9 @@
# import sys
# print (sys.path)
import logging
from .messagesBackend import create_kafka_config, create_kafka_config_a
from src.telemetry_frontend.tests.Messages import create_collector_request
from typing import Tuple
from common.proto.context_pb2 import Empty
from src.telemetry_frontend.tests.Messages import create_collector_request, create_collector_id
from src.telemetry_frontend.backend.service.KafkaProducerService import KafkaProducerService
from src.telemetry_frontend.backend.service.KafkaProducerServiceImpl import KafkaProducerServiceImpl

@@ -28,17 +29,8 @@ LOGGER = logging.getLogger(__name__)
###########################
def test_get_kafka_configs():
    LOGGER.warning('test_get_kafka_configs requesting')
    response = KafkaProducerService.generate_kafka_configs(
        create_kafka_config()
        )
    LOGGER.debug(str(response))
    assert isinstance(response, dict)

def test_get_kafka_configs_a():
    LOGGER.warning('test_get_kafka_configs_a requesting')
    response = KafkaProducerService.generate_kafka_configs(
        create_kafka_config_a('ip:port', 'ip:port', 'test_topic', 10, 3)
        )
    KafkaProducerServiceObj = KafkaProducerService()
    response = KafkaProducerServiceObj.generate_kafka_configs()
    LOGGER.debug(str(response))
    assert isinstance(response, dict)

@@ -48,4 +40,11 @@ def test_export_collector_value():
        create_collector_request('1')
    )
    LOGGER.debug(str(response))
    assert isinstance(response, str)
 No newline at end of file
    assert isinstance(response, Tuple)

def test_write_to_kafka():
    LOGGER.warning('test_write_to_kafka requesting')
    collector_value = KafkaProducerServiceImpl.export_collector_value(create_collector_request('1'))
    response = KafkaProducerServiceImpl.write_to_kafka(collector_value)      # type: ignore (don't know why warning here)
    LOGGER.debug(str(response))
    assert isinstance(response, Empty)