def test_read_size():
conn = _get_test_conn()
body_size = 6
body_size_packed = struct_l.pack(body_size)
conn._read_size(body_size_packed)
conn.stream.read_bytes.assert_called_once_with(body_size, conn._read_body)
# now test that we get the right behavior when we get malformed data
# for this, we'll want to stick on mock on conn.close
conn.stream.read_bytes.reset_mock()
with patch.object(conn, 'close', autospec=True) as mock_close:
conn._read_size('asdfasdf')
mock_close.assert_called_once_with()
assert not conn.stream.read_bytes.called
def test_read_size():
conn = _get_test_conn()
body_size = 6
body_size_packed = struct_l.pack(body_size)
conn._read_size(body_size_packed)
conn.stream.read_bytes.assert_called_once_with(body_size, conn._read_body)
# now test that we get the right behavior when we get malformed data
# for this, we'll want to stick on mock on conn.close
conn.stream.read_bytes.reset_mock()
with patch.object(conn, 'close', autospec=True) as mock_close:
conn._read_size('asdfasdf')
mock_close.assert_called_once_with()
assert not conn.stream.read_bytes.called
def pytest_generate_tests(metafunc):
identify_dict_ascii = {'a': 1, 'b': 2}
identify_dict_unicode = {'c': u'w\xc3\xa5\xe2\x80\xa0'}
identify_body_ascii = to_bytes(json.dumps(identify_dict_ascii))
identify_body_unicode = to_bytes(json.dumps(identify_dict_unicode))
msgs = [b'asdf', b'ghjk', b'abcd']
mpub_body = struct_l.pack(len(msgs)) + b''.join(
struct_l.pack(len(m)) + m for m in msgs)
if metafunc.function == test_command:
for cmd_method, kwargs, result in [
(protocol.identify, {
'data': identify_dict_ascii
}, b'IDENTIFY\n' + struct_l.pack(len(identify_body_ascii)) +
to_bytes(identify_body_ascii)),
(protocol.identify, {
'data': identify_dict_unicode
}, b'IDENTIFY\n' + struct_l.pack(len(identify_body_unicode)) +
to_bytes(identify_body_unicode)),
(protocol.subscribe, {
'topic': 'test_topic',
'channel': 'test_channel'
}, b'SUB test_topic test_channel\n'),
(protocol.finish, {
'id': 'test'
}, b'FIN test\n'),
(protocol.finish, {
'id': u'\u2020est \xfcn\xee\xe7\xf8\u2202\xe9'
},
b'FIN \xe2\x80\xa0est \xc3\xbcn\xc3\xae\xc3\xa7\xc3\xb8\xe2\x88\x82\xc3\xa9\n'
), (protocol.requeue, {
'id': 'test'
}, b'REQ test 0\n'),
(protocol.requeue, {
'id': 'test',
'time_ms': 60
}, b'REQ test 60\n'),
(protocol.touch, {
'id': 'test'
}, b'TOUCH test\n'), (protocol.ready, {
'count': 100
}, b'RDY 100\n'), (protocol.nop, {}, b'NOP\n'),
(protocol.pub, {
'topic': 'test',
'data': msgs[0]
},
b'PUB test\n' + struct_l.pack(len(msgs[0])) + to_bytes(msgs[0])),
(protocol.mpub, {
'topic': 'test',
'data': msgs
}, b'MPUB test\n' + struct_l.pack(len(mpub_body)) +
to_bytes(mpub_body))
]:
metafunc.addcall(funcargs=dict(
cmd_method=cmd_method, kwargs=kwargs, result=result))
def pytest_generate_tests(metafunc):
identify_dict_ascii = {'a': 1, 'b': 2}
identify_dict_unicode = {'c': u'w\xc3\xa5\xe2\x80\xa0'}
identify_body_ascii = to_bytes(json.dumps(identify_dict_ascii))
identify_body_unicode = to_bytes(json.dumps(identify_dict_unicode))
msgs = [b'asdf', b'ghjk', b'abcd']
mpub_body = struct_l.pack(len(msgs)) + b''.join(struct_l.pack(len(m)) + m for m in msgs)
if metafunc.function == test_command:
for cmd_method, kwargs, result in [
(protocol.identify,
{'data': identify_dict_ascii},
b'IDENTIFY\n' + struct_l.pack(len(identify_body_ascii)) +
to_bytes(identify_body_ascii)),
(protocol.identify,
{'data': identify_dict_unicode},
b'IDENTIFY\n' + struct_l.pack(len(identify_body_unicode)) +
to_bytes(identify_body_unicode)),
(protocol.subscribe,
{'topic': 'test_topic', 'channel': 'test_channel'},
b'SUB test_topic test_channel\n'),
(protocol.finish,
{'id': 'test'},
b'FIN test\n'),
(protocol.finish,
{'id': u'\u2020est \xfcn\xee\xe7\xf8\u2202\xe9'},
b'FIN \xe2\x80\xa0est \xc3\xbcn\xc3\xae\xc3\xa7\xc3\xb8\xe2\x88\x82\xc3\xa9\n'),
(protocol.requeue,
{'id': 'test'},
b'REQ test 0\n'),
(protocol.requeue,
{'id': 'test', 'time_ms': 60},
b'REQ test 60\n'),
(protocol.touch,
{'id': 'test'},
b'TOUCH test\n'),
(protocol.ready,
{'count': 100},
b'RDY 100\n'),
(protocol.nop,
{},
b'NOP\n'),
(protocol.pub,
{'topic': 'test', 'data': msgs[0]},
b'PUB test\n' + struct_l.pack(len(msgs[0])) + to_bytes(msgs[0])),
(protocol.mpub,
{'topic': 'test', 'data': msgs},
b'MPUB test\n' + struct_l.pack(len(mpub_body)) + to_bytes(mpub_body))
]:
metafunc.addcall(funcargs=dict(cmd_method=cmd_method, kwargs=kwargs, result=result))
# shunt '..' into sys.path since we are in a 'tests' subdirectory
base_dir = os.path.abspath(os.path.join(os.path.dirname(os.path.abspath(__file__)), '..'))
if base_dir not in sys.path:
sys.path.insert(0, base_dir)
from nsq._compat import to_bytes
from nsq._compat import struct_l
from nsq import protocol
identify_dict_ascii = {'a': 1, 'b': 2}
identify_dict_unicode = {'c': u'w\xc3\xa5\xe2\x80\xa0'}
identify_body_ascii = to_bytes(json.dumps(identify_dict_ascii))
identify_body_unicode = to_bytes(json.dumps(identify_dict_unicode))
msgs = [b'asdf', b'ghjk', b'abcd']
mpub_body = struct_l.pack(len(msgs)) + b''.join(struct_l.pack(len(m)) + m for m in msgs)
@pytest.mark.parametrize(['cmd_method', 'kwargs', 'result'], [
pytest.param(protocol.identify,
{'data': identify_dict_ascii},
b'IDENTIFY\n' + struct_l.pack(len(identify_body_ascii))
+ to_bytes(identify_body_ascii), # noqa: W503
id="identify-ascii"),
pytest.param(protocol.identify,
{'data': identify_dict_unicode},
b'IDENTIFY\n' + struct_l.pack(len(identify_body_unicode))
+ to_bytes(identify_body_unicode), # noqa: W503
id="identify-unicode"),
pytest.param(protocol.subscribe,
{'topic': 'test_topic', 'channel': 'test_channel'},
def mock_response_write(c, frame_type, data):
body_size = 4 + len(data)
body_size_packed = struct_l.pack(body_size)
frame_type_packed = struct_l.pack(frame_type)
mock_write(c, body_size_packed + frame_type_packed + data)
# shunt '..' into sys.path since we are in a 'tests' subdirectory
base_dir = os.path.abspath(
os.path.join(os.path.dirname(os.path.abspath(__file__)), '..'))
if base_dir not in sys.path:
sys.path.insert(0, base_dir)
from nsq._compat import to_bytes
from nsq._compat import struct_l
from nsq import protocol
identify_dict_ascii = {'a': 1, 'b': 2}
identify_dict_unicode = {'c': u'w\xc3\xa5\xe2\x80\xa0'}
identify_body_ascii = to_bytes(json.dumps(identify_dict_ascii))
identify_body_unicode = to_bytes(json.dumps(identify_dict_unicode))
msgs = [b'asdf', b'ghjk', b'abcd']
mpub_body = struct_l.pack(len(msgs)) + b''.join(
struct_l.pack(len(m)) + m for m in msgs)
@pytest.mark.parametrize(
['cmd_method', 'kwargs', 'result'],
[
pytest.param(
protocol.identify,
{'data': identify_dict_ascii},
b'IDENTIFY\n' + struct_l.pack(len(identify_body_ascii)) +
to_bytes(identify_body_ascii), # noqa: W503
id="identify-ascii"),
pytest.param(
protocol.identify,
{'data': identify_dict_unicode},
def mock_response_write(c, frame_type, data):
body_size = 4 + len(data)
body_size_packed = struct_l.pack(body_size)
frame_type_packed = struct_l.pack(frame_type)
mock_write(c, body_size_packed + frame_type_packed + data)