From ed77dac801e0e6e859980c92b3c52ca67eb22092 Mon Sep 17 00:00:00 2001 From: Waleed Akbar Date: Tue, 14 May 2024 13:57:49 +0000 Subject: [PATCH] Update the messages format --- src/telemetry/frontend/tests/Messages.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/src/telemetry/frontend/tests/Messages.py b/src/telemetry/frontend/tests/Messages.py index 9e7b6c049..93a6066ee 100644 --- a/src/telemetry/frontend/tests/Messages.py +++ b/src/telemetry/frontend/tests/Messages.py @@ -12,20 +12,27 @@ # See the License for the specific language governing permissions and # limitations under the License. +import uuid +import random from common.proto import telemetry_frontend_pb2 from common.proto.kpi_sample_types_pb2 import KpiSampleType +def create_collector_id(): + _collector_id = telemetry_frontend_pb2.CollectorId() + _collector_id.collector_id.uuid = uuid.uuid4() + return _collector_id + def create_collector_id(coll_id_str : str): _collector_id = telemetry_frontend_pb2.CollectorId() _collector_id.collector_id.uuid = str(coll_id_str) return _collector_id -def create_collector_request(coll_id_str : str): +def create_collector_request(): _create_collector_request = telemetry_frontend_pb2.Collector() - _create_collector_request.collector_id.collector_id.uuid = str(coll_id_str) - _create_collector_request.kpi_id.kpi_id.uuid = 'KPIid' + str(coll_id_str) - _create_collector_request.duration_s = float(-1) - _create_collector_request.interval_s = float(-1) + _create_collector_request.collector_id.collector_id.uuid = str(uuid.uuid4()) + _create_collector_request.kpi_id.kpi_id.uuid = str(uuid.uuid4()) + _create_collector_request.duration_s = float(random.randint(8, 16)) + _create_collector_request.interval_s = float(random.randint(2, 3)) return _create_collector_request def create_collector_request_a(): -- GitLab