def test_examples_protocol_http_server_persistence(env, extra_data): # Acquire DUT dut1 = env.get_dut('http_server', 'examples/protocols/http_server/persistent_sockets', dut_class=ttfw_idf.ESP32DUT) # Get binary file binary_file = os.path.join(dut1.app.binary_path, 'persistent_sockets.bin') bin_size = os.path.getsize(binary_file) ttfw_idf.log_performance('http_server_bin_size', '{}KB'.format(bin_size // 1024)) # Upload binary and start testing Utility.console_log('Starting http_server persistance test app') dut1.start_app() # Parse IP address of STA Utility.console_log('Waiting to connect with AP') got_ip = dut1.expect( re.compile(r'(?:[\s\S]*)IPv4 address: (\d+.\d+.\d+.\d+)'), timeout=30)[0] got_port = dut1.expect( re.compile(r"(?:[\s\S]*)Starting server on port: '(\d+)'"), timeout=30)[0] Utility.console_log('Got IP : ' + got_ip) Utility.console_log('Got Port : ' + got_port) # Expected Logs dut1.expect('Registering URI handlers', timeout=30) # Run test script conn = client.start_session(got_ip, got_port) visitor = 0 adder = 0 # Test PUT request and initialize session context num = random.randint(0, 100) client.putreq(conn, '/adder', str(num)) visitor += 1 dut1.expect('/adder visitor count = ' + str(visitor), timeout=30) dut1.expect('/adder PUT handler read ' + str(num), timeout=30) dut1.expect('PUT allocating new session', timeout=30) # Retest PUT request and change session context value num = random.randint(0, 100) Utility.console_log('Adding: ' + str(num)) client.putreq(conn, '/adder', str(num)) visitor += 1 adder += num dut1.expect('/adder visitor count = ' + str(visitor), timeout=30) dut1.expect('/adder PUT handler read ' + str(num), timeout=30) try: # Re allocation shouldn't happen dut1.expect('PUT allocating new session', timeout=30) # Not expected raise RuntimeError except Exception: # As expected pass # Test POST request and session persistence random_nums = [random.randint(0, 100) for _ in range(100)] for num in random_nums: Utility.console_log('Adding: ' + str(num)) client.postreq(conn, '/adder', str(num)) visitor += 1 adder += num dut1.expect('/adder visitor count = ' + str(visitor), timeout=30) dut1.expect('/adder handler read ' + str(num), timeout=30) # Test GET request and session persistence Utility.console_log('Matching final sum: ' + str(adder)) if client.getreq(conn, '/adder').decode() != str(adder): raise RuntimeError visitor += 1 dut1.expect('/adder visitor count = ' + str(visitor), timeout=30) dut1.expect('/adder GET handler send ' + str(adder), timeout=30) Utility.console_log('Ending session') # Close connection and check for invocation of context "Free" function client.end_session(conn) dut1.expect('/adder Free Context function called', timeout=30) Utility.console_log('Validating user context data') # Start another session to check user context data client.start_session(got_ip, got_port) num = random.randint(0, 100) client.putreq(conn, '/adder', str(num)) visitor += 1 dut1.expect('/adder visitor count = ' + str(visitor), timeout=30) dut1.expect('/adder PUT handler read ' + str(num), timeout=30) dut1.expect('PUT allocating new session', timeout=30) client.end_session(conn) dut1.expect('/adder Free Context function called', timeout=30)
def test_examples_protocol_http_server_persistence(dut: Dut) -> None: # Get binary file binary_file = os.path.join(dut.app.binary_path, 'persistent_sockets.bin') bin_size = os.path.getsize(binary_file) logging.info('http_server_bin_size : {}KB'.format(bin_size // 1024)) # Upload binary and start testing logging.info('Starting http_server persistance test app') # Parse IP address of STA logging.info('Waiting to connect with AP') got_ip = dut.expect(r'IPv4 address: (\d+\.\d+\.\d+\.\d+)', timeout=30)[1].decode() got_port = dut.expect(r"(?:[\s\S]*)Starting server on port: '(\d+)'", timeout=30)[1].decode() logging.info('Got IP : {}'.format(got_ip)) logging.info('Got Port : {}'.format(got_port)) # Expected Logs dut.expect('Registering URI handlers', timeout=30) # Run test script conn = client.start_session(got_ip, got_port) visitor = 0 adder = 0 # Test PUT request and initialize session context num = random.randint(0,100) client.putreq(conn, '/adder', str(num)) visitor += 1 dut.expect('/adder visitor count = ' + str(visitor), timeout=30) dut.expect('/adder PUT handler read ' + str(num), timeout=30) dut.expect('PUT allocating new session', timeout=30) # Retest PUT request and change session context value num = random.randint(0,100) logging.info('Adding: {}'.format(num)) client.putreq(conn, '/adder', str(num)) visitor += 1 adder += num dut.expect('/adder visitor count = ' + str(visitor), timeout=30) dut.expect('/adder PUT handler read ' + str(num), timeout=30) try: # Re allocation shouldn't happen dut.expect('PUT allocating new session', timeout=30) # Not expected raise RuntimeError except Exception: # As expected pass # Test POST request and session persistence random_nums = [random.randint(0,100) for _ in range(100)] for num in random_nums: logging.info('Adding: {}'.format(num)) client.postreq(conn, '/adder', str(num)) visitor += 1 adder += num dut.expect('/adder visitor count = ' + str(visitor), timeout=30) dut.expect('/adder handler read ' + str(num), timeout=30) # Test GET request and session persistence logging.info('Matching final sum: {}'.format(adder)) if client.getreq(conn, '/adder').decode() != str(adder): raise RuntimeError visitor += 1 dut.expect('/adder visitor count = ' + str(visitor), timeout=30) dut.expect('/adder GET handler send ' + str(adder), timeout=30) logging.info('Ending session') # Close connection and check for invocation of context "Free" function client.end_session(conn) dut.expect('/adder Free Context function called', timeout=30) logging.info('Validating user context data') # Start another session to check user context data client.start_session(got_ip, got_port) num = random.randint(0,100) client.putreq(conn, '/adder', str(num)) visitor += 1 dut.expect('/adder visitor count = ' + str(visitor), timeout=30) dut.expect('/adder PUT handler read ' + str(num), timeout=30) dut.expect('PUT allocating new session', timeout=30) client.end_session(conn) dut.expect('/adder Free Context function called', timeout=30)