Commit 362464e1 authored by Andrea Sgambelluri's avatar Andrea Sgambelluri
Browse files

refactor test_Complete_MGON_Integration to enhance telemetry service...

refactor test_Complete_MGON_Integration to enhance telemetry service initialization and collector handling
parent de9ecf8e
Loading
Loading
Loading
Loading
+12 −3
Original line number Diff line number Diff line
@@ -10,10 +10,11 @@ import threading
from common.proto import kpi_manager_pb2
from common.proto.kpi_sample_types_pb2 import KpiSampleType

from src.telemetry.backend.service.collector_api import DriverFactory
from tests.ofc26_flexscale.test_ofc26_messages import create_kpi_descriptor_request, create_collector_request
from src.tests.ofc26_flexscale.test_ofc26_messages import create_basic_sub_request_parameters

from src.telemetry.backend.service.TelemetryBackendService import TelemetryBackendService
from src.telemetry.backend.service.TelemetryBackendService import DriverInstanceCache, TelemetryBackendService


WITH_TFS = True     #True/False
@@ -83,6 +84,14 @@ def test_Complete_MGON_Integration(kpi_manager_client, telemetry_frontend_client
    #     assert isinstance(response, CollectorId)
    
    # step 2: Telemetry Collector backup option
    from telemetry.backend.service.collectors import COLLECTORS
    from telemetry.backend.service.collector_api.DriverFactory import DriverFactory
    from telemetry.backend.service.collector_api.DriverInstanceCache import DriverInstanceCache, preload_drivers
    
    driver_factory        = DriverFactory(COLLECTORS)
    driver_instance_cache = DriverInstanceCache(driver_factory)
    _service              = TelemetryBackendService(driver_instance_cache)
    
    _collector_request = create_collector_request()
    _collector         = create_basic_sub_request_parameters()
    _coll_id           = "mgon_collector_id"
@@ -93,10 +102,10 @@ def test_Complete_MGON_Integration(kpi_manager_client, telemetry_frontend_client
    
    stop_event       = threading.Event()
    collector_thread = threading.Thread(
        target=TelemetryBackendService.GenericCollectorHandler,
        target=_service.GenericCollectorHandler,
        args=(
            _coll_id, _collector, None, _duration, _interval,
            None, None, None, None, stop_event
            None, None, None, "43813baf-195e-5da6-af20-b3d0922e71a7", stop_event
        ),
        daemon=False
    )