Add expiration time in mock connector

This patch adds an expiration time of 10 seconds per message in script mock connector.
CESARBR · Dec 26, 2019 · 5410478 · 5410478
1 parent 01a0a48
commit 5410478
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/test/mock-connector.py b/test/mock-connector.py
@@ -29,6 +29,8 @@
 
 QUEUE_CLOUD_NAME = 'connIn-messages'
 
+MESSAGE_EXPIRATION_TIME_MS = '10000'
+
 EVENT_REGISTER = 'device.register'
 KEY_REGISTERED = 'device.registered'
 
@@ -170,12 +172,18 @@ def msg_update(args):
     channel = __amqp_start()
     msg = __parse_update_message(args.json_msg_file)
     channel.basic_publish(exchange=fog_exchange,
+                          properties=pika.BasicProperties(
+                            expiration=MESSAGE_EXPIRATION_TIME_MS,
+                          ),
                           routing_key=KEY_UPDATE, body=msg)
 
 def msg_request(args):
     channel = __amqp_start()
     msg = __parse_request_message(args.json_msg_file)
     channel.basic_publish(exchange=fog_exchange,
+                          properties=pika.BasicProperties(
+                            expiration=MESSAGE_EXPIRATION_TIME_MS,
+                          ),
                           routing_key=KEY_REQUEST, body=msg)
 
 def no_command(args):