alex/esp-idf
alex/esp-idf
1
0
Fork 0
mirror of https://github.com/espressif/esp-idf.git synced 2024-10-05 20:47:46 -04:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

ci/mqtt: Make publish test-app message properties configurable

Browse Source 
Also increase the default sizes and repeat-counts to send more data and
exercise the library more intensly
This commit is contained in:
David Cermak 2022-01-27 07:43:05 +01:00
parent 9de81ed2e2
commit c2b037572c
1 changed files with 20 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
tools/test_apps/protocols/mqtt/publish_connect_test/app_test.py
View File

@ -10,6 +10,7 @@ import string
import subprocess import subprocess
import sys import sys
import time import time
from itertools import count
from threading import Event, Lock, Thread from threading import Event, Lock, Thread
import paho.mqtt.client as mqtt import paho.mqtt.client as mqtt
@ -383,16 +384,31 @@ def test_app_protocol_mqtt_publish_connect(env, extra_data):
with MqttPublisher(dut1, transport, qos, repeat, published, queue, publish_cfg): with MqttPublisher(dut1, transport, qos, repeat, published, queue, publish_cfg):
pass pass
# Initialize message sizes and repeat counts (if defined in the environment)
messages = []
for i in count(0):
# Check env variable: MQTT_PUBLISH_MSG_{len|repeat}_{x}
env_dict = {var:'MQTT_PUBLISH_MSG_' + var + '_' + str(i) for var in ['len', 'repeat']}
if os.getenv(env_dict['len']) and os.getenv(env_dict['repeat']):
messages.append({var: int(os.getenv(env_dict[var])) for var in ['len', 'repeat']})
continue
break
if not messages: # No message sizes present in the env - set defaults
messages = [{'len':0, 'repeat':5}, # zero-sized messages
{'len':2, 'repeat':10}, # short messages
{'len':200, 'repeat':3}, # long messages
{'len':20, 'repeat':50} # many medium sized
]
# Iterate over all publish message properties
for qos in [0, 1, 2]: for qos in [0, 1, 2]:
for transport in ['tcp', 'ssl', 'ws', 'wss']: for transport in ['tcp', 'ssl', 'ws', 'wss']:
for q in [0, 1]: for q in [0, 1]:
if publish_cfg['broker_host_' + transport] is None: if publish_cfg['broker_host_' + transport] is None:
print('Skipping transport: {}...'.format(transport)) print('Skipping transport: {}...'.format(transport))
continue continue
start_publish_case(transport, qos, 0, 5, q) for msg in messages:
start_publish_case(transport, qos, 2, 5, q) start_publish_case(transport, qos, msg['len'], msg['repeat'], q)
start_publish_case(transport, qos, 50, 1, q)
start_publish_case(transport, qos, 10, 20, q)
if __name__ == '__main__': if __name__ == '__main__':