esp_eth: new test_app: emac layer tests

components/esp_eth/test_apps/component_ut_test.py

@@ -0,0 +1,102 @@
+import os
+import re
+import socket
+
+import tiny_test_fw
+import ttfw_idf
+from ttfw_idf import TestFormat
+
+try:
+    import typing  # noqa: F401 # pylint: disable=unused-import
+except ImportError:
+    pass
+
+
+def configure_eth_if(func):         # type: (typing.Any) -> typing.Any
+    def inner(*args, **kwargs):     # type: (typing.Any, typing.Any) -> typing.Any
+        # try to determine which interface to use
+        netifs = os.listdir('/sys/class/net/')
+        target_if = ''
+        print('detected interfaces: ' + str(netifs))
+        for netif in netifs:
+            if netif.find('eth') == 0 or netif.find('enp') == 0 or netif.find('eno') == 0:
+                target_if = netif
+                break
+        if target_if == '':
+            raise Exception('no network interface found')
+        print('Use ' + target_if + ' for testing')
+        so = socket.socket(socket.AF_PACKET, socket.SOCK_RAW, 0x2222)
+        so.bind((target_if, 0))
+
+        func(so, *args, **kwargs)
+
+        so.close()
+
+    return inner
+
+
+@configure_eth_if
+def check_eth_recv_packet(so):    # type: (socket.socket) -> None
+    so.settimeout(10)
+    try:
+        pkt = so.recv(1024)
+        for i in range(128, 1024):
+            if pkt[i] != i & 0xff:
+                raise Exception('Packet content mismatch')
+    except Exception as e:
+        raise e
+
+
+@configure_eth_if
+def send_eth_packet(so, mac):    # type: (socket.socket, bytes) -> None
+    so.settimeout(10)
+    pkt = bytearray()
+    pkt += mac                      # dest
+    pkt += so.getsockname()[4]      # src
+    pkt += bytes.fromhex('2222')    # proto
+    pkt += bytes(1010)              # padding to 1024
+    for i in range(128, 1024):
+        pkt[i] = i & 0xff
+    try:
+        so.send(pkt)
+    except Exception as e:
+        raise e
+
+
+def test_component_ut_esp_eth(env, appname):  # type: (tiny_test_fw.Env, str) -> None
+    dut = env.get_dut('esp_eth', 'components/esp_eth/test_apps', app_config_name=appname)
+    dut.start_app()
+    stdout = dut.expect('Press ENTER to see the list of tests', full_stdout=True)
+    dut.write('"start_and_stop"')
+    stdout += dut.expect("Enter next test, or 'enter' to see menu", full_stdout=True)
+    ttfw_idf.ComponentUTResult.parse_result(stdout, test_format=TestFormat.UNITY_BASIC)
+    dut.write('"get_set_mac"')
+    stdout = dut.expect("Enter next test, or 'enter' to see menu", full_stdout=True)
+    ttfw_idf.ComponentUTResult.parse_result(stdout, test_format=TestFormat.UNITY_BASIC)
+    dut.write('"ethernet_broadcast_transmit"')
+    check_eth_recv_packet()
+    stdout = dut.expect("Enter next test, or 'enter' to see menu", full_stdout=True)
+    ttfw_idf.ComponentUTResult.parse_result(stdout, test_format=TestFormat.UNITY_BASIC)
+    dut.write('"recv_pkt"')
+    expect_result = dut.expect(re.compile(r'([\s\S]*)DUT MAC: ([0-9a-zA-Z:]*)'), timeout=10)
+    stdout = expect_result[0]
+    send_eth_packet(bytes.fromhex('ffffffffffff'))      # broadcast frame
+    send_eth_packet(bytes.fromhex('010000000000'))      # multicast frame
+    send_eth_packet(bytes.fromhex(expect_result[1].replace(':', '')))   # unicast frame
+    stdout += dut.expect("Enter next test, or 'enter' to see menu", full_stdout=True)
+    ttfw_idf.ComponentUTResult.parse_result(stdout, test_format=TestFormat.UNITY_BASIC)
+
+
+@ttfw_idf.idf_component_unit_test(env_tag='COMPONENT_UT_IP101', target=['esp32'])
+def test_component_ut_esp_eth_ip101(env, _):  # type: (tiny_test_fw.Env, typing.Any) -> None
+    test_component_ut_esp_eth(env, 'ip101')
+
+
+@ttfw_idf.idf_component_unit_test(env_tag='COMPONENT_UT_LAN8720', target=['esp32'])
+def test_component_ut_esp_eth_lan8720(env, _):  # type: (tiny_test_fw.Env, typing.Any) -> None
+    test_component_ut_esp_eth(env, 'lan8720')
+
+
+if __name__ == '__main__':
+    test_component_ut_esp_eth_ip101()
+    test_component_ut_esp_eth_lan8720()