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)
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)
def setUp(self): self._server = server.S6aServer(Mock(), Mock(), "mai.facebook.com", "hss.mai.facebook.com") # Mock the message handler self._server._handle_msg = Mock() # 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)