tools: binman: change shebang from python into python2
[oweals/u-boot.git] / test / dm / eth.c
index 196eba85a2bbc860ce3ded5bf94ed1c7a7f4305b..6288ae24abb4467580bb5e1cbd1515c59a6f69b5 100644 (file)
@@ -9,16 +9,20 @@
 
 #include <common.h>
 #include <dm.h>
-#include <dm/test.h>
-#include <dm/ut.h>
 #include <fdtdec.h>
 #include <malloc.h>
 #include <net.h>
+#include <dm/test.h>
+#include <dm/device-internal.h>
+#include <dm/uclass-internal.h>
 #include <asm/eth.h>
+#include <test/ut.h>
 
 DECLARE_GLOBAL_DATA_PTR;
 
-static int dm_test_eth(struct dm_test_state *dms)
+#define DM_TEST_ETH_NUM                4
+
+static int dm_test_eth(struct unit_test_state *uts)
 {
        net_ping_ip = string_to_ip("1.1.2.2");
 
@@ -38,7 +42,7 @@ static int dm_test_eth(struct dm_test_state *dms)
 }
 DM_TEST(dm_test_eth, DM_TESTF_SCAN_FDT);
 
-static int dm_test_eth_alias(struct dm_test_state *dms)
+static int dm_test_eth_alias(struct unit_test_state *uts)
 {
        net_ping_ip = string_to_ip("1.1.2.2");
        setenv("ethact", "eth0");
@@ -62,7 +66,7 @@ static int dm_test_eth_alias(struct dm_test_state *dms)
 }
 DM_TEST(dm_test_eth_alias, DM_TESTF_SCAN_FDT);
 
-static int dm_test_eth_prime(struct dm_test_state *dms)
+static int dm_test_eth_prime(struct unit_test_state *uts)
 {
        net_ping_ip = string_to_ip("1.1.2.2");
 
@@ -82,15 +86,69 @@ static int dm_test_eth_prime(struct dm_test_state *dms)
 }
 DM_TEST(dm_test_eth_prime, DM_TESTF_SCAN_FDT);
 
-static int dm_test_eth_rotate(struct dm_test_state *dms)
+/**
+ * This test case is trying to test the following scenario:
+ *     - All ethernet devices are not probed
+ *     - "ethaddr" for all ethernet devices are not set
+ *     - "ethact" is set to a valid ethernet device name
+ *
+ * With Sandbox default test configuration, all ethernet devices are
+ * probed after power-up, so we have to manually create such scenario:
+ *     - Remove all ethernet devices
+ *     - Remove all "ethaddr" environment variables
+ *     - Set "ethact" to the first ethernet device
+ *
+ * Do a ping test to see if anything goes wrong.
+ */
+static int dm_test_eth_act(struct unit_test_state *uts)
 {
-       char ethaddr[18];
+       struct udevice *dev[DM_TEST_ETH_NUM];
+       const char *ethname[DM_TEST_ETH_NUM] = {"eth@10002000", "eth@10003000",
+                                               "sbe5", "eth@10004000"};
+       const char *addrname[DM_TEST_ETH_NUM] = {"ethaddr", "eth5addr",
+                                                "eth3addr", "eth1addr"};
+       char ethaddr[DM_TEST_ETH_NUM][18];
+       int i;
 
-       /* Invalidate eth1's MAC address */
        net_ping_ip = string_to_ip("1.1.2.2");
-       strcpy(ethaddr, getenv("eth1addr"));
-       setenv("eth1addr", NULL);
 
+       /* Prepare the test scenario */
+       for (i = 0; i < DM_TEST_ETH_NUM; i++) {
+               ut_assertok(uclass_find_device_by_name(UCLASS_ETH,
+                                                      ethname[i], &dev[i]));
+               ut_assertok(device_remove(dev[i]));
+
+               /* Invalidate MAC address */
+               strcpy(ethaddr[i], getenv(addrname[i]));
+               /* Must disable access protection for ethaddr before clearing */
+               setenv(".flags", addrname[i]);
+               setenv(addrname[i], NULL);
+       }
+
+       /* Set ethact to "eth@10002000" */
+       setenv("ethact", ethname[0]);
+
+       /* Segment fault might happen if something is wrong */
+       ut_asserteq(-ENODEV, net_loop(PING));
+
+       for (i = 0; i < DM_TEST_ETH_NUM; i++) {
+               /* Restore the env */
+               setenv(".flags", addrname[i]);
+               setenv(addrname[i], ethaddr[i]);
+
+               /* Probe the device again */
+               ut_assertok(device_probe(dev[i]));
+       }
+       setenv(".flags", NULL);
+       setenv("ethact", NULL);
+
+       return 0;
+}
+DM_TEST(dm_test_eth_act, DM_TESTF_SCAN_FDT);
+
+/* The asserts include a return on fail; cleanup in the caller */
+static int _dm_test_eth_rotate1(struct unit_test_state *uts)
+{
        /* Make sure that the default is to rotate to the next interface */
        setenv("ethact", "eth@10004000");
        ut_assertok(net_loop(PING));
@@ -102,32 +160,66 @@ static int dm_test_eth_rotate(struct dm_test_state *dms)
        ut_asserteq(-EINVAL, net_loop(PING));
        ut_asserteq_str("eth@10004000", getenv("ethact"));
 
-       /* Restore the env */
-       setenv("eth1addr", ethaddr);
-       setenv("ethrotate", NULL);
-
-       /* Invalidate eth0's MAC address */
-       strcpy(ethaddr, getenv("ethaddr"));
-       setenv(".flags", "ethaddr");
-       setenv("ethaddr", NULL);
+       return 0;
+}
 
