From 6119f835713ed4df5ccc0795387aafa7a25ff18f Mon Sep 17 00:00:00 2001 From: Roger Light Date: Wed, 9 Jan 2019 13:56:15 +0000 Subject: [PATCH] Add receive-maximum=1 test for client. --- .../03-publish-c2b-qos2-receive-maximum-1.py | 118 ++++++++++++++++++ ... 03-publish-c2b-qos2-receive-maximum-2.py} | 0 test/lib/Makefile | 3 +- ...> 03-publish-c2b-qos2-receive-maximum-1.c} | 0 .../c/03-publish-c2b-qos2-receive-maximum-2.c | 52 ++++++++ test/lib/c/Makefile | 3 +- test/lib/ptest.py | 3 +- 7 files changed, 176 insertions(+), 3 deletions(-) create mode 100755 test/lib/03-publish-c2b-qos2-receive-maximum-1.py rename test/lib/{03-publish-c2b-qos2-receive-maximum.py => 03-publish-c2b-qos2-receive-maximum-2.py} (100%) rename test/lib/c/{03-publish-c2b-qos2-receive-maximum.c => 03-publish-c2b-qos2-receive-maximum-1.c} (100%) create mode 100644 test/lib/c/03-publish-c2b-qos2-receive-maximum-2.c diff --git a/test/lib/03-publish-c2b-qos2-receive-maximum-1.py b/test/lib/03-publish-c2b-qos2-receive-maximum-1.py new file mode 100755 index 00000000..cb27cf55 --- /dev/null +++ b/test/lib/03-publish-c2b-qos2-receive-maximum-1.py @@ -0,0 +1,118 @@ +#!/usr/bin/env python + +# Test whether a client responds correctly to multiple PUBLISH with QoS 2, with +# receive maximum set to 1. + +from mosq_test_helper import * + +port = mosq_test.get_lib_port() + +rc = 1 +keepalive = 60 +connect_packet = mosq_test.gen_connect("publish-qos2-test", keepalive=keepalive, proto_ver=5) + +props = mqtt5_props.gen_uint16_prop(mqtt5_props.PROP_RECEIVE_MAXIMUM, 1) +connack_packet = mosq_test.gen_connack(rc=0, proto_ver=5, properties=props) + +disconnect_packet = mosq_test.gen_disconnect(proto_ver=5) + +mid = 1 +publish_1_packet = mosq_test.gen_publish("topic", qos=2, mid=mid, payload="12345", proto_ver=5) +pubrec_1_packet = mosq_test.gen_pubrec(mid, proto_ver=5) +pubrel_1_packet = mosq_test.gen_pubrel(mid, proto_ver=5) +pubcomp_1_packet = mosq_test.gen_pubcomp(mid, proto_ver=5) + +mid = 2 +publish_2_packet = mosq_test.gen_publish("topic", qos=2, mid=mid, payload="12345", proto_ver=5) +pubrec_2_packet = mosq_test.gen_pubrec(mid, proto_ver=5) +pubrel_2_packet = mosq_test.gen_pubrel(mid, proto_ver=5) +pubcomp_2_packet = mosq_test.gen_pubcomp(mid, proto_ver=5) + +mid = 3 +publish_3_packet = mosq_test.gen_publish("topic", qos=2, mid=mid, payload="12345", proto_ver=5) +pubrec_3_packet = mosq_test.gen_pubrec(mid, proto_ver=5) +pubrel_3_packet = mosq_test.gen_pubrel(mid, proto_ver=5) +pubcomp_3_packet = mosq_test.gen_pubcomp(mid, proto_ver=5) + +mid = 4 +publish_4_packet = mosq_test.gen_publish("topic", qos=2, mid=mid, payload="12345", proto_ver=5) +pubrec_4_packet = mosq_test.gen_pubrec(mid, proto_ver=5) +pubrel_4_packet = mosq_test.gen_pubrel(mid, proto_ver=5) +pubcomp_4_packet = mosq_test.gen_pubcomp(mid, proto_ver=5) + +mid = 5 +publish_5_packet = mosq_test.gen_publish("topic", qos=2, mid=mid, payload="12345", proto_ver=5) +pubrec_5_packet = mosq_test.gen_pubrec(mid, proto_ver=5) +pubrel_5_packet = mosq_test.gen_pubrel(mid, proto_ver=5) +pubcomp_5_packet = mosq_test.gen_pubcomp(mid, proto_ver=5) + + +sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) +sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) +sock.settimeout(10) +sock.bind(('', port)) +sock.listen(5) + + +client_args = sys.argv[1:] +env = dict(os.environ) +env['LD_LIBRARY_PATH'] = '../../lib:../../lib/cpp' +try: + pp = env['PYTHONPATH'] +except KeyError: + pp = '' +env['PYTHONPATH'] = '../../lib/python:'+pp +client = mosq_test.start_client(filename=sys.argv[1].replace('/', '-'), cmd=client_args, env=env, port=port) + + +try: + (conn, address) = sock.accept() + conn.settimeout(10) + + if mosq_test.expect_packet(conn, "connect", connect_packet): + conn.send(connack_packet) + + if mosq_test.expect_packet(conn, "publish 1", publish_1_packet): + conn.send(pubrec_1_packet) + if mosq_test.expect_packet(conn, "pubrel 1", pubrel_1_packet): + conn.send(pubcomp_1_packet) + + if mosq_test.expect_packet(conn, "publish 2", publish_2_packet): + conn.send(pubrec_2_packet) + if mosq_test.expect_packet(conn, "pubrel 2", pubrel_2_packet): + conn.send(pubcomp_2_packet) + + if mosq_test.expect_packet(conn, "publish 3", publish_3_packet): + conn.send(pubrec_3_packet) + if mosq_test.expect_packet(conn, "pubrel 3", pubrel_3_packet): + conn.send(pubcomp_3_packet) + + if mosq_test.expect_packet(conn, "publish 4", publish_4_packet): + conn.send(pubrec_4_packet) + if mosq_test.expect_packet(conn, "pubrel 4", pubrel_4_packet): + conn.send(pubcomp_4_packet) + + if mosq_test.expect_packet(conn, "publish 5", publish_5_packet): + conn.send(pubrec_5_packet) + if mosq_test.expect_packet(conn, "pubrel 5", pubrel_5_packet): + conn.send(pubcomp_5_packet) + rc = 0 + + conn.close() +finally: + for i in range(0, 5): + if client.returncode != None: + break + time.sleep(0.1) + + try: + client.terminate() + except OSError: + pass + + client.wait() + sock.close() + if client.returncode != 0: + exit(1) + +exit(rc) diff --git a/test/lib/03-publish-c2b-qos2-receive-maximum.py b/test/lib/03-publish-c2b-qos2-receive-maximum-2.py similarity index 100% rename from test/lib/03-publish-c2b-qos2-receive-maximum.py rename to test/lib/03-publish-c2b-qos2-receive-maximum-2.py diff --git a/test/lib/Makefile b/test/lib/Makefile index 2d4cba92..f7eb8858 100644 --- a/test/lib/Makefile +++ b/test/lib/Makefile @@ -37,7 +37,8 @@ c : test-compile ./03-publish-c2b-qos2.py $@/03-publish-c2b-qos2.test ./03-publish-c2b-qos2-disconnect.py $@/03-publish-c2b-qos2-disconnect.test ./03-publish-c2b-qos1-receive-maximum.py $@/03-publish-c2b-qos1-receive-maximum.test - ./03-publish-c2b-qos2-receive-maximum.py $@/03-publish-c2b-qos2-receive-maximum.test + ./03-publish-c2b-qos2-receive-maximum-1.py $@/03-publish-c2b-qos2-receive-maximum-1.test + ./03-publish-c2b-qos2-receive-maximum-2.py $@/03-publish-c2b-qos2-receive-maximum-2.test ./03-publish-b2c-qos1.py $@/03-publish-b2c-qos1.test ./03-publish-b2c-qos2.py $@/03-publish-b2c-qos2.test ./03-request-response.py $@/03-request-response.test diff --git a/test/lib/c/03-publish-c2b-qos2-receive-maximum.c b/test/lib/c/03-publish-c2b-qos2-receive-maximum-1.c similarity index 100% rename from test/lib/c/03-publish-c2b-qos2-receive-maximum.c rename to test/lib/c/03-publish-c2b-qos2-receive-maximum-1.c diff --git a/test/lib/c/03-publish-c2b-qos2-receive-maximum-2.c b/test/lib/c/03-publish-c2b-qos2-receive-maximum-2.c new file mode 100644 index 00000000..a4b8e2a2 --- /dev/null +++ b/test/lib/c/03-publish-c2b-qos2-receive-maximum-2.c @@ -0,0 +1,52 @@ +#include +#include +#include +#include +#include + +static int run = -1; + +void on_connect(struct mosquitto *mosq, void *obj, int rc, int flags, const mosquitto_property *properties) +{ + if(rc){ + exit(1); + } + + for(int i=0; i<5; i++){ + mosquitto_publish_v5(mosq, NULL, "topic", 5, "12345", 2, false, NULL); + } +} + +void on_publish(struct mosquitto *mosq, void *obj, int mid, const mosquitto_property *properties) +{ + if(mid == 5){ + mosquitto_disconnect(mosq); + run = 0; + } +} + +int main(int argc, char *argv[]) +{ + int rc; + struct mosquitto *mosq; + mosquitto_property *props = NULL; + + int port = atoi(argv[1]); + + mosquitto_lib_init(); + + mosq = mosquitto_new("publish-qos2-test", true, &run); + mosquitto_int_option(mosq, MOSQ_OPT_PROTOCOL_VERSION, MQTT_PROTOCOL_V5); + + mosquitto_connect_v5_callback_set(mosq, on_connect); + mosquitto_publish_v5_callback_set(mosq, on_publish); + + rc = mosquitto_connect_bind_v5(mosq, "localhost", port, 60, NULL, NULL); + + while(run == -1){ + mosquitto_loop(mosq, 300, 1); + } + + mosquitto_lib_cleanup(); + return run; +} diff --git a/test/lib/c/Makefile b/test/lib/c/Makefile index 69dc56a3..289d33d6 100644 --- a/test/lib/c/Makefile +++ b/test/lib/c/Makefile @@ -20,7 +20,8 @@ SRC = \ 03-publish-c2b-qos2.c \ 03-publish-c2b-qos2-disconnect.c \ 03-publish-c2b-qos1-receive-maximum.c \ - 03-publish-c2b-qos2-receive-maximum.c \ + 03-publish-c2b-qos2-receive-maximum-1.c \ + 03-publish-c2b-qos2-receive-maximum-2.c \ 03-publish-b2c-qos1.c \ 03-publish-b2c-qos2.c \ 03-request-response-1.c \ diff --git a/test/lib/ptest.py b/test/lib/ptest.py index 34b53323..56d478bc 100755 --- a/test/lib/ptest.py +++ b/test/lib/ptest.py @@ -22,7 +22,8 @@ tests = [ ('./03-publish-c2b-qos2-disconnect.py', 'c/03-publish-c2b-qos2-disconnect.test'), ('./03-publish-c2b-qos2.py', 'c/03-publish-c2b-qos2.test'), ('./03-publish-c2b-qos1-receive-maximum.py', 'c/03-publish-c2b-qos1-receive-maximum.test'), - ('./03-publish-c2b-qos2-receive-maximum.py', 'c/03-publish-c2b-qos2-receive-maximum.test'), + ('./03-publish-c2b-qos2-receive-maximum-1.py', 'c/03-publish-c2b-qos2-receive-maximum-1.test'), + ('./03-publish-c2b-qos2-receive-maximum-2.py', 'c/03-publish-c2b-qos2-receive-maximum-2.test'), ('./03-publish-qos0-no-payload.py', 'c/03-publish-qos0-no-payload.test'), ('./03-publish-qos0.py', 'c/03-publish-qos0.test'), ('./03-request-response.py', 'c/03-request-response.test'),