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)
