def test_zmq_sleep(self):
from zmq.cffi_core._cffi import C
try:
C.zmq_sleep(1)
except Exception as e:
raise AssertionError("Error executing zmq_sleep(int)")
def test_zmq_ctx_new_destroy(self):
from zmq.cffi_core._cffi import ffi, C
ctx = C.zmq_ctx_new()
assert ctx != ffi.NULL
assert 0 == C.zmq_ctx_destroy(ctx)
def test_zmq_stopwatch_functions(self):
from zmq.cffi_core._cffi import C, ffi
stopwatch = C.zmq_stopwatch_start()
ret = C.zmq_stopwatch_stop(stopwatch)
assert ffi.NULL != stopwatch
assert 0 < int(ret)
def test_zmq_msg_init_size(self):
from zmq.cffi_core._cffi import C, ffi
zmq_msg = ffi.new('zmq_msg_t*')
assert ffi.NULL != zmq_msg
assert 0 == C.zmq_msg_init_size(zmq_msg, 10)
assert 0 == C.zmq_msg_close(zmq_msg)
def test_zmq_socket_open_close(self):
from zmq.cffi_core._cffi import ffi, C
from zmq.cffi_core.constants import PUSH
ctx = C.zmq_ctx_new()
socket = C.zmq_socket(ctx, PUSH)
assert ctx != ffi.NULL
assert ffi.NULL != socket
assert 0 == C.zmq_close(socket)
assert 0 == C.zmq_ctx_destroy(ctx)
def test_zmq_bind(self):
from zmq.cffi_core._cffi import C, ffi
from zmq.cffi_core.constants import PUSH
ctx = C.zmq_ctx_new()
socket = C.zmq_socket(ctx, 8)
assert 0 == C.zmq_bind(socket, 'tcp://*:4444')
assert ctx != ffi.NULL
assert ffi.NULL != socket
assert 0 == C.zmq_close(socket)
assert 0 == C.zmq_ctx_destroy(ctx)
def test_zmq_msg_init_data(self):
from zmq.cffi_core._cffi import C, ffi
zmq_msg = ffi.new('zmq_msg_t*')
message = ffi.new('char[5]', 'Hello')
assert 0 == C.zmq_msg_init_data(zmq_msg,
ffi.cast('void*', message),
5,
ffi.NULL,
ffi.NULL)
assert ffi.NULL != zmq_msg
assert 0 == C.zmq_msg_close(zmq_msg)
def test_zmq_setsockopt(self):
from zmq.cffi_core._cffi import C, ffi
from zmq.cffi_core.constants import PUSH, IDENTITY
ctx = C.zmq_ctx_new()
socket = C.zmq_socket(ctx, PUSH)
identity = ffi.new('char[3]', 'zmq')
ret = C.zmq_setsockopt(socket, IDENTITY, ffi.cast('void*', identity), 3)
assert ret == 0
assert ctx != ffi.NULL
assert ffi.NULL != socket
assert 0 == C.zmq_close(socket)
assert 0 == C.zmq_ctx_destroy(ctx)
def stop(self):
if self.watch == ffi.NULL:
raise ZMQError('Must start the Stopwatch before calling stop.')
else:
time = C.zmq_stopwatch_stop(self.watch)
self.watch = ffi.NULL
return time
def stop(self):
if self.watch == ffi.NULL:
raise ZMQError('Must start the Stopwatch before calling stop.')
else:
time = C.zmq_stopwatch_stop(self.watch)
self.watch = ffi.NULL
return time
def test_zmq_getsockopt(self):
from zmq.cffi_core._cffi import C, ffi
from zmq.cffi_core.constants import PUSH, IDENTITY
ctx = C.zmq_ctx_new()
socket = C.zmq_socket(ctx, PUSH)
identity = ffi.new('char[]', 'zmq')
ret = C.zmq_setsockopt(socket, IDENTITY, ffi.cast('void*', identity), 3)
assert ret == 0
option_len = ffi.new('size_t*', 3)
option = ffi.new('char*')
ret = C.zmq_getsockopt(socket,
IDENTITY,
ffi.cast('void*', option),
option_len)
assert ret == 0
assert ffi.string(ffi.cast('char*', option))[0] == "z"
assert ffi.string(ffi.cast('char*', option))[1] == "m"
assert ffi.string(ffi.cast('char*', option))[2] == "q"
assert ctx != ffi.NULL
assert ffi.NULL != socket
assert 0 == C.zmq_close(socket)
assert 0 == C.zmq_ctx_destroy(ctx)
def test_zmq_bind_connect(self):
from zmq.cffi_core._cffi import C, ffi
from zmq.cffi_core.constants import PUSH, PULL
ctx = C.zmq_ctx_new()
socket1 = C.zmq_socket(ctx, PUSH)
socket2 = C.zmq_socket(ctx, PULL)
assert 0 == C.zmq_bind(socket1, 'tcp://*:4444')
assert 0 == C.zmq_connect(socket2, 'tcp://127.0.0.1:4444')
assert ctx != ffi.NULL
assert ffi.NULL != socket1
assert ffi.NULL != socket2
assert 0 == C.zmq_close(socket1)
assert 0 == C.zmq_close(socket2)
assert 0 == C.zmq_ctx_destroy(ctx)
def sleep(self, seconds):
C.zmq_sleep(seconds)
def test_zmq_recv(self):
from zmq.cffi_core.constants import REQ, REP
from zmq.cffi_core._cffi import C, ffi
ctx = C.zmq_ctx_new()
sender = C.zmq_socket(ctx, REQ)
receiver = C.zmq_socket(ctx, REP)
assert 0 == C.zmq_bind(receiver, 'tcp://*:2222')
assert 0 == C.zmq_connect(sender, 'tcp://127.0.0.1:2222')
import time
time.sleep(0.1)
zmq_msg = ffi.new('zmq_msg_t*')
message = ffi.new('char[5]', 'Hello')
C.zmq_msg_init_data(zmq_msg,
ffi.cast('void*', message),
ffi.cast('size_t', 5),
ffi.NULL,
ffi.NULL)
zmq_msg2 = ffi.new('zmq_msg_t*')
C.zmq_msg_init(zmq_msg2)
assert 5 == C.zmq_msg_send(zmq_msg, sender, 0)
assert 5 == C.zmq_msg_recv(zmq_msg2, receiver, 0)
assert 5 == C.zmq_msg_size(zmq_msg2)
assert b"Hello" == ffi.buffer(C.zmq_msg_data(zmq_msg2),
C.zmq_msg_size(zmq_msg2))[:]
assert C.zmq_close(sender) == 0
assert C.zmq_close(receiver) == 0
assert C.zmq_ctx_destroy(ctx) == 0
def start(self):
if self.watch == ffi.NULL:
self.watch = C.zmq_stopwatch_start()
else:
raise ZMQError('Stopwatch is already runing.')
def sleep(self, seconds):
C.zmq_sleep(seconds)
def test_zmq_poll(self):
from zmq.cffi_core.constants import REQ, REP, NOBLOCK, POLLIN, POLLOUT
from zmq.cffi_core._cffi import C, ffi
ctx = C.zmq_ctx_new()
sender = C.zmq_socket(ctx, REQ)
receiver = C.zmq_socket(ctx, REP)
r1 = C.zmq_bind(receiver, 'tcp://*:3333')
r2 = C.zmq_connect(sender, 'tcp://127.0.0.1:3333')
zmq_msg = ffi.new('zmq_msg_t*')
message = ffi.new('char[5]', 'Hello')
C.zmq_msg_init_data(zmq_msg,
ffi.cast('void*', message),
ffi.cast('size_t', 5),
ffi.NULL,
ffi.NULL)
receiver_pollitem = ffi.new('zmq_pollitem_t*')
receiver_pollitem.socket = receiver
receiver_pollitem.fd = 0
receiver_pollitem.events = POLLIN | POLLOUT
receiver_pollitem.revents = 0
ret = C.zmq_poll(ffi.NULL, 0, 0)
assert ret == 0
ret = C.zmq_poll(receiver_pollitem, 1, 0)
assert ret == 0
ret = C.zmq_msg_send(zmq_msg, sender, 0)
print(ffi.string(C.zmq_strerror(C.zmq_errno())))
assert ret == 5
import time
time.sleep(0.2)
ret = C.zmq_poll(receiver_pollitem, 1, 0)
assert ret == 1
assert int(receiver_pollitem.revents) & POLLIN
assert not int(receiver_pollitem.revents) & POLLOUT
zmq_msg2 = ffi.new('zmq_msg_t*')
C.zmq_msg_init(zmq_msg2)
ret_recv = C.zmq_msg_recv(zmq_msg2, receiver, 0)
assert ret_recv == 5
assert 5 == C.zmq_msg_size(zmq_msg2)
assert b"Hello" == ffi.buffer(C.zmq_msg_data(zmq_msg2),
C.zmq_msg_size(zmq_msg2))[:]
sender_pollitem = ffi.new('zmq_pollitem_t*')
sender_pollitem.socket = sender
sender_pollitem.fd = 0
sender_pollitem.events = POLLIN | POLLOUT
sender_pollitem.revents = 0
ret = C.zmq_poll(sender_pollitem, 1, 0)
assert ret == 0
zmq_msg_again = ffi.new('zmq_msg_t*')
message_again = ffi.new('char[11]', 'Hello Again')
C.zmq_msg_init_data(zmq_msg_again,
ffi.cast('void*', message_again),
ffi.cast('size_t', 11),
ffi.NULL,
ffi.NULL)
assert 11 == C.zmq_msg_send(zmq_msg_again, receiver, 0)
import time
time.sleep(0.2)
assert 0 <= C.zmq_poll(sender_pollitem, 1, 0)
assert int(sender_pollitem.revents) & POLLIN
assert 11 == C.zmq_msg_recv(zmq_msg2, sender, 0)
assert 11 == C.zmq_msg_size(zmq_msg2)
assert b"Hello Again" == ffi.buffer(C.zmq_msg_data(zmq_msg2),
int(C.zmq_msg_size(zmq_msg2)))[:]
assert 0 == C.zmq_close(sender)
assert 0 == C.zmq_close(receiver)
assert 0 == C.zmq_ctx_destroy(ctx)
assert 0 == C.zmq_msg_close(zmq_msg)
assert 0 == C.zmq_msg_close(zmq_msg2)
assert 0 == C.zmq_msg_close(zmq_msg_again)
def start(self):
if self.watch == ffi.NULL:
self.watch = C.zmq_stopwatch_start()
else:
raise ZMQError('Stopwatch is already runing.')