add twisted/blocking interop test
This commit is contained in:
parent
fd9a62e8ff
commit
9827a2e50c
64
src/wormhole/test/test_interop.py
Normal file
64
src/wormhole/test/test_interop.py
Normal file
|
@ -0,0 +1,64 @@
|
|||
from __future__ import print_function
|
||||
import sys
|
||||
from twisted.trial import unittest
|
||||
from twisted.internet.defer import gatherResults
|
||||
from twisted.internet.threads import deferToThread
|
||||
from ..twisted.transcribe import Wormhole as twisted_Wormhole
|
||||
from ..blocking.transcribe import Wormhole as blocking_Wormhole
|
||||
from .common import ServerBase
|
||||
|
||||
# make sure the two implementations (Twisted-style and blocking-style) can
|
||||
# interoperate
|
||||
|
||||
APPID = u"appid"
|
||||
|
||||
class Basic(ServerBase, unittest.TestCase):
|
||||
|
||||
def doBoth(self, call1, d2):
|
||||
f1 = call1[0]
|
||||
f1args = call1[1:]
|
||||
return gatherResults([deferToThread(f1, *f1args), d2], True)
|
||||
|
||||
def test_twisted_to_blocking(self):
|
||||
tw = twisted_Wormhole(APPID, self.relayurl)
|
||||
bw = blocking_Wormhole(APPID, self.relayurl)
|
||||
d = tw.get_code()
|
||||
def _got_code(code):
|
||||
bw.set_code(code)
|
||||
return self.doBoth([bw.send_data, b"data2"], tw.send_data(b"data1"))
|
||||
d.addCallback(_got_code)
|
||||
def _sent(res):
|
||||
return self.doBoth([bw.get_data], tw.get_data())
|
||||
d.addCallback(_sent)
|
||||
def _done(dl):
|
||||
(dataX, dataY) = dl
|
||||
self.assertEqual(dataX, b"data1")
|
||||
self.assertEqual(dataY, b"data2")
|
||||
return self.doBoth([bw.close], tw.close())
|
||||
d.addCallback(_done)
|
||||
return d
|
||||
|
||||
def test_blocking_to_twisted(self):
|
||||
bw = blocking_Wormhole(APPID, self.relayurl)
|
||||
tw = twisted_Wormhole(APPID, self.relayurl)
|
||||
d = deferToThread(bw.get_code)
|
||||
def _got_code(code):
|
||||
tw.set_code(code)
|
||||
return self.doBoth([bw.send_data, b"data1"], tw.send_data(b"data2"))
|
||||
d.addCallback(_got_code)
|
||||
def _sent(res):
|
||||
return self.doBoth([bw.get_data], tw.get_data())
|
||||
d.addCallback(_sent)
|
||||
def _done(dl):
|
||||
(dataX, dataY) = dl
|
||||
self.assertEqual(dataX, b"data2")
|
||||
self.assertEqual(dataY, b"data1")
|
||||
return self.doBoth([bw.close], tw.close())
|
||||
d.addCallback(_done)
|
||||
return d
|
||||
|
||||
if sys.version_info[0] >= 3:
|
||||
Basic.skip = "twisted is not yet sufficiently ported to py3"
|
||||
# as of 15.4.0, Twisted is still missing:
|
||||
# * web.client.Agent (for all non-EventSource POSTs in transcribe.py)
|
||||
# * python.logfile (to allow daemonization of 'wormhole server')
|
Loading…
Reference in New Issue
Block a user