def test_reqrep(self):
def do_test(s0, s1):
f = tests.spawn(s0.send, b'hello world')
d = s1.recv()
f.result()
self.assertEqual(d, b'hello world')
f = tests.spawn(s1.send, b'spam egg')
d = s0.recv()
f.result()
self.assertEqual(d, b'spam egg')
with contextlib.ExitStack() as stack:
url = 'inproc://%s' % uuid.uuid4()
sock1 = stack.enter_context(sockets.Socket(nng.Protocols.REP0))
sock1.listen(url)
sock0 = stack.enter_context(sockets.Socket(nng.Protocols.REQ0))
sock0.dial(url)
with self.subTest((sock0, sock1)):
do_test(sock0, sock1)
c0 = stack.enter_context(sockets.Context(sock0))
c1 = stack.enter_context(sockets.Context(sock1))
with self.subTest((c0, c1)):
do_test(c0, c1)
def test_reqrep0(self):
with contextlib.ExitStack() as stack:
url = 'inproc://%s' % uuid.uuid4()
s1 = stack.enter_context(sockets.Socket(nng.Protocols.REP0))
s2 = stack.enter_context(sockets.Socket(nng.Protocols.REQ0))
s1.listen(url)
s2.dial(url)
_send(s2, b'hello world')
self.assertEqual(_recv(s1), b'hello world')
_send(s1, b'foo bar')
self.assertEqual(_recv(s2), b'foo bar')
def test_bus0(self):
with contextlib.ExitStack() as stack:
url = 'inproc://%s' % uuid.uuid4()
s1 = stack.enter_context(sockets.Socket(nng.Protocols.BUS0))
s2 = stack.enter_context(sockets.Socket(nng.Protocols.BUS0))
s3 = stack.enter_context(sockets.Socket(nng.Protocols.BUS0))
s1.listen(url)
s2.dial(url)
s3.dial(url)
_send(s1, b'hello world')
self.assertEqual(_recv(s2), b'hello world')
self.assertEqual(_recv(s3), b'hello world')
def test_survey0(self):
with contextlib.ExitStack() as stack:
url = 'inproc://%s' % uuid.uuid4()
s1 = stack.enter_context(sockets.Socket(nng.Protocols.SURVEYOR0))
s2 = stack.enter_context(sockets.Socket(nng.Protocols.RESPONDENT0))
s3 = stack.enter_context(sockets.Socket(nng.Protocols.RESPONDENT0))
s1.listen(url)
s2.dial(url)
s3.dial(url)
_send(s1, b'hello world')
self.assertEqual(_recv(s2), b'hello world')
self.assertEqual(_recv(s3), b'hello world')
_send(s2, b'spam')
_send(s3, b'egg')
self.assertEqual(sorted([_recv(s1), _recv(s1)]), [b'egg', b'spam'])
def test_dialer_start(self):
with contextlib.ExitStack() as stack:
url = 'inproc://%s' % uuid.uuid4()
s1 = stack.enter_context(sockets.Socket(nng.Protocols.REP0))
s1.listen(url)
s0 = stack.enter_context(sockets.Socket(nng.Protocols.REQ0))
d = s0.dial(url, create_only=True)
d.start()
f = tests.spawn(s0.send, b'hello world')
d = s1.recv()
f.result()
self.assertEqual(d, b'hello world')
d = s0.dial('inproc://%s' % uuid.uuid4())
with self.assertRaises(nng.Errors.ESTATE):
d.start()
def test_message(self):
def do_test(s0, s1):
m0 = nng.Message(b'hello world')
self.assertEqual(m0.header.memory_view, b'')
self.assertEqual(m0.body.memory_view, b'hello world')
f = tests.spawn(s0.sendmsg, m0)
m1 = s1.recvmsg()
f.result()
del m0
self.assertEqual(m1.header.memory_view, b'')
self.assertEqual(m1.body.memory_view, b'hello world')
m1.body.memory_view[:] = bytes(reversed(b'hello world'))
self.assertEqual(m1.header.memory_view, b'')
self.assertEqual(m1.body.memory_view, b'dlrow olleh')
f = tests.spawn(s1.sendmsg, m1)
m2 = s0.recvmsg()
f.result()
# Ownership is transferred on success.
self.assertIsNone(m1._msg_p)
self.assertEqual(m2.body.memory_view, b'dlrow olleh')
with contextlib.ExitStack() as stack:
url = 'inproc://%s' % uuid.uuid4()
sock1 = stack.enter_context(sockets.Socket(nng.Protocols.REP0))
sock1.listen(url)
sock0 = stack.enter_context(sockets.Socket(nng.Protocols.REQ0))
sock0.dial(url)
with self.subTest((sock0, sock1)):
do_test(sock0, sock1)
c0 = stack.enter_context(sockets.Context(sock0))
c1 = stack.enter_context(sockets.Context(sock1))
with self.subTest((c0, c1)):
do_test(c0, c1)
def test_reqrep_incorrect_sequence(self):
def do_test(s0, s1):
with self.assertRaises(nng.Errors.ESTATE):
s0.recv()
with self.assertRaises(nng.Errors.ESTATE):
s1.send(b'')
with contextlib.ExitStack() as stack:
url = 'inproc://%s' % uuid.uuid4()
sock1 = stack.enter_context(sockets.Socket(nng.Protocols.REP0))
sock1.listen(url)
sock0 = stack.enter_context(sockets.Socket(nng.Protocols.REQ0))
sock0.dial(url)
with self.subTest((sock0, sock1)):
do_test(sock0, sock1)
c0 = stack.enter_context(sockets.Context(sock0))
c1 = stack.enter_context(sockets.Context(sock1))
with self.subTest((c0, c1)):
do_test(c0, c1)
def test_del_not_resurrecting(self):
tests.assert_del_not_resurrecting(
self, lambda: sockets.Socket(nng.Protocols.REP0))
with sockets.Socket(nng.Protocols.REP0) as s:
tests.assert_del_not_resurrecting(self, lambda: sockets.Context(s))
