65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
import sys
|
|
import logging
|
|
import tempfile
|
|
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
|
|
from cls_ci_helper import TestCaseCtx
|
|
import cls_cmd
|
|
|
|
class TestModule(unittest.TestCase):
|
|
def setUp(self):
|
|
self.ctx = TestCaseCtx.Default(tempfile.mkdtemp())
|
|
def tearDown(self):
|
|
with cls_cmd.LocalCmd() as c:
|
|
c.run(f'rm -rf {self.ctx.logPath}')
|
|
|
|
def test_simple_module(self):
|
|
c = cls_module.Module_UE("test", filename="tests/config/test_module_infra.yaml")
|
|
self.assertFalse(c.trace)
|
|
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
|
|
|
|
def test_simple_trace(self):
|
|
c = cls_module.Module_UE("test-trace", filename="tests/config/test_module_infra.yaml")
|
|
self.assertTrue(c.trace)
|
|
success = c.initialize()
|
|
self.assertTrue(success)
|
|
ip = c.attach()
|
|
self.assertEqual(ip, "127.0.0.1")
|
|
self.assertTrue(c.checkMTU())
|
|
c.detach()
|
|
tmp = tempfile.mkdtemp()
|
|
log_file = c.terminate(self.ctx)
|
|
# undeploy uses archiveArtifact(), which writes to {prefix}-logs
|
|
self.assertEqual(log_file, [f"{self.ctx.baseFilename()}-tttf"]) # matches test-trace UE collection file
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|