diff --git a/src/common/tests/MockServicerImpl_Context.py b/src/common/tests/MockServicerImpl_Context.py index 3f4af19eced89843552737ae6f8b773d3fa26a58..6d1e0b5043687f363049920cb3ea18f0a7ca2e4e 100644 --- a/src/common/tests/MockServicerImpl_Context.py +++ b/src/common/tests/MockServicerImpl_Context.py @@ -22,7 +22,7 @@ from common.proto.context_pb2 import ( Link, LinkEvent, LinkId, LinkIdList, LinkList, Service, ServiceEvent, ServiceFilter, ServiceId, ServiceIdList, ServiceList, Slice, SliceEvent, SliceFilter, SliceId, SliceIdList, SliceList, - Topology, TopologyEvent, TopologyId, TopologyIdList, TopologyList) + Topology, TopologyDetails, TopologyEvent, TopologyId, TopologyIdList, TopologyList) from common.proto.context_pb2_grpc import ContextServiceServicer from common.tests.MockMessageBroker import ( TOPIC_CONNECTION, TOPIC_CONTEXT, TOPIC_DEVICE, TOPIC_LINK, TOPIC_SERVICE, TOPIC_SLICE, TOPIC_TOPOLOGY, @@ -162,6 +162,22 @@ class MockServicerImpl_Context(ContextServiceServicer): LOGGER.info('[GetTopology] reply={:s}'.format(grpc_message_to_json_string(reply))) return reply + def GetTopologyDetails(self, request : TopologyId, context : grpc.ServicerContext) -> TopologyDetails: + LOGGER.info('[GetTopologyDetails] request={:s}'.format(grpc_message_to_json_string(request))) + container_name = 'topology[{:s}]'.format(str(request.context_id.context_uuid.uuid)) + reply = get_entry(context, self.database, container_name, request.topology_uuid.uuid) + topology_details = TopologyDetails() + topology_details.topology_id.CopyFrom(reply.topology_id) + topology_details.name = reply.name + for device_id in reply.device_ids: + device = get_entry(context, self.database, 'device', device_id.device_uuid.uuid) + topology_details.devices.append(device) + for link_id in reply.link_ids: + link = get_entry(context, self.database, 'link', link_id.link_uuid.uuid) + topology_details.links.append(link) + LOGGER.info('[GetTopologyDetails] reply={:s}'.format(grpc_message_to_json_string(reply))) + return reply + def SetTopology(self, request: Topology, context : grpc.ServicerContext) -> TopologyId: LOGGER.info('[SetTopology] request={:s}'.format(grpc_message_to_json_string(request))) container_name = 'topology[{:s}]'.format(str(request.topology_id.context_id.context_uuid.uuid))