| import unittest |
| import tempfile |
| import os |
| import zipfile |
| import sys |
| from unittest.mock import patch, MagicMock |
| if "/www/server/panel" not in sys.path: |
| sys.path.append("/www/server/panel") |
|
|
| from mod.project.java.utils import get_jar_war_config, to_utf8, parse_application_yaml, TomCat |
|
|
|
|
| class TestJarWarConfig(unittest.TestCase): |
| def setUp(self): |
| |
| self.test_jar = tempfile.NamedTemporaryFile(delete=False) |
| self.test_jar_name = self.test_jar.name |
| self.test_jar.close() |
|
|
| |
| with zipfile.ZipFile(self.test_jar_name, 'w') as jar: |
| jar.writestr('application.yaml', 'spring:\n datasource:\n url: jdbc:mysql://localhost:3306/testdb') |
|
|
| |
| self.test_jar_no_config = tempfile.NamedTemporaryFile(delete=False) |
| self.test_jar_no_config_name = self.test_jar_no_config.name |
| self.test_jar_no_config.close() |
|
|
| with zipfile.ZipFile(self.test_jar_no_config_name, 'w') as jar: |
| jar.writestr('README.txt', 'This is a test JAR without configuration') |
|
|
| |
| self.test_non_zip = tempfile.NamedTemporaryFile(delete=False) |
| self.test_non_zip_name = self.test_non_zip.name |
| self.test_non_zip.close() |
|
|
| |
| self.test_non_existent = 'non_existent.jar' |
|
|
| def tearDown(self): |
| |
| os.unlink(self.test_jar_name) |
| os.unlink(self.test_jar_no_config_name) |
| os.unlink(self.test_non_zip_name) |
|
|
| def test_get_jar_war_config_with_valid_jar(self): |
| |
| result = get_jar_war_config(self.test_jar_name) |
| self.assertIsNotNone(result) |
| self.assertEqual(len(result), 1) |
| self.assertIn('application.yaml', result[0][0]) |
|
|
| def test_get_jar_war_config_with_no_config(self): |
| |
| result = get_jar_war_config(self.test_jar_no_config_name) |
| self.assertIsNone(result) |
|
|
| def test_get_jar_war_config_with_non_zip(self): |
| |
| result = get_jar_war_config(self.test_non_zip_name) |
| self.assertIsNone(result) |
|
|
| def test_get_jar_war_config_with_non_existent(self): |
| |
| result = get_jar_war_config(self.test_non_existent) |
| self.assertIsNone(result) |
|
|
| def test_to_utf8(self): |
| |
| byte_data = 'test'.encode('utf-8') |
| file_data_list = [('test.txt', byte_data)] |
| result = to_utf8(file_data_list) |
| self.assertIsNotNone(result) |
| self.assertEqual(result, [('test.txt', 'test')]) |
|
|
| def test_parse_application_yaml_valid(self): |
| |
| byte_data = 'spring:\n datasource:\n url: jdbc:mysql://localhost:3306/testdb' |
| file_data_list = [('application.yaml', byte_data.encode('utf-8'))] |
| result = parse_application_yaml(file_data_list) |
| self.assertIsNotNone(result) |
| self.assertEqual(len(result), 1) |
| self.assertIn('application.yaml', result[0][0]) |
| self.assertIsInstance(result[0][1], dict) |
| self.assertEqual(result[0][1]['spring']['datasource']['url'], 'jdbc:mysql://localhost:3306/testdb') |
|
|
| def test_parse_application_yaml_invalid(self): |
| |
| invalid_byte_data = 'not --- a valid yaml' |
| file_data_list = [('application.yaml', invalid_byte_data.encode('utf-8'))] |
| result = parse_application_yaml(file_data_list) |
| self.assertIsNotNone(result) |
| self.assertEqual(len(result), 0) |
|
|
|
|
| class TestTomCat(unittest.TestCase): |
| def setUp(self): |
| self.tomcat_path = "/usr/local/bttomcat/tomcat10" |
| self.tomcat = TomCat(self.tomcat_path) |
|
|
| def test_jdk_path(self): |
| self.assertEqual(self.tomcat.jdk_path, "/usr/local/btjdk/jdk8") |
|
|
| def test_config_xml(self): |
| self.assertIsNotNone(self.tomcat.config_xml) |
| print(self.tomcat.config_xml) |
|
|
| def test_save_config_xml(self): |
| self.tomcat.add_host("taaaaa", "/tmp/aaaa") |
| self.tomcat.add_host("taaaaa", "/tmp/aaaa") |
| self.tomcat.add_host("tbbb", "/tmp/tbbb") |
| self.assertEqual(self.tomcat.save_config_xml(), True) |
| with open(self.tomcat_path + "/conf/server.xml", "r") as f: |
| print(f.read()) |
|
|
| self.tomcat.remove_host("tbbb") |
| self.assertEqual(self.tomcat.save_config_xml(), True) |
| with open(self.tomcat_path + "/conf/server.xml", "r") as f: |
| print(f.read()) |
|
|
| def test_status(self): |
| print(self.tomcat.status()) |
|
|
|
|
| if __name__ == '__main__': |
| unittest.main() |
|
|
