from mock import patch, MagicMock

import pytest


@pytest.fixture(autouse=True)
def mock_zmq_context():
    with patch("zmq.Context") as mock:
        mock.instance.return_value = MagicMock()
        yield mock