Newer
Older
# Copyright 2021-2023 H2020 TeraFlow (https://www.teraflow-h2020.eu/)
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import copy, grpc, pytest
from common.Constants import DEFAULT_CONTEXT_UUID, DEFAULT_TOPOLOGY_UUID
from common.proto.context_pb2 import (
Context, ContextId, Device, DeviceId, Service, ServiceId, ServiceStatusEnum, ServiceTypeEnum, Topology, TopologyId)
from context.client.ContextClient import ContextClient
#from context.client.EventsCollector import EventsCollector
from .Objects import (
CONTEXT, CONTEXT_ID, DEVICE_R1, DEVICE_R1_ID, DEVICE_R1_UUID, DEVICE_R2, DEVICE_R2_ID, DEVICE_R2_UUID,
SERVICE_R1_R2, SERVICE_R1_R2_ID, SERVICE_R1_R2_UUID, TOPOLOGY, TOPOLOGY_ID)
def grpc_service(context_client_grpc : ContextClient) -> None:
# ----- Initialize the EventsCollector -----------------------------------------------------------------------------
#events_collector = EventsCollector(
# context_client_grpc, log_events_received=True,
# activate_context_collector = False, activate_topology_collector = False, activate_device_collector = False,
# activate_link_collector = False, activate_service_collector = True, activate_slice_collector = False,
# activate_connection_collector = False)
#events_collector.start()
# ----- Prepare dependencies for the test and capture related events -----------------------------------------------
response = context_client_grpc.SetContext(Context(**CONTEXT))
assert response.context_uuid.uuid == DEFAULT_CONTEXT_UUID
response = context_client_grpc.SetTopology(Topology(**TOPOLOGY))
assert response.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
assert response.topology_uuid.uuid == DEFAULT_TOPOLOGY_UUID
response = context_client_grpc.SetDevice(Device(**DEVICE_R1))
assert response.device_uuid.uuid == DEVICE_R1_UUID
response = context_client_grpc.SetDevice(Device(**DEVICE_R2))
assert response.device_uuid.uuid == DEVICE_R2_UUID
# events = events_collector.get_events(block=True, count=4)
# assert isinstance(events[0], ContextEvent)
# assert events[0].event.event_type == EventTypeEnum.EVENTTYPE_CREATE
# assert events[0].context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
# assert isinstance(events[1], TopologyEvent)
# assert events[1].event.event_type == EventTypeEnum.EVENTTYPE_CREATE
# assert events[1].topology_id.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
# assert events[1].topology_id.topology_uuid.uuid == DEFAULT_TOPOLOGY_UUID
# assert isinstance(events[2], DeviceEvent)
# assert events[2].event.event_type == EventTypeEnum.EVENTTYPE_CREATE
# assert events[2].device_id.device_uuid.uuid == DEVICE_R1_UUID
# assert isinstance(events[3], DeviceEvent)
# assert events[3].event.event_type == EventTypeEnum.EVENTTYPE_CREATE
# assert events[3].device_id.device_uuid.uuid == DEVICE_R2_UUID
# ----- Get when the object does not exist -------------------------------------------------------------------------
with pytest.raises(grpc.RpcError) as e:
context_client_grpc.GetService(ServiceId(**SERVICE_R1_R2_ID))
assert e.value.code() == grpc.StatusCode.NOT_FOUND
assert e.value.details() == 'Service({:s}) not found'.format(SERVICE_R1_R2_UUID)
# ----- List when the object does not exist ------------------------------------------------------------------------
response = context_client_grpc.GetContext(ContextId(**CONTEXT_ID))
assert len(response.topology_ids) == 1
assert len(response.service_ids) == 0
assert len(response.slice_ids) == 0
response = context_client_grpc.ListServiceIds(ContextId(**CONTEXT_ID))
assert len(response.service_ids) == 0
response = context_client_grpc.ListServices(ContextId(**CONTEXT_ID))
assert len(response.services) == 0
# ----- Create the object ------------------------------------------------------------------------------------------
with pytest.raises(grpc.RpcError) as e:
WRONG_SERVICE = copy.deepcopy(SERVICE_R1_R2)
WRONG_SERVICE['service_endpoint_ids'][0]\
['topology_id']['context_id']['context_uuid']['uuid'] = 'ca1ea172-728f-441d-972c-feeae8c9bffc'
context_client_grpc.SetService(Service(**WRONG_SERVICE))
assert e.value.code() == grpc.StatusCode.INVALID_ARGUMENT
msg = 'request.service_endpoint_ids[0].topology_id.context_id.context_uuid.uuid(ca1ea172-728f-441d-972c-feeae8c9bffc) is invalid; '\
'should be == request.service_id.context_id.context_uuid.uuid({:s})'.format(DEFAULT_CONTEXT_UUID)
assert e.value.details() == msg
response = context_client_grpc.SetService(Service(**SERVICE_R1_R2))
assert response.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
assert response.service_uuid.uuid == SERVICE_R1_R2_UUID
# ----- Check create event -----------------------------------------------------------------------------------------
#event = events_collector.get_event(block=True)
#assert isinstance(event, ServiceEvent)
#assert event.event.event_type == EventTypeEnum.EVENTTYPE_CREATE
#assert event.service_id.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
#assert event.service_id.service_uuid.uuid == SERVICE_R1_R2_UUID
# ----- Get when the object exists ---------------------------------------------------------------------------------
response = context_client_grpc.GetContext(ContextId(**CONTEXT_ID))
assert response.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
assert response.name == ''
assert len(response.topology_ids) == 1
assert len(response.service_ids) == 1
assert response.service_ids[0].context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
assert response.service_ids[0].service_uuid.uuid == SERVICE_R1_R2_UUID
assert len(response.slice_ids) == 0
response = context_client_grpc.GetService(ServiceId(**SERVICE_R1_R2_ID))
assert response.service_id.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
assert response.service_id.service_uuid.uuid == SERVICE_R1_R2_UUID
assert response.name == ''
assert response.service_type == ServiceTypeEnum.SERVICETYPE_L3NM
assert len(response.service_endpoint_ids) == 2
assert len(response.service_constraints) == 2
assert response.service_status.service_status == ServiceStatusEnum.SERVICESTATUS_PLANNED
assert len(response.service_config.config_rules) == 3
# ----- List when the object exists --------------------------------------------------------------------------------
response = context_client_grpc.ListServiceIds(ContextId(**CONTEXT_ID))
assert len(response.service_ids) == 1
assert response.service_ids[0].context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
assert response.service_ids[0].service_uuid.uuid == SERVICE_R1_R2_UUID
response = context_client_grpc.ListServices(ContextId(**CONTEXT_ID))
assert len(response.services) == 1
assert response.services[0].service_id.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
assert response.services[0].service_id.service_uuid.uuid == SERVICE_R1_R2_UUID
assert response.services[0].name == ''
assert response.service_type == ServiceTypeEnum.SERVICETYPE_L3NM
assert len(response.service_endpoint_ids) == 2
assert len(response.service_constraints) == 2
assert response.service_status.service_status == ServiceStatusEnum.SERVICESTATUS_PLANNED
assert len(response.service_config.config_rules) == 3
# ----- Update the object ------------------------------------------------------------------------------------------
new_service_name = 'svc:r1-r2'
SERVICE_UPDATED = copy.deepcopy(SERVICE_R1_R2)
SERVICE_UPDATED['name'] = new_service_name
response = context_client_grpc.SetService(Service(**SERVICE_UPDATED))
assert response.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
assert response.service_uuid.uuid == SERVICE_R1_R2_UUID
# ----- Check update event -----------------------------------------------------------------------------------------
#event = events_collector.get_event(block=True)
#assert isinstance(event, ServiceEvent)
#assert event.event.event_type == EventTypeEnum.EVENTTYPE_UPDATE
#assert event.service_id.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
#assert event.service_id.service_uuid.uuid == SERVICE_R1_R2_UUID
# ----- Get when the object is modified ----------------------------------------------------------------------------
response = context_client_grpc.GetService(ServiceId(**SERVICE_R1_R2_ID))
assert response.service_id.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
assert response.service_id.service_uuid.uuid == SERVICE_R1_R2_UUID
assert response.service_type == ServiceTypeEnum.SERVICETYPE_L3NM
assert len(response.service_endpoint_ids) == 2
assert len(response.service_constraints) == 2
assert response.service_status.service_status == ServiceStatusEnum.SERVICESTATUS_PLANNED
assert len(response.service_config.config_rules) == 3
# ----- List when the object is modified ---------------------------------------------------------------------------
response = context_client_grpc.ListServiceIds(ContextId(**CONTEXT_ID))
assert len(response.service_ids) == 1
assert response.service_ids[0].context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
assert response.service_ids[0].service_uuid.uuid == SERVICE_R1_R2_UUID
response = context_client_grpc.ListServices(ContextId(**CONTEXT_ID))
assert len(response.services) == 1
assert response.services[0].service_id.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
assert response.services[0].service_id.service_uuid.uuid == SERVICE_R1_R2_UUID
assert response.services[0].name == new_service_name
assert response.services[0].service_type == ServiceTypeEnum.SERVICETYPE_L3NM
assert len(response.services[0].service_endpoint_ids) == 2
assert len(response.services[0].service_constraints) == 2
assert response.services[0].service_status.service_status == ServiceStatusEnum.SERVICESTATUS_PLANNED
assert len(response.services[0].service_config.config_rules) == 3
# ----- Remove the object ------------------------------------------------------------------------------------------
context_client_grpc.RemoveService(ServiceId(**SERVICE_R1_R2_ID))
# ----- Check remove event -----------------------------------------------------------------------------------------
#event = events_collector.get_event(block=True)
#assert isinstance(event, ServiceEvent)
#assert event.service_id.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
#assert event.service_id.service_uuid.uuid == SERVICE_R1_R2_UUID
# ----- List after deleting the object -----------------------------------------------------------------------------
response = context_client_grpc.GetContext(ContextId(**CONTEXT_ID))
assert len(response.topology_ids) == 1
assert len(response.service_ids) == 0
assert len(response.slice_ids) == 0
response = context_client_grpc.ListServiceIds(ContextId(**CONTEXT_ID))
assert len(response.service_ids) == 0
response = context_client_grpc.ListServices(ContextId(**CONTEXT_ID))
assert len(response.services) == 0
# ----- Clean dependencies used in the test and capture related events ---------------------------------------------
context_client_grpc.RemoveDevice(DeviceId(**DEVICE_R1_ID))
context_client_grpc.RemoveDevice(DeviceId(**DEVICE_R2_ID))
context_client_grpc.RemoveTopology(TopologyId(**TOPOLOGY_ID))
context_client_grpc.RemoveContext(ContextId(**CONTEXT_ID))
#events = events_collector.get_events(block=True, count=4)
#assert isinstance(events[0], DeviceEvent)
#assert events[0].event.event_type == EventTypeEnum.EVENTTYPE_REMOVE
#assert events[0].device_id.device_uuid.uuid == DEVICE_R1_UUID
#assert isinstance(events[1], DeviceEvent)
#assert events[1].event.event_type == EventTypeEnum.EVENTTYPE_REMOVE
#assert events[1].device_id.device_uuid.uuid == DEVICE_R2_UUID
#assert isinstance(events[2], TopologyEvent)
#assert events[2].event.event_type == EventTypeEnum.EVENTTYPE_REMOVE
#assert events[2].topology_id.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
#assert events[2].topology_id.topology_uuid.uuid == DEFAULT_TOPOLOGY_UUID
#assert isinstance(events[3], ContextEvent)
#assert events[3].event.event_type == EventTypeEnum.EVENTTYPE_REMOVE
#assert events[3].context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID
# ----- Stop the EventsCollector -----------------------------------------------------------------------------------