mirror of
https://github.com/u-boot/u-boot.git
synced 2025-04-16 01:44:34 +00:00
752c376987
This fixture name is quite long and results in lots of verbose code. We know this is U-Boot so the 'u_boot_' part is not necessary. But it is also a bit of a misnomer, since it provides access to all the information available to tests. It is not just the console. It would be too confusing to use con as it would be confused with config and it is probably too short. So shorten it to 'ubman'. Signed-off-by: Simon Glass <sjg@chromium.org> Link: https://lore.kernel.org/u-boot/CAFLszTgPa4aT_J9h9pqeTtLCVn4x2JvLWRcWRD8NaN3uoSAtyA@mail.gmail.com/
116 lines
3.7 KiB
Python
116 lines
3.7 KiB
Python
# SPDX-License-Identifier: GPL-2.0
|
|
# (C) Copyright 2023, Advanced Micro Devices, Inc.
|
|
|
|
import pytest
|
|
import random
|
|
import re
|
|
|
|
"""
|
|
Note: This test relies on boardenv_* containing configuration values to define
|
|
the i2c device info including the bus list and eeprom address/value. This test
|
|
will be automatically skipped without this.
|
|
|
|
For example:
|
|
|
|
# Setup env__i2c_device_test to set the i2c bus list and probe_all boolean
|
|
# parameter. For i2c_probe_all_buses case, if probe_all parameter is set to
|
|
# False then it probes all the buses listed in bus_list instead of probing all
|
|
# the buses available.
|
|
env__i2c_device_test = {
|
|
'bus_list': [0, 2, 5, 12, 16, 18],
|
|
'probe_all': False,
|
|
}
|
|
|
|
# Setup env__i2c_eeprom_device_test to set the i2c bus number, eeprom address
|
|
# and configured value for i2c_eeprom test case. Test will be skipped if
|
|
# env__i2c_eeprom_device_test is not set
|
|
env__i2c_eeprom_device_test = {
|
|
'bus': 3,
|
|
'eeprom_addr': 0x54,
|
|
'eeprom_val': '30 31',
|
|
}
|
|
"""
|
|
|
|
def get_i2c_test_env(ubman):
|
|
f = ubman.config.env.get("env__i2c_device_test", None)
|
|
if not f:
|
|
pytest.skip("No I2C device to test!")
|
|
else:
|
|
bus_list = f.get("bus_list", None)
|
|
if not bus_list:
|
|
pytest.skip("I2C bus list is not provided!")
|
|
probe_all = f.get("probe_all", False)
|
|
return bus_list, probe_all
|
|
|
|
@pytest.mark.buildconfigspec("cmd_i2c")
|
|
def test_i2c_bus(ubman):
|
|
bus_list, probe = get_i2c_test_env(ubman)
|
|
bus = random.choice(bus_list)
|
|
expected_response = f"Bus {bus}:"
|
|
response = ubman.run_command("i2c bus")
|
|
assert expected_response in response
|
|
|
|
@pytest.mark.buildconfigspec("cmd_i2c")
|
|
def test_i2c_dev(ubman):
|
|
bus_list, probe = get_i2c_test_env(ubman)
|
|
expected_response = "Current bus is"
|
|
response = ubman.run_command("i2c dev")
|
|
assert expected_response in response
|
|
|
|
@pytest.mark.buildconfigspec("cmd_i2c")
|
|
def test_i2c_probe(ubman):
|
|
bus_list, probe = get_i2c_test_env(ubman)
|
|
bus = random.choice(bus_list)
|
|
expected_response = f"Setting bus to {bus}"
|
|
response = ubman.run_command(f"i2c dev {bus}")
|
|
assert expected_response in response
|
|
expected_response = "Valid chip addresses:"
|
|
response = ubman.run_command("i2c probe")
|
|
assert expected_response in response
|
|
|
|
@pytest.mark.buildconfigspec("cmd_i2c")
|
|
def test_i2c_eeprom(ubman):
|
|
f = ubman.config.env.get("env__i2c_eeprom_device_test", None)
|
|
if not f:
|
|
pytest.skip("No I2C eeprom to test!")
|
|
|
|
bus = f.get("bus", 0)
|
|
if bus < 0:
|
|
pytest.fail("No bus specified via env__i2c_eeprom_device_test!")
|
|
|
|
addr = f.get("eeprom_addr", -1)
|
|
if addr < 0:
|
|
pytest.fail("No eeprom address specified via env__i2c_eeprom_device_test!")
|
|
|
|
value = f.get("eeprom_val")
|
|
if not value:
|
|
pytest.fail(
|
|
"No eeprom configured value provided via env__i2c_eeprom_device_test!"
|
|
)
|
|
|
|
# Enable i2c mux bridge
|
|
ubman.run_command("i2c dev %x" % bus)
|
|
ubman.run_command("i2c probe")
|
|
output = ubman.run_command("i2c md %x 0 5" % addr)
|
|
assert value in output
|
|
|
|
@pytest.mark.buildconfigspec("cmd_i2c")
|
|
def test_i2c_probe_all_buses(ubman):
|
|
bus_list, probe = get_i2c_test_env(ubman)
|
|
bus = random.choice(bus_list)
|
|
expected_response = f"Bus {bus}:"
|
|
response = ubman.run_command("i2c bus")
|
|
assert expected_response in response
|
|
|
|
# Get all the bus list
|
|
if probe:
|
|
buses = re.findall("Bus (.+?):", response)
|
|
bus_list = [int(x) for x in buses]
|
|
|
|
for dev in bus_list:
|
|
expected_response = f"Setting bus to {dev}"
|
|
response = ubman.run_command(f"i2c dev {dev}")
|
|
assert expected_response in response
|
|
expected_response = "Valid chip addresses:"
|
|
response = ubman.run_command("i2c probe")
|
|
assert expected_response in response
|