diff --git a/src/dlt/connector/main_test.py b/src/dlt/connector/main_test.py index a5ad9c33c0c1d7f218a01e2b83d4c1e6dfc6df6a..6324972200c8e21943bf58a4082e4efbdff6e6db 100644 --- a/src/dlt/connector/main_test.py +++ b/src/dlt/connector/main_test.py @@ -1,15 +1,43 @@ -import sys +# pip install grpcio==1.47.0 grpcio-tools==1.47.0 protobuf==3.20.1 +# PYTHONPATH=/home/cttc/teraflow/src python -m dlt.connector.main_test + +import logging, sys, time +from common.proto.dlt_gateway_pb2 import DLTRECORDOPERATION_ADD, DLTRECORDTYPE_DEVICE, DltRecord +from common.tools.object_factory.Device import json_device +from common.tools.grpc.Tools import grpc_message_to_json_string +from src.common.proto.context_pb2 import DEVICEOPERATIONALSTATUS_ENABLED, Device from .client.DltGatewayClient import DltGatewayClient from .client.DltEventsCollector import DltEventsCollector +logging.basicConfig(level=logging.INFO) +LOGGER = logging.getLogger(__name__) + def main(): - dltgateway_client_1 = DltGatewayClient(host='', port=0) - dltgateway_client_2 = DltGatewayClient(host='', port=0) - dltgateway_client_3 = DltGatewayClient(host='', port=0) + dltgateway_client = DltGatewayClient(host='127.0.0.1', port=50051) + dltgateway_collector = DltEventsCollector(dltgateway_client, log_events_received=True) + dltgateway_collector.start() + + time.sleep(3) + + device = Device(**json_device('dev-1', 'packet-router', DEVICEOPERATIONALSTATUS_ENABLED)) + + r2dlt_req = DltRecord() + r2dlt_req.record_id.domain_uuid.uuid = 'tfs-a' + r2dlt_req.record_id.type = DLTRECORDTYPE_DEVICE + r2dlt_req.record_id.record_uuid.uuid = device.device_id.device_uuid.uuid + r2dlt_req.operation = DLTRECORDOPERATION_ADD + r2dlt_req.data_json = grpc_message_to_json_string(device) + LOGGER.info('r2dlt_req = {:s}'.format(grpc_message_to_json_string(r2dlt_req))) + r2dlt_rep = dltgateway_client.RecordToDlt(r2dlt_req) + LOGGER.info('r2dlt_rep = {:s}'.format(grpc_message_to_json_string(r2dlt_rep))) + + dlt2r_req = r2dlt_req.record_id + LOGGER.info('dlt2r_req = {:s}'.format(grpc_message_to_json_string(dlt2r_req))) + dlt2r_rep = dltgateway_client.RecordToDlt(dlt2r_req) + LOGGER.info('dlt2r_rep = {:s}'.format(grpc_message_to_json_string(dlt2r_rep))) - dltgateway_collector_1 = DltEventsCollector(dltgateway_client_1, log_events_received=True) - dltgateway_collector_2 = DltEventsCollector(dltgateway_client_2, log_events_received=True) - dltgateway_collector_3 = DltEventsCollector(dltgateway_client_3, log_events_received=True) + dltgateway_collector.stop() + return 0 if __name__ == '__main__': sys.exit(main())