http examples pytest migration

examples/protocols/http_server/advanced_tests/pytest_http_server_advanced.py

@@ -1,27 +1,23 @@
 #!/usr/bin/env python
 #
-# Copyright 2018 Espressif Systems (Shanghai) PTE LTD
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# SPDX-FileCopyrightText: 2018-2022 Espressif Systems (Shanghai) CO LTD
+# SPDX-License-Identifier: Apache-2.0
 
 from __future__ import division, print_function, unicode_literals
 
+import logging
 import os
-import re
+import sys
 
-import ttfw_idf
-from idf_http_server_test import test as client
-from tiny_test_fw import Utility
+import pytest
+
+try:
+    from idf_http_server_test import test as client
+except ModuleNotFoundError:
+    sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..', '..', '..', 'tools', 'ci', 'python_packages'))
+    from idf_http_server_test import test as client
+
+from pytest_embedded import Dut
 
 # When running on local machine execute the following before running this script
 # > make app bootloader
@@ -35,46 +31,47 @@ from tiny_test_fw import Utility
 # features to this component.
 
 
-@ttfw_idf.idf_example_test(env_tag='Example_WIFI_Protocols')
-def test_examples_protocol_http_server_advanced(env, extra_data):
-    # Acquire DUT
-    dut1 = env.get_dut('http_server', 'examples/protocols/http_server/advanced_tests', dut_class=ttfw_idf.ESP32DUT)
+@pytest.mark.esp32
+@pytest.mark.esp32c3
+@pytest.mark.esp32s2
+@pytest.mark.esp32s3
+@pytest.mark.wifi
+def test_examples_protocol_http_server_advanced(dut: Dut) -> None:
 
     # Get binary file
-    binary_file = os.path.join(dut1.app.binary_path, 'tests.bin')
+    binary_file = os.path.join(dut.app.binary_path, 'tests.bin')
     bin_size = os.path.getsize(binary_file)
