def setUp(self): base_manager = base.BaseApplication(self.REALM, self.HOST, self.HOST_ADDR) s6a_manager = s6a_relay.S6AApplication( Mock(), self.REALM, self.HOST, self.HOST_ADDR, ) base_manager.register(s6a_manager) self._server = server.S6aServer( base_manager, s6a_manager, self.REALM, self.HOST, ) # Mock the writes to check responses self._writes = Mock() def convert_memview_to_bytes(memview): """ Deep copy the memoryview for checking later """ return self._writes(memview.tobytes()) self._transport = asyncio.Transport() self._transport.write = Mock(side_effect=convert_memview_to_bytes) # Here goes nothing.. self._server.connection_made(self._transport)
async def serve(): # noqa: WPS430 if not store.list_subscribers(): # Waiting for subscribers to be added to store await store.on_ready() if service.config['s6a_over_grpc']: logging.info('Running s6a over grpc') s6a_proxy_servicer = S6aProxyRpcServicer( processor, service.config.get('print_grpc_payload', False), ) s6a_proxy_servicer.add_to_server(service.rpc_server) else: logging.info('Running s6a over DIAMETER') base_manager = base.BaseApplication( service.config['mme_realm'], service.config['mme_host_name'], service.config['mme_host_address'], ) s6a_manager = _get_s6a_manager(service, processor) base_manager.register(s6a_manager) # Setup the Diameter/s6a MME s6a_server = service.loop.create_server( lambda: S6aServer( base_manager, s6a_manager, service.config['mme_realm'], service.config['mme_host_name'], loop=service.loop, ), service.config['host_address'], service.config['mme_port'], ) asyncio.ensure_future(s6a_server, loop=service.loop)
async def serve(): # noqa: WPS430 if not store.list_subscribers(): # Waiting for subscribers to be added to store await store.on_ready() if service.config.get('enable5g_features', service.mconfig.enable5g_features): logging.info('Cater to 5G Authentication') m5g_subs_auth_servicer = M5GAuthRpcServicer( processor, service.config.get('print_grpc_payload', False), ) m5g_subs_auth_servicer.add_to_server(service.rpc_server) if service.config.get('enable5g_features', service.mconfig.enable5g_features): logging.info('Cater to 5G SUPI Registration') m5g_suci_reg_servicer = M5GSUCIRegRpcServicer( processor, suciprofile_db_dict, service.config.get('print_grpc_payload', False), ) m5g_suci_reg_servicer.add_to_server(service.rpc_server) if service.config.get('s6a_over_grpc'): logging.info('Running s6a over grpc') s6a_proxy_servicer = S6aProxyRpcServicer( processor, service.config.get('print_grpc_payload', False), ) s6a_proxy_servicer.add_to_server(service.rpc_server) else: logging.info('Running s6a over DIAMETER') base_manager = base.BaseApplication( service.config.get('mme_realm'), service.config.get('mme_host_name'), service.config.get('mme_host_address'), ) s6a_manager = _get_s6a_manager(service, processor) base_manager.register(s6a_manager) # Setup the Diameter/s6a MME s6a_server = service.loop.create_server( lambda: S6aServer( base_manager, s6a_manager, service.config.get('mme_realm'), service.config.get('mme_host_name'), loop=service.loop, ), service.config.get('host_address'), service.config.get('mme_port'), ) asyncio.ensure_future(s6a_server, loop=service.loop)
def setUp(self): ServiceRegistry.add_service('s6a_proxy', '0.0.0.0', 0) proxy_config = { 'local_port': 1234, 'cloud_address': 'test', 'proxy_cloud_connections': True, } self._base_manager = base.BaseApplication( self.REALM, self.HOST, self.HOST_ADDR, ) self._proxy_client = Mock() self._s6a_manager = s6a_relay.S6ARelayApplication( Mock(), self.REALM, self.HOST, self.HOST_ADDR, proxy_client=self._proxy_client, ) self._server = server.S6aServer( self._base_manager, self._s6a_manager, self.REALM, self.HOST, ) self.get_proxy_config_patcher = patch.object( s6a_relay.ServiceRegistry, 'get_proxy_config', Mock(return_value=proxy_config), ) self.mock_get_proxy_config = self.get_proxy_config_patcher.start() self.addCleanup(self.get_proxy_config_patcher.stop) # Mock the writes to check responses self._writes = Mock() def convert_memview_to_bytes(memview): """ Deep copy the memoryview for checking later """ return self._writes(memview.tobytes()) self._transport = MockTransport() self._transport.write = Mock(side_effect=convert_memview_to_bytes) # Here goes nothing.. self._server.connection_made(self._transport)