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