def test_create_session(self, grpc_server: CoreGrpcServer, session_id: Optional[int]): # given client = CoreGrpcClient() # when with client.context_connect(): created_session = client.create_session(session_id) # then assert isinstance(created_session, wrappers.Session) session = grpc_server.coreemu.sessions.get(created_session.id) assert session is not None if session_id is not None: assert created_session.id == session_id assert session.id == session_id
def test_create_session(self, grpc_server, session_id): # given client = CoreGrpcClient() # when with client.context_connect(): response = client.create_session(session_id) # then assert isinstance(response.session_id, int) assert isinstance(response.state, int) session = grpc_server.coreemu.sessions.get(response.session_id) assert session is not None assert session.state == response.state if session_id is not None: assert response.session_id == session_id assert session.id == session_id
def test_start_session(self, grpc_server: CoreGrpcServer, definition): # given client = CoreGrpcClient() with client.context_connect(): session = client.create_session() position = Position(x=50, y=100) node1 = session.add_node(1, position=position) position = Position(x=100, y=100) node2 = session.add_node(2, position=position) position = Position(x=200, y=200) wlan_node = session.add_node(3, _type=NodeType.WIRELESS_LAN, position=position) iface_helper = InterfaceHelper(ip4_prefix="10.83.0.0/16") iface1_id = 0 iface1 = iface_helper.create_iface(node1.id, iface1_id) iface2_id = 0 iface2 = iface_helper.create_iface(node2.id, iface2_id) link = Link(node1_id=node1.id, node2_id=node2.id, iface1=iface1, iface2=iface2) session.links = [link] hook = Hook(state=SessionState.RUNTIME, file="echo.sh", data="echo hello") session.hooks = {hook.file: hook} location_x = 5 location_y = 10 location_z = 15 location_lat = 20 location_lon = 30 location_alt = 40 location_scale = 5 session.location = SessionLocation( x=location_x, y=location_y, z=location_z, lat=location_lat, lon=location_lon, alt=location_alt, scale=location_scale, ) # setup wlan config wlan_config_key = "range" wlan_config_value = "333" wlan_node.set_wlan({wlan_config_key: wlan_config_value}) # setup mobility config mobility_config_key = "refresh_ms" mobility_config_value = "60" wlan_node.set_mobility({mobility_config_key: mobility_config_value}) # setup service config service_name = "DefaultRoute" service_validate = ["echo hello"] node1.service_configs[service_name] = NodeServiceData( executables=[], dependencies=[], dirs=[], configs=[], startup=[], validate=service_validate, validation_mode=ServiceValidationMode.NON_BLOCKING, validation_timer=0, shutdown=[], meta="", ) # setup service file config service_file = "defaultroute.sh" service_file_data = "echo hello" node1.service_file_configs[service_name] = { service_file: service_file_data } # setup session option option_key = "controlnet" option_value = "172.16.0.0/24" session.set_options({option_key: option_value}) # when with patch.object(CoreXmlWriter, "write"): with client.context_connect(): client.start_session(session, definition=definition) # then real_session = grpc_server.coreemu.sessions[session.id] if definition: state = EventTypes.DEFINITION_STATE else: state = EventTypes.RUNTIME_STATE assert real_session.state == state assert node1.id in real_session.nodes assert node2.id in real_session.nodes assert wlan_node.id in real_session.nodes assert iface1_id in real_session.nodes[node1.id].ifaces assert iface2_id in real_session.nodes[node2.id].ifaces hook_file, hook_data = real_session.hooks[EventTypes.RUNTIME_STATE][0] assert hook_file == hook.file assert hook_data == hook.data assert real_session.location.refxyz == (location_x, location_y, location_z) assert real_session.location.refgeo == ( location_lat, location_lon, location_alt, ) assert real_session.location.refscale == location_scale set_wlan_config = real_session.mobility.get_model_config( wlan_node.id, BasicRangeModel.name) assert set_wlan_config[wlan_config_key] == wlan_config_value set_mobility_config = real_session.mobility.get_model_config( wlan_node.id, Ns2ScriptedMobility.name) assert set_mobility_config[ mobility_config_key] == mobility_config_value service = real_session.services.get_service(node1.id, service_name, default_service=True) assert service.validate == tuple(service_validate) real_node1 = real_session.get_node(node1.id, CoreNode) service_file = real_session.services.get_service_file( real_node1, service_name, service_file) assert service_file.data == service_file_data assert option_value == real_session.options.get_config(option_key)