def test1(self):
hook(PITA, 'asyncMethod')
pita = PITA()
pita.synchronousMethod(1)
call = yield pita.asyncMethod.called()
with ShouldRaise(Exception(1)):
yield call.result
def setUp(self):
context = Context()
server = makeTCPServer(context, ServerProtocol)
client = yield makeTCPClient(context, EchoClient, server)
hook(EchoClient, 'lineReceived')
self.client = client.clientProtocol
self.server = client.serverProtocol
self.addCleanup(context.cleanup)
def setUp(self):
context = Context()
server = makeTCPServer(context, MyServerProtocol,
WebSocketServerFactory())
client = yield makeWebSocketClient(context, server)
hook(WebSocketClientProtocol,
'onMessage',
decoder=lambda payload, _: payload)
self.client = client.clientProtocol
self.addCleanup(context.cleanup)
def setUp(self):
context = Context()
server = makeTCPServer(context, WebSocketServerProtocol,
WebSocketServerFactory())
hook(WebSocketServerProtocol,
'onMessage',
decoder=lambda payload, _: payload)
client = yield makeWebSocketClient(context, server, MyClientProtocol)
self.server = client.serverProtocol
# cancel the loop:
self.addCleanup(context.cleanup, delayedCalls=1)
def testUnconsumedCalls(self):
hook(Sample, 'method')
obj = Sample()
obj.method('foo')
with ShouldRaise(
UnconsumedCalls({
(Sample, 'method'): {
obj: (Call(protocol=obj,
args=('foo', ),
kw={},
result='foo'), )
}
})):
cleanup()
def setUp(self):
context = self.context = Context()
self.dir = TempDirectory()
self.addCleanup(self.dir.cleanup)
# make sure that we trap unexpected close messages:
hook(ServerProtocol, 'sendClose')
self.server = makeTCPServer(context,
ServerProtocol,
buildSite(b'gotit', self.dir.path),
installProtocol=False)
client = yield makeWebSocketClient(context,
self.server,
factoryClass=ServerTesterFactory,
close='close',
endpoint='/spam')
self.client = client.clientProtocol
self.addCleanup(context.cleanup, threads=True, delayedCalls=1)
def testReconnect(self):
hook(ServerTester, 'onOpen')
self.client.sendMessage(b'hello', isBinary=True)
yield ServerProtocol.sendClose.called()
yield self.client.connectionLost.called()
client = yield waitForClient(self.context, ServerTester.onOpen,
ServerProtocol.connectionMade, 'close')
server = client.serverProtocol
client = client.clientProtocol
client.sendMessage(b'hello')
payload = yield client.onMessage.called()
compare(payload, expected=b'gotit')
# autobahn just doesn't use client protocol instances that aren't open, so
# we manually check we're stripped out closed connections:
compare(server.factory.state.clients, expected={server})
yield advanceTime(seconds=2.1)
payload = yield client.onMessage.called()
compare(payload, expected=b'ping 1')