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

src/tests/ofc26_flexscale/test_ofc26_mgon_integration_V2.py

+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

    )