Introduce list of tests and loop on it in main body.

Tests will now just take care of calling setup with a list of
"units" they need, and return 0 for failure, 1 for success, 2 if
the test had to be skipped.

Main script body will take care of displaying results and
cleaning up after every test. Introduce guard variable so that
we don't clean up twice in case of interrupts or unexpected
failures.

Signed-off-by: Stefano Brivio <sbri...@redhat.com>
---
 tools/testing/selftests/net/pmtu.sh | 58 ++++++++++++++++++++++---------------
 1 file changed, 34 insertions(+), 24 deletions(-)

diff --git a/tools/testing/selftests/net/pmtu.sh 
b/tools/testing/selftests/net/pmtu.sh
index fc79cc6f49a6..feebb6fcff21 100755
--- a/tools/testing/selftests/net/pmtu.sh
+++ b/tools/testing/selftests/net/pmtu.sh
@@ -5,13 +5,15 @@
 #
 # Tests currently implemented:
 #
-# - test_pmtu_vti6_exception
+# - pmtu_vti6_exception
 #      Set up vti6 tunnel on top of veth, with xfrm states and policies, in two
 #      namespaces with matching endpoints. Check that route exception is
 #      created by exceeding link layer MTU with ping to other endpoint. Then
 #      decrease and increase MTU of tunnel, checking that route exception PMTU
 #      changes accordingly
 
+tests="pmtu_vti6_exception"
+
 NS_A="ns-$(mktemp -u XXXXXX)"
 NS_B="ns-$(mktemp -u XXXXXX)"
 ns_a="ip netns exec ${NS_A}"
@@ -25,6 +27,8 @@ vti6_a_addr="fd00:2::a"
 vti6_b_addr="fd00:2::b"
 vti6_mask="64"
 
+cleanup_done=1
+
 setup_namespaces() {
        ip netns add ${NS_A} || return 0
        ip netns add ${NS_B}
@@ -75,26 +79,21 @@ setup_xfrm() {
 }
 
 setup() {
-       tunnel_type="$1"
+       [ "$(id -u)" -ne 0 ] && echo "  need to run as root" && return 0
 
-       [ "$(id -u)" -ne 0 ] && echo "SKIP: need to run as root" && exit 0
-
-       setup_namespaces && echo "SKIP: namespaces not supported" && exit 0
-       setup_veth && echo "SKIP: veth not supported" && exit 0
+       cleanup_done=0
+       for arg do
+               eval setup_${arg} && echo "  ${arg} not supported" && return 0
+       done
 
-       case ${tunnel_type} in
-       "vti6")
-               setup_vti6 && echo "SKIP: vti6 not supported" && exit 0
-               setup_xfrm && echo "SKIP: xfrm not supported" && exit 0
-               ;;
-       *)
-               ;;
-       esac
+       return 1
 }
 
 cleanup() {
+       [ ${cleanup_done} -eq 1 ] && return
        ip netns del ${NS_A} 2 > /dev/null
        ip netns del ${NS_B} 2 > /dev/null
+       cleanup_done=1
 }
 
 mtu() {
@@ -130,7 +129,7 @@ route_get_dst_pmtu_from_exception() {
 }
 
 test_pmtu_vti6_exception() {
-       setup vti6
+       setup namespaces veth vti6 xfrm && return 2
 
        # Create route exception by exceeding link layer MTU
        mtu "${ns_a}" veth_a 4000
@@ -141,30 +140,41 @@ test_pmtu_vti6_exception() {
 
        # Check that exception was created
        if [ "$(route_get_dst_pmtu_from_exception "${ns_a}" ${vti6_b_addr})" = 
"" ]; then
-               echo "FAIL: Tunnel exceeding link layer MTU didn't create route 
exception"
-               exit 1
+               echo "  tunnel exceeding link layer MTU didn't create route 
exception"
+               return 0
        fi
 
        # Decrease tunnel MTU, check for PMTU decrease in route exception
        mtu "${ns_a}" vti_a 3000
 
        if [ "$(route_get_dst_pmtu_from_exception "${ns_a}" ${vti6_b_addr})" 
-ne 3000 ]; then
-               echo "FAIL: Decreasing tunnel MTU didn't decrease route 
exception PMTU"
-               exit 1
+               echo "  decreasing tunnel MTU didn't decrease route exception 
PMTU"
+               return 0
        fi
 
        # Increase tunnel MTU, check for PMTU increase in route exception
        mtu "${ns_a}" vti_a 9000
        if [ "$(route_get_dst_pmtu_from_exception "${ns_a}" ${vti6_b_addr})" 
-ne 9000 ]; then
-               echo "FAIL: Increasing tunnel MTU didn't increase route 
exception PMTU"
-               exit 1
+               echo "  increasing tunnel MTU didn't increase route exception 
PMTU"
+               return 0
        fi
 
-       echo "PASS"
+       return 1
 }
 
 trap cleanup EXIT
 
-test_pmtu_vti6_exception
+exitcode=0
+for name in ${tests}; do
+       echo "${name}: START"
+       eval test_${name}
+       ret=$?
+       cleanup
+
+       if [ $ret -eq 0 ]; then echo "${name}: FAIL"; exitcode=1
+       elif [ $ret -eq 1 ]; then echo "${name}: PASS"
+       elif [ $ret -eq 2 ]; then echo "${name}: SKIP"
+       fi
+done
 
-exit 0
+exit ${exitcode}
-- 
2.15.1

Reply via email to