From cc1b2315af6f354afb4bde5d057a5ab282a241ec Mon Sep 17 00:00:00 2001
From: Robert Schmidt <robert.schmidt@openairinterface.org>
Date: Thu, 31 Jul 2025 16:32:23 +0200
Subject: [PATCH] CI: add simple unit test for UE handling

---
 .../tests/config/test_module_infra.yaml       | 19 +++++++++
 ci-scripts/tests/module.py                    | 41 +++++++++++++++++++
 2 files changed, 60 insertions(+)
 create mode 100644 ci-scripts/tests/config/test_module_infra.yaml
 create mode 100644 ci-scripts/tests/module.py

diff --git a/ci-scripts/tests/config/test_module_infra.yaml b/ci-scripts/tests/config/test_module_infra.yaml
new file mode 100644
index 0000000000..fadc721232
--- /dev/null
+++ b/ci-scripts/tests/config/test_module_infra.yaml
@@ -0,0 +1,19 @@
+test:
+  Host: localhost
+  InitScript: ""
+  TermScript: ""
+  AttachScript: "true"
+  DetachScript: "true"
+  NetworkScript: ip a show dev lo
+  IF: lo
+  MTU: 65536
+
+test-fail:
+  Host: localhost
+  InitScript: ""
+  TermScript: ""
+  AttachScript: "false"
+  DetachScript: "true"
+  NetworkScript: ip a show dev TESTDEVICEDOESNOTEXIST
+  IF: TESTDEVICEDOESNOTEXIST
+  MTU: 123
diff --git a/ci-scripts/tests/module.py b/ci-scripts/tests/module.py
new file mode 100644
index 0000000000..b6898405d4
--- /dev/null
+++ b/ci-scripts/tests/module.py
@@ -0,0 +1,41 @@
+import sys
+import logging
+logging.basicConfig(
+	level=logging.DEBUG,
+	stream=sys.stdout,
+	format="[%(asctime)s] %(levelname)8s: %(message)s"
+)
+import os
+
+import unittest
+
+sys.path.append('./') # to find OAI imports below
+import cls_module
+import cls_cmd
+
+class TestModule(unittest.TestCase):
+    def test_simple_module(self):
+        c = cls_module.Module_UE("test", filename="tests/config/test_module_infra.yaml")
+        success = c.initialize()
+        self.assertTrue(success)
+        ip = c.attach()
+        self.assertEqual(ip, "127.0.0.1")
+        self.assertTrue(c.checkMTU())
+        c.detach()
+        logs = c.terminate()
+        self.assertEqual(logs, None) # no tracing
+
+    @unittest.skip("this test takes long: it verifies the UE cannot attach")
+    def test_simple_fail(self):
+        c = cls_module.Module_UE("test-fail", filename="tests/config/test_module_infra.yaml")
+        success = c.initialize()
+        self.assertTrue(success)
+        ip = c.attach()
+        self.assertEqual(ip, None)
+        self.assertFalse(c.checkMTU())
+        c.detach()
+        logs = c.terminate()
+        self.assertEqual(logs, None) # no tracing
+
+if __name__ == '__main__':
+    unittest.main()