Add db_dump test to verify current broker output.
parent
6b1b0bf7bc
commit
27c99bc7e7
@ -0,0 +1,92 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
# Check whether output from the current broker can be read by the current db_dump.
|
||||
|
||||
from mosq_test_helper import *
|
||||
import shutil
|
||||
|
||||
def write_config(conf_file, port):
|
||||
with open(conf_file, 'w') as f:
|
||||
f.write(f"listener {port}\n")
|
||||
f.write("allow_anonymous true\n")
|
||||
f.write("persistence true\n")
|
||||
f.write(f"persistence_location {port}\n")
|
||||
|
||||
def check_db(port, counts):
|
||||
stdout = f"DB_CHUNK_CFG: {counts[0]}\n" + \
|
||||
f"DB_CHUNK_BASE_MSG: {counts[1]}\n" + \
|
||||
f"DB_CHUNK_CLIENT_MSG: {counts[2]}\n" + \
|
||||
f"DB_CHUNK_RETAIN: {counts[3]}\n" + \
|
||||
f"DB_CHUNK_SUB: {counts[4]}\n" + \
|
||||
f"DB_CHUNK_CLIENT: {counts[5]}\n"
|
||||
|
||||
cmd = ['../../apps/db_dump/mosquitto_db_dump',
|
||||
'--stats',
|
||||
f'{port}/mosquitto.db'
|
||||
]
|
||||
res = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, timeout=1, encoding='utf-8')
|
||||
if res.stdout != stdout:
|
||||
print(res.stdout)
|
||||
raise mosq_test.TestError
|
||||
|
||||
|
||||
def do_test(counts):
|
||||
rc = 1
|
||||
|
||||
port = mosq_test.get_port()
|
||||
conf_file = os.path.basename(__file__).replace('.py', '.conf')
|
||||
|
||||
try:
|
||||
if not os.path.exists(str(port)):
|
||||
os.mkdir(str(port))
|
||||
except FileExistsError:
|
||||
pass
|
||||
write_config(conf_file, port)
|
||||
|
||||
try:
|
||||
broker = mosq_test.start_broker(filename=os.path.basename(__file__), port=port, use_conf=True)
|
||||
env = {
|
||||
'LD_LIBRARY_PATH': mosq_test.get_build_root() + '/lib',
|
||||
'XDG_CONFIG_HOME':'/tmp/missing'
|
||||
}
|
||||
|
||||
# Set up persistent client session, including a subscription
|
||||
cmd = ['../../client/mosquitto_sub',
|
||||
'-c',
|
||||
'-i', 'client-id',
|
||||
'-p', str(port),
|
||||
'-q', '1',
|
||||
'-t', 'sub-topic',
|
||||
'-E'
|
||||
]
|
||||
subprocess.run(cmd, timeout=1, env=env)
|
||||
|
||||
# Publish a retained message which is also queued for the subscriber
|
||||
cmd = ['../../client/mosquitto_pub',
|
||||
'-p', str(port),
|
||||
'-q', '1',
|
||||
'-t', 'sub-topic',
|
||||
'-m', 'message',
|
||||
'-r'
|
||||
]
|
||||
subprocess.run(cmd, timeout=1, env=env)
|
||||
|
||||
broker.terminate()
|
||||
if mosq_test.wait_for_subprocess(broker):
|
||||
print("broker not terminated")
|
||||
raise mosq_test.TestError
|
||||
check_db(port, counts)
|
||||
rc = 0
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
finally:
|
||||
os.remove(conf_file)
|
||||
os.remove(f"{port}/mosquitto.db")
|
||||
shutil.rmtree(str(port))
|
||||
if broker is not None:
|
||||
broker.terminate()
|
||||
|
||||
exit(rc)
|
||||
|
||||
do_test([1,1,1,1,1,1])
|
Loading…
Reference in New Issue