Improve client test for unexpected pubrel.

test/lib/03-publish-b2c-qos2-unexpected-pubrel.py

@@ -20,6 +20,8 @@ pubrec_packet = mosq_test.gen_pubrec(mid)
 pubrel_packet = mosq_test.gen_pubrel(mid)
 pubcomp_packet = mosq_test.gen_pubcomp(mid)
 
+publish_quit_packet = mosq_test.gen_publish("quit", qos=0, payload="quit")
+
 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
 sock.settimeout(10)
@@ -52,6 +54,7 @@ try:
                 conn.send(pubrel_packet)
 
                 if mosq_test.expect_packet(conn, "pubcomp", pubcomp_packet):
+                    conn.send(publish_quit_packet)
                     rc = 0
 
     conn.close()

test/lib/c/03-publish-b2c-qos2-unexpected-pubrel.c

@@ -15,6 +15,10 @@ void on_connect(struct mosquitto *mosq, void *obj, int rc)
 
 void on_message(struct mosquitto *mosq, void *obj, const struct mosquitto_message *msg)
 {
+	if(!strcmp(msg->topic, "quit")){
+		run = 0;
+		return;
+	}
 	if(msg->mid != 13423){
 		printf("Invalid mid (%d)\n", msg->mid);
 		exit(1);
@@ -40,7 +44,6 @@ void on_message(struct mosquitto *mosq, void *obj, const struct mosquitto_messag
 		exit(1);
 	}
 
-	run = 0;
 }
 
 int main(int argc, char *argv[])