mirror of
https://github.com/espressif/esp-idf.git
synced 2024-10-05 20:47:46 -04:00
304 lines
14 KiB
Python
304 lines
14 KiB
Python
import logging
|
|
import os
|
|
import re
|
|
from threading import Thread
|
|
|
|
import ttfw_idf
|
|
from tiny_test_fw import DUT
|
|
|
|
LOG_LEVEL = logging.DEBUG
|
|
LOGGER_NAME = 'modbus_test'
|
|
|
|
# Allowed options for the test
|
|
TEST_READ_MAX_ERR_COUNT = 3 # Maximum allowed read errors during initialization
|
|
TEST_THREAD_JOIN_TIMEOUT = 60 # Test theread join timeout in seconds
|
|
TEST_EXPECT_STR_TIMEOUT = 30 # Test expect timeout in seconds
|
|
TEST_MASTER_TCP = 'mb_tcp_master'
|
|
TEST_SLAVE_TCP = 'mb_tcp_slave'
|
|
|
|
STACK_DEFAULT = 0
|
|
STACK_IPV4 = 1
|
|
STACK_IPV6 = 2
|
|
STACK_INIT = 3
|
|
STACK_CONNECT = 4
|
|
STACK_START = 5
|
|
STACK_PAR_OK = 6
|
|
STACK_PAR_FAIL = 7
|
|
STACK_DESTROY = 8
|
|
|
|
pattern_dict_slave = {STACK_IPV4: (r'.*I \([0-9]+\) example_connect: - IPv4 address: ([0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}).*'),
|
|
STACK_IPV6: (r'.*I \([0-9]+\) example_connect: - IPv6 address: (([A-Fa-f0-9]{1,4}::?){1,7}[A-Fa-f0-9]{1,4}).*'),
|
|
STACK_INIT: (r'.*I \(([0-9]+)\) MB_TCP_SLAVE_PORT: (Protocol stack initialized).'),
|
|
STACK_CONNECT: (r'.*I\s\(([0-9]+)\) MB_TCP_SLAVE_PORT: Socket \(#[0-9]+\), accept client connection from address: '
|
|
r'([0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}).*'),
|
|
STACK_START: (r'.*I\s\(([0-9]+)\) SLAVE_TEST: (Start modbus test).*'),
|
|
STACK_PAR_OK: (r'.*I\s\(([0-9]+)\) SLAVE_TEST: ([A-Z]+ [A-Z]+) \([a-zA-Z0-9_]+ us\),\s'
|
|
r'ADDR:([0-9]+), TYPE:[0-9]+, INST_ADDR:0x[a-zA-Z0-9]+, SIZE:[0-9]+'),
|
|
STACK_PAR_FAIL: (r'.*E \(([0-9]+)\) SLAVE_TEST: Response time exceeds configured [0-9]+ [ms], ignore packet.*'),
|
|
STACK_DESTROY: (r'.*I\s\(([0-9]+)\) SLAVE_TEST: (Modbus controller destroyed).')}
|
|
|
|
pattern_dict_master = {STACK_IPV4: (r'.*I \([0-9]+\) example_connect: - IPv4 address: ([0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}).*'),
|
|
STACK_IPV6: (r'.*I \([0-9]+\) example_connect: - IPv6 address: (([A-Fa-f0-9]{1,4}::?){1,7}[A-Fa-f0-9]{1,4}).*'),
|
|
STACK_INIT: (r'.*I \(([0-9]+)\) MASTER_TEST: (Modbus master stack initialized).*'),
|
|
STACK_CONNECT: (r'.*.*I\s\(([0-9]+)\) MB_TCP_MASTER_PORT: (Connected [0-9]+ slaves), start polling.*'),
|
|
STACK_START: (r'.*I \(([0-9]+)\) MASTER_TEST: (Start modbus test).*'),
|
|
STACK_PAR_OK: (r'.*I\s\(([0-9]+)\) MASTER_TEST: Characteristic #[0-9]+ ([a-zA-Z0-9_]+)'
|
|
r'\s\([a-zA-Z\%\/]+\) value = [a-zA-Z0-9\.]+ \(0x[a-zA-Z0-9]+\) read successful.*'),
|
|
STACK_PAR_FAIL: (r'.*E \(([0-9]+)\) MASTER_TEST: Characteristic #[0-9]+\s\(([a-zA-Z0-9_]+)\)\s'
|
|
r'read fail, err = [0-9]+ \([_A-Z]+\).*'),
|
|
STACK_DESTROY: (r'.*I\s\(([0-9]+)\) MASTER_TEST: (Destroy master).*')}
|
|
|
|
logger = logging.getLogger(LOGGER_NAME)
|
|
|
|
|
|
class DutTestThread(Thread):
|
|
""" Test thread class
|
|
"""
|
|
def __init__(self, dut=None, name=None, ip_addr=None, expect=None):
|
|
""" Initialize the thread parameters
|
|
"""
|
|
self.tname = name
|
|
self.dut = dut
|
|
self.expected = expect
|
|
self.data = None
|
|
self.ip_addr = ip_addr
|
|
self.test_finish = False
|
|
self.param_fail_count = 0
|
|
self.param_ok_count = 0
|
|
self.test_stage = STACK_DEFAULT
|
|
super(DutTestThread, self).__init__()
|
|
|
|
def __enter__(self):
|
|
logger.debug('Restart %s.' % self.tname)
|
|
# Reset DUT first
|
|
self.dut.reset()
|
|
# Capture output from the DUT
|
|
self.dut.start_capture_raw_data(capture_id=self.dut.name)
|
|
return self
|
|
|
|
def __exit__(self, exc_type, exc_value, traceback):
|
|
""" The exit method of context manager
|
|
"""
|
|
if exc_type is not None or exc_value is not None:
|
|
logger.info('Thread %s rised an exception type: %s, value: %s' % (self.tname, str(exc_type), str(exc_value)))
|
|
|
|
def run(self):
|
|
""" The function implements thread functionality
|
|
"""
|
|
# Initialize slave IP for master board
|
|
if (self.ip_addr is not None):
|
|
self.set_ip(0)
|
|
|
|
# Check expected strings in the listing
|
|
self.test_start(TEST_EXPECT_STR_TIMEOUT)
|
|
|
|
# Check DUT exceptions
|
|
dut_exceptions = self.dut.get_exceptions()
|
|
if 'Guru Meditation Error:' in dut_exceptions:
|
|
raise Exception('%s generated an exception: %s\n' % (str(self.dut), dut_exceptions))
|
|
|
|
# Mark thread has run to completion without any exceptions
|
|
self.data = self.dut.stop_capture_raw_data(capture_id=self.dut.name)
|
|
|
|
def set_ip(self, index=0):
|
|
""" The method to send slave IP to master application
|
|
"""
|
|
message = r'.*Waiting IP([0-9]{1,2}) from stdin.*'
|
|
# Read all data from previous restart to get prompt correctly
|
|
self.dut.read()
|
|
result = self.dut.expect(re.compile(message), TEST_EXPECT_STR_TIMEOUT)
|
|
if int(result[0]) != index:
|
|
raise Exception('Incorrect index of IP=%d for %s\n' % (int(result[0]), str(self.dut)))
|
|
message = 'IP%s=%s' % (result[0], self.ip_addr)
|
|
self.dut.write(message, '\r\n', False)
|
|
logger.debug('Sent message for %s: %s' % (self.tname, message))
|
|
message = r'.*IP\([0-9]+\) = \[([0-9a-zA-Z\.\:]+)\] set from stdin.*'
|
|
result = self.dut.expect(re.compile(message), TEST_EXPECT_STR_TIMEOUT)
|
|
logger.debug('Thread %s initialized with slave IP (%s).' % (self.tname, result[0]))
|
|
|
|
def test_start(self, timeout_value):
|
|
""" The method to initialize and handle test stages
|
|
"""
|
|
def handle_get_ip4(data):
|
|
""" Handle get_ip v4
|
|
"""
|
|
logger.debug('%s[STACK_IPV4]: %s' % (self.tname, str(data)))
|
|
self.test_stage = STACK_IPV4
|
|
|
|
def handle_get_ip6(data):
|
|
""" Handle get_ip v6
|
|
"""
|
|
logger.debug('%s[STACK_IPV6]: %s' % (self.tname, str(data)))
|
|
self.test_stage = STACK_IPV6
|
|
|
|
def handle_init(data):
|
|
""" Handle init
|
|
"""
|
|
logger.debug('%s[STACK_INIT]: %s' % (self.tname, str(data)))
|
|
self.test_stage = STACK_INIT
|
|
|
|
def handle_connect(data):
|
|
""" Handle connect
|
|
"""
|
|
logger.debug('%s[STACK_CONNECT]: %s' % (self.tname, str(data)))
|
|
self.test_stage = STACK_CONNECT
|
|
|
|
def handle_test_start(data):
|
|
""" Handle connect
|
|
"""
|
|
logger.debug('%s[STACK_START]: %s' % (self.tname, str(data)))
|
|
self.test_stage = STACK_START
|
|
|
|
def handle_par_ok(data):
|
|
""" Handle parameter ok
|
|
"""
|
|
logger.debug('%s[READ_PAR_OK]: %s' % (self.tname, str(data)))
|
|
if self.test_stage >= STACK_START:
|
|
self.param_ok_count += 1
|
|
self.test_stage = STACK_PAR_OK
|
|
|
|
def handle_par_fail(data):
|
|
""" Handle parameter fail
|
|
"""
|
|
logger.debug('%s[READ_PAR_FAIL]: %s' % (self.tname, str(data)))
|
|
self.param_fail_count += 1
|
|
self.test_stage = STACK_PAR_FAIL
|
|
|
|
def handle_destroy(data):
|
|
""" Handle destroy
|
|
"""
|
|
logger.debug('%s[DESTROY]: %s' % (self.tname, str(data)))
|
|
self.test_stage = STACK_DESTROY
|
|
self.test_finish = True
|
|
|
|
while not self.test_finish:
|
|
try:
|
|
self.dut.expect_any((re.compile(self.expected[STACK_IPV4]), handle_get_ip4),
|
|
(re.compile(self.expected[STACK_IPV6]), handle_get_ip6),
|
|
(re.compile(self.expected[STACK_INIT]), handle_init),
|
|
(re.compile(self.expected[STACK_CONNECT]), handle_connect),
|
|
(re.compile(self.expected[STACK_START]), handle_test_start),
|
|
(re.compile(self.expected[STACK_PAR_OK]), handle_par_ok),
|
|
(re.compile(self.expected[STACK_PAR_FAIL]), handle_par_fail),
|
|
(re.compile(self.expected[STACK_DESTROY]), handle_destroy),
|
|
timeout=timeout_value)
|
|
except DUT.ExpectTimeout:
|
|
logger.debug('%s, expect timeout on stage #%d (%s seconds)' % (self.tname, self.test_stage, timeout_value))
|
|
self.test_finish = True
|
|
|
|
|
|
def test_check_mode(dut=None, mode_str=None, value=None):
|
|
""" Check communication mode for dut
|
|
"""
|
|
global logger
|
|
try:
|
|
opt = dut.app.get_sdkconfig()[mode_str]
|
|
logger.debug('%s {%s} = %s.\n' % (str(dut), mode_str, opt))
|
|
return value == opt
|
|
except Exception:
|
|
logger.error('ENV_TEST_FAILURE: %s: Cannot find option %s in sdkconfig.' % (str(dut), mode_str))
|
|
return False
|
|
|
|
|
|
@ttfw_idf.idf_example_test(env_tag='Example_Modbus_TCP')
|
|
def test_modbus_communication(env, comm_mode):
|
|
global logger
|
|
|
|
rel_project_path = os.path.join('examples', 'protocols', 'modbus', 'tcp')
|
|
# Get device under test. Both duts must be able to be connected to WiFi router
|
|
dut_master = env.get_dut('modbus_tcp_master', os.path.join(rel_project_path, TEST_MASTER_TCP))
|
|
dut_slave = env.get_dut('modbus_tcp_slave', os.path.join(rel_project_path, TEST_SLAVE_TCP))
|
|
log_file = os.path.join(env.log_path, 'modbus_tcp_test.log')
|
|
print('Logging file name: %s' % log_file)
|
|
|
|
try:
|
|
# create file handler which logs even debug messages
|
|
logger.setLevel(logging.DEBUG)
|
|
fh = logging.FileHandler(log_file)
|
|
fh.setLevel(logging.DEBUG)
|
|
# set format of output for both handlers
|
|
formatter = logging.Formatter('%(levelname)s:%(message)s')
|
|
fh.setFormatter(formatter)
|
|
logger.addHandler(fh)
|
|
# create console handler
|
|
ch = logging.StreamHandler()
|
|
ch.setLevel(logging.INFO)
|
|
# set format of output for both handlers
|
|
formatter = logging.Formatter('%(levelname)s:%(message)s')
|
|
ch.setFormatter(formatter)
|
|
logger.addHandler(ch)
|
|
|
|
# Check Kconfig configuration options for each built example
|
|
if (test_check_mode(dut_master, 'CONFIG_FMB_COMM_MODE_TCP_EN', 'y') and
|
|
test_check_mode(dut_slave, 'CONFIG_FMB_COMM_MODE_TCP_EN', 'y')):
|
|
slave_name = TEST_SLAVE_TCP
|
|
master_name = TEST_MASTER_TCP
|
|
else:
|
|
logger.error('ENV_TEST_FAILURE: IP resolver mode do not match in the master and slave implementation.\n')
|
|
raise Exception('ENV_TEST_FAILURE: IP resolver mode do not match in the master and slave implementation.\n')
|
|
address = None
|
|
if test_check_mode(dut_master, 'CONFIG_MB_SLAVE_IP_FROM_STDIN', 'y'):
|
|
logger.info('ENV_TEST_INFO: Set slave IP address through STDIN.\n')
|
|
# Flash app onto DUT (Todo: Debug case when the slave flashed before master then expect does not work correctly for no reason
|
|
dut_slave.start_app()
|
|
dut_master.start_app()
|
|
if test_check_mode(dut_master, 'CONFIG_EXAMPLE_CONNECT_IPV6', 'y'):
|
|
address = dut_slave.expect(re.compile(pattern_dict_slave[STACK_IPV6]), TEST_EXPECT_STR_TIMEOUT)
|
|
else:
|
|
address = dut_slave.expect(re.compile(pattern_dict_slave[STACK_IPV4]), TEST_EXPECT_STR_TIMEOUT)
|
|
if address is not None:
|
|
print('Found IP slave address: %s' % address[0])
|
|
else:
|
|
raise Exception('ENV_TEST_FAILURE: Slave IP address is not found in the output. Check network settings.\n')
|
|
else:
|
|
raise Exception('ENV_TEST_FAILURE: Slave IP resolver is not configured correctly.\n')
|
|
|
|
# Create thread for each dut
|
|
with DutTestThread(dut=dut_master, name=master_name, ip_addr=address[0], expect=pattern_dict_master) as dut_master_thread:
|
|
with DutTestThread(dut=dut_slave, name=slave_name, ip_addr=None, expect=pattern_dict_slave) as dut_slave_thread:
|
|
|
|
# Start each thread
|
|
dut_slave_thread.start()
|
|
dut_master_thread.start()
|
|
|
|
# Wait for threads to complete
|
|
dut_slave_thread.join(timeout=TEST_THREAD_JOIN_TIMEOUT)
|
|
dut_master_thread.join(timeout=TEST_THREAD_JOIN_TIMEOUT)
|
|
|
|
if dut_slave_thread.isAlive():
|
|
logger.error('ENV_TEST_FAILURE: The thread %s is not completed successfully after %d seconds.\n' %
|
|
(dut_slave_thread.tname, TEST_THREAD_JOIN_TIMEOUT))
|
|
raise Exception('ENV_TEST_FAILURE: The thread %s is not completed successfully after %d seconds.\n' %
|
|
(dut_slave_thread.tname, TEST_THREAD_JOIN_TIMEOUT))
|
|
|
|
if dut_master_thread.isAlive():
|
|
logger.error('TEST_FAILURE: The thread %s is not completed successfully after %d seconds.\n' %
|
|
(dut_master_thread.tname, TEST_THREAD_JOIN_TIMEOUT))
|
|
raise Exception('TEST_FAILURE: The thread %s is not completed successfully after %d seconds.\n' %
|
|
(dut_master_thread.tname, TEST_THREAD_JOIN_TIMEOUT))
|
|
|
|
logger.info('TEST_INFO: %s error count = %d, %s error count = %d.\n' %
|
|
(dut_master_thread.tname, dut_master_thread.param_fail_count,
|
|
dut_slave_thread.tname, dut_slave_thread.param_fail_count))
|
|
logger.info('TEST_INFO: %s ok count = %d, %s ok count = %d.\n' %
|
|
(dut_master_thread.tname, dut_master_thread.param_ok_count,
|
|
dut_slave_thread.tname, dut_slave_thread.param_ok_count))
|
|
|
|
if ((dut_master_thread.param_fail_count > TEST_READ_MAX_ERR_COUNT) or
|
|
(dut_slave_thread.param_fail_count > TEST_READ_MAX_ERR_COUNT) or
|
|
(dut_slave_thread.param_ok_count == 0) or
|
|
(dut_master_thread.param_ok_count == 0)):
|
|
raise Exception('TEST_FAILURE: %s parameter read error(ok) count = %d(%d), %s parameter read error(ok) count = %d(%d).\n' %
|
|
(dut_master_thread.tname, dut_master_thread.param_fail_count, dut_master_thread.param_ok_count,
|
|
dut_slave_thread.tname, dut_slave_thread.param_fail_count, dut_slave_thread.param_ok_count))
|
|
logger.info('TEST_SUCCESS: The Modbus parameter test is completed successfully.\n')
|
|
|
|
finally:
|
|
dut_master.close()
|
|
dut_slave.close()
|
|
logging.shutdown()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
test_modbus_communication()
|