def test_send_message_zmq_error(self):
mock_zmq_context = mock.Mock(spec=zmq.Context)
testPeer = peer.Peer(mock_zmq_context, "local_identity",
"peer_identity")
testPeer.start()
testPeer._zmq_dealer.send.side_effect = zmq.ZMQError
testPeer.send_message(b"test_message")
def test_init(self):
testPeer = peer.Peer(zmq.Context.instance(), "local_identity",
"peer_identity")
assert isinstance(testPeer._context, zmq.Context)
assert testPeer._local_identity == "local_identity"
assert testPeer._peer_identity == "peer_identity"
assert not testPeer._is_started
def test_send_message(self):
mock_zmq_context = mock.Mock(spec=zmq.Context)
testPeer = peer.Peer(mock_zmq_context, "local_identity",
"peer_identity")
testPeer.start()
testPeer.send_message(b"test_message")
testPeer._zmq_dealer.send.assert_called_once_with(b"test_message",
flags=zmq.NOBLOCK)
def test_stop(self):
mock_zmq_context = mock.Mock(spec=zmq.Context)
testPeer = peer.Peer(mock_zmq_context, "local_identity",
"peer_identity")
testPeer.start()
testPeer.stop()
testPeer._zmq_dealer.close.assert_called_once_with()
assert not testPeer._is_started
def test_start(self):
mock_zmq_context = mock.Mock(spec=zmq.Context)
testPeer = peer.Peer(mock_zmq_context, "local_identity",
"peer_identity")
testPeer.start()
mock_connect = testPeer._zmq_dealer.connect
mock_connect.assert_called_once_with("tcp://peer_identity")
assert testPeer._is_started
def test_setup(self):
testPeer = peer.Peer(zmq.Context.instance(), "local_identity",
"peer_identity")
testPeer._setup()
assert isinstance(testPeer._zmq_dealer, zmq.sugar.socket.Socket)
assert (testPeer._zmq_dealer.getsockopt(
zmq.IDENTITY) == six.b("local_identity"))
assert testPeer._zmq_dealer.getsockopt(zmq.LINGER) == 0