esp-idf/examples/wifi/power_save/pytest_wifi_power_save.py
2023-08-23 18:02:03 +08:00

78 lines
2.0 KiB
Python

# SPDX-FileCopyrightText: 2021-2023 Espressif Systems (Shanghai) CO LTD
# SPDX-License-Identifier: Apache-2.0
import logging
import pexpect
import pytest
from common_test_methods import get_env_config_variable
from pytest_embedded import Dut
bad_event_str = [
'bcn_timout',
'm f probe req l',
'abort() was called',
'Guru Meditation Error',
]
def _run_test(dut: Dut) -> None:
if dut.app.sdkconfig.get('EXAMPLE_GET_AP_INFO_FROM_STDIN') is True:
env_name = 'wifi_ap'
ap_ssid = get_env_config_variable(env_name, 'ap_ssid')
ap_password = get_env_config_variable(env_name, 'ap_password')
tag = 'power_save'
dut.expect(f'{tag}: Input SSID:', timeout=5)
dut.write(f'{ap_ssid}')
dut.expect(f'{tag}: Input Password:', timeout=5)
dut.write(f'{ap_password}')
else:
# for local test may config ssid/password from menuconfig
pass
try:
dut.expect(r'got ip: (\d+\.\d+\.\d+\.\d+)[^\d]', timeout=20)
log_after_got_ip = dut.expect(pexpect.TIMEOUT, timeout=10).decode()
if any(s in log_after_got_ip for s in bad_event_str):
logging.info('Abnormal connection log:')
logging.info('\n' + log_after_got_ip)
raise RuntimeError('Something abnormal happened after got ip')
except pexpect.exceptions.TIMEOUT:
raise RuntimeError('Failed to get ip in power save mode')
@pytest.mark.esp32
@pytest.mark.esp32c2
@pytest.mark.esp32s2
@pytest.mark.esp32c3
@pytest.mark.esp32s3
@pytest.mark.esp32c6
@pytest.mark.wifi_ap
def test_wifi_power_save(dut: Dut) -> None:
_run_test(dut)
@pytest.mark.esp32c6
@pytest.mark.wifi_ap
@pytest.mark.parametrize(
'config',
[
'pd_top',
],
indirect=True,
)
def test_wifi_power_save_pd_top(dut: Dut) -> None:
_run_test(dut)
@pytest.mark.esp32c2
@pytest.mark.wifi_ap
@pytest.mark.xtal_26mhz
@pytest.mark.parametrize(
'config, baud', [
('c2_xtal26m', '74880'),
], indirect=True
)
def test_wifi_power_save_esp32c2_26mhz(dut: Dut) -> None:
_run_test(dut)