Commit 7ea968e5 authored by Lluis Gifre Renom's avatar Lluis Gifre Renom
Browse files

Common - Mock Context Servicer: 

- corrected SetContext
- corrected SetSlice/RemoveSlice
- implemented GetSliceEvents
- corrected RemoveService
parent 6d62d7f4

src/common/tests/MockServicerImpl_Context.py

+8 −6
Original line number Diff line number Diff line
@@ -115,7 +115,7 @@ class MockServicerImpl_Context(ContextServiceServicer): 


    def SetContext(self, request: Context, context : grpc.ServicerContext) -> ContextId:

        LOGGER.info('[SetContext] request={:s}'.format(grpc_message_to_json_string(request)))

        return self._set(request, 'context', request.context_uuid.uuid, 'context_id', TOPIC_CONTEXT)

        return self._set(request, 'context', request.context_id.context_uuid.uuid, 'context_id', TOPIC_CONTEXT)



    def RemoveContext(self, request: ContextId, context : grpc.ServicerContext) -> Empty:

        LOGGER.info('[RemoveContext] request={:s}'.format(grpc_message_to_json_string(request)))
@@ -233,17 +233,19 @@ class MockServicerImpl_Context(ContextServiceServicer): 


    def SetSlice(self, request: Slice, context : grpc.ServicerContext) -> SliceId:

        LOGGER.info('[SetSlice] request={:s}'.format(grpc_message_to_json_string(request)))

        return set_entry(

            self.database, 'slice[{:s}]'.format(str(request.slice_id.context_id.context_uuid.uuid)),

            request.slice_id.slice_uuid.uuid, request).slice_id

        container_name = 'slice[{:s}]'.format(str(request.slice_id.context_id.context_uuid.uuid))

        slice_uuid = request.slice_id.slice_uuid.uuid

        return self._set(request, container_name, slice_uuid, 'slice_id', TOPIC_SLICE)



    def RemoveSlice(self, request: SliceId, context : grpc.ServicerContext) -> Empty:

        LOGGER.info('[RemoveSlice] request={:s}'.format(grpc_message_to_json_string(request)))

        container_name = 'slice[{:s}]'.format(str(request.context_id.context_uuid.uuid))

        return del_entry(context, self.database, container_name, request.slice_uuid.uuid)

        slice_uuid = request.slice_uuid.uuid

        return self._del(request, container_name, slice_uuid, 'slice_id', TOPIC_SLICE, context)



    def GetSliceEvents(self, request: Empty, context : grpc.ServicerContext) -> Iterator[SliceEvent]:

        LOGGER.info('[GetSliceEvents] request={:s}'.format(grpc_message_to_json_string(request)))

        for message in self.msg_broker.consume({TOPIC_SLICE}): yield SliceEvent(**json.loads(message.content))





    # ----- Service ----------------------------------------------------------------------------------------------------
@@ -272,7 +274,7 @@ class MockServicerImpl_Context(ContextServiceServicer): 
    def RemoveService(self, request: ServiceId, context : grpc.ServicerContext) -> Empty:

        LOGGER.info('[RemoveService] request={:s}'.format(grpc_message_to_json_string(request)))

        container_name = 'service[{:s}]'.format(str(request.context_id.context_uuid.uuid))

        service_uuid = request.service_id.service_uuid.uuid

        service_uuid = request.service_uuid.uuid

        return self._del(request, container_name, service_uuid, 'service_id', TOPIC_SERVICE, context)



    def GetServiceEvents(self, request: Empty, context : grpc.ServicerContext) -> Iterator[ServiceEvent]: