# SPDX-License-Identifier: LicenseRef-CSSL-1.0

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()