+static int _dm_test_eth_rotate2(struct unit_test_state *uts)
+{
        /* Make sure we can skip invalid devices */
        setenv("ethact", "eth@10004000");
        ut_assertok(net_loop(PING));
        ut_asserteq_str("eth@10004000", getenv("ethact"));
 
-       /* Restore the env */
-       setenv("ethaddr", ethaddr);
-       setenv(".flags", NULL);
+       /* Make sure we can handle device name which is not eth# */
+       setenv("ethact", "sbe5");
+       ut_assertok(net_loop(PING));
+       ut_asserteq_str("sbe5", getenv("ethact"));
 
        return 0;
 }
-DM_TEST(dm_test_eth_rotate, DM_TESTF_SCAN_FDT);
 
-static int dm_test_net_retry(struct dm_test_state *dms)
+static int dm_test_eth_rotate(struct unit_test_state *uts)
 {
+       char ethaddr[18];
+       int retval;
+
+       /* Set target IP to mock ping */
        net_ping_ip = string_to_ip("1.1.2.2");
 
+       /* Invalidate eth1's MAC address */
+       strcpy(ethaddr, getenv("eth1addr"));
+       /* Must disable access protection for eth1addr before clearing */
+       setenv(".flags", "eth1addr");
+       setenv("eth1addr", NULL);
+
+       retval = _dm_test_eth_rotate1(uts);
+
+       /* Restore the env */
+       setenv("eth1addr", ethaddr);
+       setenv("ethrotate", NULL);
+
+       if (!retval) {
+               /* Invalidate eth0's MAC address */
+               strcpy(ethaddr, getenv("ethaddr"));
+               /* Must disable access protection for ethaddr before clearing */
+               setenv(".flags", "ethaddr");
+               setenv("ethaddr", NULL);
+
+               retval = _dm_test_eth_rotate2(uts);
+
+               /* Restore the env */
+               setenv("ethaddr", ethaddr);
+       }
+       /* Restore the env */
+       setenv(".flags", NULL);
+
+       return retval;
+}
+DM_TEST(dm_test_eth_rotate, DM_TESTF_SCAN_FDT);
+
+/* The asserts include a return on fail; cleanup in the caller */
+static int _dm_test_net_retry(struct unit_test_state *uts)
+{
        /*
         * eth1 is disabled and netretry is yes, so the ping should succeed and
         * the active device should be eth0
@@ -149,10 +241,21 @@ static int dm_test_net_retry(struct dm_test_state *dms)
        ut_asserteq(-ETIMEDOUT, net_loop(PING));
        ut_asserteq_str("eth@10004000", getenv("ethact"));
 
+       return 0;
+}
+
+static int dm_test_net_retry(struct unit_test_state *uts)
+{
+       int retval;
+
+       net_ping_ip = string_to_ip("1.1.2.2");
+
+       retval = _dm_test_net_retry(uts);
+
        /* Restore the env */
        setenv("netretry", NULL);
        sandbox_eth_disable_response(1, false);
 
-       return 0;
+       return retval;
 }
 DM_TEST(dm_test_net_retry, DM_TESTF_SCAN_FDT);