def test_sync_receive_messages(): @mock_server def handle(socket, address): assert socket.recv(4) == b' V2' assert socket.recv(9) == b'IDENTIFY\n' size = nsq.unpack_size(socket.recv(4)) data = json.loads(socket.recv(size).decode('utf-8')) assert isinstance(data, dict) socket.sendall(mock_response(nsq.FRAME_TYPE_RESPONSE, b'OK')) msg = b'SUB topic channel\n' assert socket.recv(len(msg)) == msg socket.sendall(mock_response(nsq.FRAME_TYPE_RESPONSE, b'OK')) for i in range(10): assert socket.recv(6) == b'RDY 1\n' body = six.b(json.dumps({'data': {'test_key': i}})) ts = i * 1000 * 1000 socket.sendall(mock_response_message(ts, i, i, body)) with handle as server: conn = NsqdTCPClient('127.0.0.1', server.server_port) conn.connect() assert conn.identify() is None conn.subscribe('topic', 'channel') frame, data = conn.read_response() assert frame == nsq.FRAME_TYPE_RESPONSE assert data == b'OK' for i in range(10): conn.ready(1) frame, msg = conn.read_response() assert frame == nsq.FRAME_TYPE_MESSAGE assert isinstance(msg, Message) assert msg.timestamp == i * 1000 * 1000 assert msg.id == six.b('%016d' % i) assert msg.attempts == i assert json.loads( msg.body.decode('utf-8'))['data']['test_key'] == i
def test_sync_receive_messages(): @mock_server def handle(socket, address): assert socket.recv(4) == b' V2' assert socket.recv(9) == b'IDENTIFY\n' size = nsq.unpack_size(socket.recv(4)) data = json.loads(socket.recv(size).decode('utf-8')) assert isinstance(data, dict) socket.sendall(mock_response(nsq.FRAME_TYPE_RESPONSE, b'OK')) msg = b'SUB topic channel\n' assert socket.recv(len(msg)) == msg socket.sendall(mock_response(nsq.FRAME_TYPE_RESPONSE, b'OK')) for i in range(10): assert socket.recv(6) == b'RDY 1\n' body = six.b(json.dumps({'data': {'test_key': i}})) ts = i * 1000 * 1000 socket.sendall(mock_response_message(ts, i, i, body)) with handle as server: conn = NsqdTCPClient('127.0.0.1', server.server_port) conn.connect() assert conn.identify() is None conn.subscribe('topic', 'channel') frame, data = conn.read_response() assert frame == nsq.FRAME_TYPE_RESPONSE assert data == b'OK' for i in range(10): conn.ready(1) frame, msg = conn.read_response() assert frame == nsq.FRAME_TYPE_MESSAGE assert isinstance(msg, Message) assert msg.timestamp == i * 1000 * 1000 assert msg.id == six.b('%016d' % i) assert msg.attempts == i assert json.loads(msg.body.decode('utf-8'))['data']['test_key'] == i
def test_socket_upgrades(tls, deflate, snappy): with NsqdIntegrationServer() as server: options = { 'address': server.address, 'port': server.tcp_port, 'deflate': deflate, 'snappy': snappy, } if tls: options.update({ 'tls_v1': True, 'tls_options': { 'keyfile': server.tls_key, 'certfile': server.tls_cert, } }) conn = NsqdTCPClient(**options) conn.connect() assert conn.state == states.CONNECTED if deflate and snappy: with pytest.raises(errors.NSQErrorCode): conn.identify() return if tls and BAD_GEVENT: with pytest.raises(AttributeError): conn.identify() return if tls and server.version < (0, 2, 28): with pytest.raises(ssl.SSLError): conn.identify() return resp = conn.identify() assert isinstance(resp, dict) assert resp['tls_v1'] is tls assert resp['deflate'] is deflate assert resp['snappy'] is snappy if tls and (deflate or snappy): assert isinstance(conn.stream.socket._socket, SSLSocket) elif tls: assert isinstance(conn.stream.socket, SSLSocket) if deflate: assert isinstance(conn.stream.socket, DefalteSocket) if snappy: assert isinstance(conn.stream.socket, SnappySocket) conn.publish('topic', b'sup') frame, data = conn.read_response() assert frame == nsq.FRAME_TYPE_RESPONSE assert data == b'OK' conn.subscribe('topic', 'channel') frame, data = conn.read_response() assert frame == nsq.FRAME_TYPE_RESPONSE assert data == b'OK' conn.ready(1) frame, data = conn.read_response() assert frame == nsq.FRAME_TYPE_MESSAGE assert data.body == b'sup' conn.close_stream()
def test_socket_upgrades(tls, deflate, snappy): with NsqdIntegrationServer() as server: options = { 'address': server.address, 'port': server.tcp_port, 'deflate': deflate, 'snappy': snappy, } if tls: options.update({ 'tls_v1': True, 'tls_options': { 'keyfile': server.tls_key, 'certfile': server.tls_cert, } }) conn = NsqdTCPClient(**options) conn.connect() assert conn.state == states.CONNECTED if deflate and snappy: with pytest.raises(errors.NSQErrorCode): conn.identify() return if tls and BAD_GEVENT: with pytest.raises(AttributeError): conn.identify() return if tls and server.version < (0, 2, 28): with pytest.raises(ssl.SSLError): conn.identify() return resp = conn.identify() assert isinstance(resp, dict) assert resp['tls_v1'] is tls assert resp['deflate'] is deflate assert resp['snappy'] is snappy if tls and (deflate or snappy): assert isinstance(conn.stream.socket._socket, SSLSocket) elif tls: assert isinstance(conn.stream.socket, SSLSocket) if deflate: assert isinstance(conn.stream.socket, DefalteSocket) if snappy: assert isinstance(conn.stream.socket, SnappySocket) conn.publish('topic', b'sup') frame, data = conn.read_response() assert frame == nsq.FRAME_TYPE_RESPONSE assert data == b'OK' conn.subscribe('topic', 'channel') frame, data = conn.read_response() assert frame == nsq.FRAME_TYPE_RESPONSE assert data == b'OK' conn.ready(1) frame, data = conn.read_response() assert frame == nsq.FRAME_TYPE_MESSAGE assert data.body == b'sup' conn.close_stream()