test/py: Add basic i2c tests

This was tested on zcu102.

Signed-off-by: Michal Simek <michal.simek@xilinx.com>
State: pending

diff --git a/test/py/tests/test_i2c.py b/test/py/tests/test_i2c.py
new file mode 100644 (file)
index 0000000..0b80504
--- /dev/null
+++ b/test/py/tests/test_i2c.py
@@ -0,0 +1,73 @@
+# Copyright (c) 2015 Stephen Warren
+#
+# SPDX-License-Identifier: GPL-2.0
+
+import pytest
+import random
+
+"""
+Note: This test doesn't rely on boardenv_* configuration value but they can
+change test behavior.
+
+# Setup env__i2c_device_test_skip to True if tests with i2c devices should be
+# skipped. For example: Missing QEMU model or broken i2c device
+env__i2c_device_test_skip = True
+
+"""
+
+@pytest.mark.buildconfigspec("cmd_i2c")
+def test_i2c_bus(u_boot_console):
+    expected_response = "Bus"
+    response = u_boot_console.run_command("i2c bus")
+    assert(expected_response in response)
+
+@pytest.mark.buildconfigspec("cmd_i2c")
+def test_i2c_dev(u_boot_console):
+    expected_response = "Current bus"
+    response = u_boot_console.run_command("i2c dev")
+    assert(expected_response in response)
+
+@pytest.mark.buildconfigspec("cmd_i2c")
+def test_i2c_probe(u_boot_console):
+    expected_response = "Setting bus to 0"
+    response = u_boot_console.run_command("i2c dev 0")
+    assert(expected_response in response)
+    expected_response = "Valid chip addresses:"
+    response = u_boot_console.run_command("i2c probe")
+    assert(expected_response in response)
+
+@pytest.mark.xfail
+@pytest.mark.boardspec("zynq_zc702")
+@pytest.mark.boardspec("zynq_zc706")
+@pytest.mark.buildconfigspec("cmd_i2c")
+def test_i2c_probe_zc70x(u_boot_console):
+    test_skip = u_boot_console.config.env.get('env__i2c_device_test_skip', False)
+    if test_skip:
+        pytest.skip('i2c device test skipped')
+
+    # Enable i2c mux bridge
+    u_boot_console.run_command("i2c mw 74 0 4")
+    u_boot_console.run_command("i2c probe")
+    val = format(random.randint(0,255), '02x')
+    u_boot_console.run_command("i2c mw 54 0 " + val + " 5")
+    response = u_boot_console.run_command("i2c md 54 0 5")
+    expected_response = "0000: " + val + " " + val + " " + val + " " + val + " " + val + " "
+    assert(expected_response in response)
+
+@pytest.mark.boardspec("xilinx_zynqmp_zcu102_rev1_0")
+@pytest.mark.boardspec("xilinx_zynqmp_zcu102_revA")
+@pytest.mark.buildconfigspec("cmd_i2c")
+def test_i2c_probe_zcu102(u_boot_console):
+    test_skip = u_boot_console.config.env.get('env__i2c_device_test_skip', False)
+    if test_skip:
+        pytest.skip('i2c device test skipped')
+
+    # This is using i2c mux wiring from config file
+    u_boot_console.run_command("i2c dev 5")
+    u_boot_console.run_command("i2c probe")
+    val = format(random.randint(0,255), '02x')
+    u_boot_console.run_command("i2c mw 54 0 " + val + " 5")
+    response = u_boot_console.run_command("i2c md 54 0 5")
+    expected_response = "0000: " + val + " " + val + " " + val + " " + val + " " + val + " "
+    print (expected_response)
+    assert(expected_response in response)