• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• jsonrpcclient-master
  • .coveragerc
  • examples
    • websockets
      • request.py
      • notification.py
      • batch.py
    • tornado
      • request.py
      • batch.py
    • aiohttp
      • request.py
      • batch.py
    • zeromq
      • request.py
      • notification.py
      • batch.py
    • http
      • request.py
      • batch.py
  • .pre-commit-config.yaml
  • LICENSE
  • jsonrpcclient
    • log.py
    • requests.py
    • response-schema.json
    • exceptions.py
    • parse.py
    • config.py
    • py.typed
    • client.py
    • id_generators.py
    • async_client.py
    • clients
      • websockets_client.py
      • http_client.py
      • aiohttp_client.py
      • socket_client.py
      • zeromq_client.py
      • __init__.py
      • tornado_client.py
    • response.py
    • __main__.py
    • __init__.py
  • CHANGELOG.md
  • setup.py
  • README.md
  • tests
    • test_exceptions.py
    • test_main.py
    • test_log.py
    • test_id_generators.py
    • test_response.py
    • clients
      • test_tornado_client.py
      • test_http_client.py
      • test_zeromq_client.py
    • response.py
    • test_client.py
    • test_requests.py
    • __init__.py
    • test_parse.py
    • test_async_client.py
  • .gitignore
  • requirements-docs.txt
  • MANIFEST.in
  • doc
    • Makefile
    • index.md
    • examples.rst
    • api.md
    • conf.py
    • _templates
      • layout.html
  • tox.ini

import itertools
import json

import pytest
import zmq
from mock import Mock, patch

from jsonrpcclient.clients.zeromq_client import ZeroMQClient
from jsonrpcclient.requests import Request


class TestZeroMQClient:
    def setup_method(self):
        # Patch Request.id_generator to ensure the request id is always 1
        Request.id_generator = itertools.count(1)

    def test_instantiate(self):
        ZeroMQClient("tcp://localhost:5555")

    @patch("zmq.Socket.send_string", Mock())
    @patch(
        "zmq.Socket.recv",
        Mock(
            side_effect=lambda: json.dumps(
                {"jsonrpc": "2.0", "result": 99, "id": 1}
            ).encode("utf-8")
        ),
    )
    @patch("jsonrpcclient.client.response_log")
    def test_send_message(self, *_):
        client = ZeroMQClient("tcp://localhost:5555")
        client.send_message(str(Request("go")), response_expected=True)

    def test_send_message_conn_error(self):
        client = ZeroMQClient("tcp://localhost:5555")
        # Set timeouts
        client.socket.setsockopt(zmq.RCVTIMEO, 5)
        client.socket.setsockopt(zmq.SNDTIMEO, 5)
        client.socket.setsockopt(zmq.LINGER, 5)
        with pytest.raises(zmq.error.ZMQError):
            client.send_message(str(Request("go")), response_expected=True)