smoketest: l2tpv3: only remove modules if they are loaded

1 files changed, 9 insertions, 3 deletions

diff --git a/smoketest/scripts/cli/test_interfaces_l2tpv3.py b/smoketest/scripts/cli/test_interfaces_l2tpv3.py
index a2091ff10..06ced5c40 100755
--- a/smoketest/scripts/cli/test_interfaces_l2tpv3.py
+++ b/smoketest/scripts/cli/test_interfaces_l2tpv3.py
@@ -14,13 +14,14 @@
 # You should have received a copy of the GNU General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
+import os
 import json
 import unittest
 
 from base_interfaces_test import BasicInterfaceTest
 from vyos.util import cmd
 
-class GeneveInterfaceTest(BasicInterfaceTest.TestCase):
+class L2TPv3InterfaceTest(BasicInterfaceTest.TestCase):
     @classmethod
     def setUpClass(cls):
         cls._test_ip = True
@@ -53,12 +54,17 @@ class GeneveInterfaceTest(BasicInterfaceTest.TestCase):
                     for opt in self._options[interface]:
                         dict.update({opt.split()[0].replace('-','_'): opt.split()[1]})
 
-                    for key in ['peer_session_id', 'peer_tunnel_id', 'session_id', 'tunnel_id']:
+                    for key in ['peer_session_id', 'peer_tunnel_id',
+                                'session_id', 'tunnel_id']:
                         self.assertEqual(str(config[key]), dict[key])
 
 
 if __name__ == '__main__':
     # when re-running this test, cleanup loaded modules first so they are
     # reloaded on demand - not needed but test more and more features
-    cmd('sudo rmmod l2tp_ip6 l2tp_eth l2tp_eth l2tp_netlink l2tp_core')
+    for module in ['l2tp_ip6', 'l2tp_ip', 'l2tp_eth', 'l2tp_eth',
+                   'l2tp_netlink', 'l2tp_core']:
+        if os.path.exists(f'/sys/module/{module}'):
+            cmd(f'sudo rmmod {module}')
+
     unittest.main(verbosity=2)