-    ttfw_idf.log_performance('http_server_bin_size', '{}KB'.format(bin_size // 1024))
+    logging.info('http_server_bin_size : {}KB'.format(bin_size // 1024))
 
-    # Upload binary and start testing
-    Utility.console_log('Starting http_server advanced test app')
-    dut1.start_app()
+    logging.info('Starting http_server advanced test 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]
+    logging.info('Waiting to connect with AP')
+    got_ip = dut.expect(r'IPv4 address: (\d+\.\d+\.\d+\.\d+)', timeout=30)[1].decode()
 
-    got_port = dut1.expect(re.compile(r"(?:[\s\S]*)Started HTTP server on port: '(\d+)'"), timeout=15)[0]
-    result = dut1.expect(re.compile(r"(?:[\s\S]*)Max URI handlers: '(\d+)'(?:[\s\S]*)Max Open Sessions: "  # noqa: W605
-                                    r"'(\d+)'(?:[\s\S]*)Max Header Length: '(\d+)'(?:[\s\S]*)Max URI Length: "
-                                    r"'(\d+)'(?:[\s\S]*)Max Stack Size: '(\d+)'"), timeout=15)
-    # max_uri_handlers = int(result[0])
-    max_sessions = int(result[1])
-    max_hdr_len = int(result[2])
-    max_uri_len = int(result[3])
-    max_stack_size = int(result[4])
+    got_port = dut.expect(r"(?:[\s\S]*)Started HTTP server on port: '(\d+)'", timeout=30)[1].decode()
 
-    Utility.console_log('Got IP   : ' + got_ip)
-    Utility.console_log('Got Port : ' + got_port)
+    result = dut.expect(r"(?:[\s\S]*)Max URI handlers: '(\d+)'(?:[\s\S]*)Max Open Sessions: "  # noqa: W605
+                        r"'(\d+)'(?:[\s\S]*)Max Header Length: '(\d+)'(?:[\s\S]*)Max URI Length: "
+                        r"'(\d+)'(?:[\s\S]*)Max Stack Size: '(\d+)'", timeout=15)
+    # max_uri_handlers = int(result[1])
+    max_sessions = int(result[2])
+    max_hdr_len = int(result[3])
+    max_uri_len = int(result[4])
+    max_stack_size = int(result[5])
+
+    logging.info('Got Port : {}'.format(got_port))
+    logging.info('Got IP   : {}'.format(got_ip))
 
     # Run test script
     # If failed raise appropriate exception
     failed = False
 
-    Utility.console_log('Sessions and Context Tests...')
+    logging.info('Sessions and Context Tests...')
     if not client.spillover_session(got_ip, got_port, max_sessions):
-        Utility.console_log('Ignoring failure')
+        logging.info('Ignoring failure')
     if not client.parallel_sessions_adder(got_ip, got_port, max_sessions):
-        Utility.console_log('Ignoring failure')
+        logging.info('Ignoring failure')
     if not client.leftover_data_test(got_ip, got_port):
         failed = True
     if not client.async_response_test(got_ip, got_port):
@@ -87,19 +84,19 @@ def test_examples_protocol_http_server_advanced(env, extra_data):
     # This test fails a lot! Enable when connection is stable
     # test_size = 50*1024 # 50KB
     # if not client.packet_size_limit_test(got_ip, got_port, test_size):
-    #    Utility.console_log("Ignoring failure")
+    #    logging.info("Ignoring failure")
 
-    Utility.console_log('Getting initial stack usage...')
+    logging.info('Getting initial stack usage...')
     if not client.get_hello(got_ip, got_port):
         failed = True
 
-    inital_stack = int(dut1.expect(re.compile(r"(?:[\s\S]*)Free Stack for server task: '(\d+)'"), timeout=15)[0])
+    inital_stack = int(dut.expect(r"(?:[\s\S]*)Free Stack for server task: '(\d+)'", timeout=15)[1])
 
     if inital_stack < 0.1 * max_stack_size:
-        Utility.console_log('More than 90% of stack being used on server start')
+        logging.info('More than 90% of stack being used on server start')
         failed = True
 
-    Utility.console_log('Basic HTTP Client Tests...')
+    logging.info('Basic HTTP Client Tests...')
     if not client.get_hello(got_ip, got_port):
         failed = True
     if not client.post_hello(got_ip, got_port):
@@ -121,7 +118,7 @@ def test_examples_protocol_http_server_advanced(env, extra_data):
     if not client.get_test_headers(got_ip, got_port):
         failed = True
 
-    Utility.console_log('Error code tests...')
+    logging.info('Error code tests...')
     if not client.code_500_server_error_test(got_ip, got_port):
         failed = True
     if not client.code_501_method_not_impl(got_ip, got_port):
@@ -137,25 +134,21 @@ def test_examples_protocol_http_server_advanced(env, extra_data):
     if not client.code_408_req_timeout(got_ip, got_port):
         failed = True
     if not client.code_414_uri_too_long(got_ip, got_port, max_uri_len):
-        Utility.console_log('Ignoring failure')
+        logging.info('Ignoring failure')
     if not client.code_431_hdr_too_long(got_ip, got_port, max_hdr_len):
-        Utility.console_log('Ignoring failure')
+        logging.info('Ignoring failure')
     if not client.test_upgrade_not_supported(got_ip, got_port):
         failed = True
 
-    Utility.console_log('Getting final stack usage...')
+    logging.info('Getting final stack usage...')
     if not client.get_hello(got_ip, got_port):
         failed = True
 
-    final_stack = int(dut1.expect(re.compile(r"(?:[\s\S]*)Free Stack for server task: '(\d+)'"), timeout=15)[0])
+    final_stack = int(dut.expect(r"(?:[\s\S]*)Free Stack for server task: '(\d+)'", timeout=15)[1])
 
     if final_stack < 0.05 * max_stack_size:
-        Utility.console_log('More than 95% of stack got used during tests')
+        logging.info('More than 95% of stack got used during tests')
         failed = True
 
     if failed:
         raise RuntimeError
-
-
-if __name__ == '__main__':
-    test_examples_protocol_http_server_advanced()