def test_secure_echo_server(self): class EchoServer(WebSocketServerProtocol): def onMessage(self, msg, is_binary): self.sendMessage(msg) agent = create_memory_agent(self.reactor, self.pumper, EchoServer) proto = yield agent.open(u"wss://localhost:1234/ws", dict()) messages = [] def got(msg, is_binary): messages.append(msg) proto.on("message", got) proto.sendMessage(b"hello") if True: # clean close proto.sendClose() else: # unclean close proto.transport.loseConnection() yield proto.is_closed self.assertEqual([b"hello"], messages)
def setUp(self): self.reactor = MemoryReactorClockResolver() self.pumper = create_pumper() self.agent = create_memory_agent( self.reactor, self.pumper, TokenAuthenticatedWebSocketServerProtocol) return self.pumper.start()
def setUp(self): super(WebSocketTests, self).setUp() self.reactor = MemoryReactorClockResolver() self.pumper = create_pumper() self.service = WebSocketStatusService() self.factory = StatusFactory(self.service) self.agent = create_memory_agent( self.reactor, self.pumper, lambda: self.factory.buildProtocol(None)) return self.pumper.start()
def setUp(self): super(TestApiMonitor, self).setUp() self.magic_config = FilePath(self.mktemp()) self.global_config = create_testing_configuration( self.magic_config, FilePath(u"/no/tahoe/node-directory"), ) self.reactor = MemoryReactorClockResolver() self.pumper = create_pumper() self.service = WebSocketStatusService( self.reactor, self.global_config, ) self.factory = StatusFactory(self.service) self.agent = create_memory_agent( self.reactor, self.pumper, lambda: self.factory.buildProtocol(None) ) return self.pumper.start()
def setUp(self): super(WebSocketTests, self).setUp() self.reactor = MemoryReactorClockResolver() self.pumper = create_pumper() self.tahoe_node_dir = FilePath(self.mktemp()) self.tahoe_node_dir.makedirs() self.global_config = create_testing_configuration( FilePath(self.mktemp()), self.tahoe_node_dir, ) self.service = WebSocketStatusService( self.reactor, self.global_config, ) self.factory = StatusFactory(self.service) self.agent = create_memory_agent( self.reactor, self.pumper, lambda: self.factory.buildProtocol(None)) return self.pumper.start()