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 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):
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()
