File size: 3,482 Bytes
3a5cf48 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
import os
import shutil
import sys
if "/www/server/panel" not in sys.path:
sys.path.insert(0, "/www/server/panel")
from mod.test.test_web_conf import WebBaseTestcase, PREFIX, NGINX_CONFIG_CASE, APACHE_CONFIG_CASE
from mod.base.web_conf import ConfigMgr, Proxy
from mod.base.web_conf.util import GET_CLASS
class TestProxy(WebBaseTestcase):
proxy_obj = Proxy(PREFIX)
def test_create_project_proxy(self):
get = GET_CLASS()
get.proxyname = "aaa"
get.sitename = self.site_name
get.proxydir = "/"
get.proxysite = "https://www.baidu.com"
get.todomain = "www.baidu.com"
get.type = "1"
get.cache = "1"
get.subfilter = '[{"sub1":"","sub2":""},{"sub1":"","sub2":""},{"sub1":"","sub2":""}]'
get.advanced = "1"
get.cachetime = "1"
print(self.proxy_obj.create_proxy(get))
# get = GET_CLASS()
# get.proxyname = "ggfff"
# get.sitename = self.site_name
# get.proxydir = "/dad"
# get.proxysite = "https://www.baidu.com"
# get.todomain = "www.baidu.com"
# get.type = "1"
# get.cache = "1"
# get.subfilter = '[{"sub1":"","sub2":""},{"sub1":"","sub2":""},{"sub1":"","sub2":""}]'
# get.advanced = "1"
# get.cachetime = "1"
# print(self.proxy_obj.create_proxy(get))
def test_modify_project_proxy(self):
get = GET_CLASS()
get.proxyname = "ggfff"
get.sitename = self.site_name
get.proxydir = "/dygvccc"
get.proxysite = "https://www.baidu.com"
get.todomain = "www.baidu.com"
get.type = "1"
get.cache = "1"
get.subfilter = '[{"sub1":"","sub2":""},{"sub1":"","sub2":""},{"sub1":"","sub2":""}]'
get.advanced = "0"
get.cachetime = "1"
print(self.proxy_obj.modify_proxy(get))
def test_remove_project_proxy(self):
get = GET_CLASS()
get.proxyname = "aaa"
get.sitename = self.site_name
print(self.proxy_obj.remove_proxy(get))
def test_get_project_proxy_list(self):
get = GET_CLASS()
get.sitename = self.site_name
print(self.proxy_obj.get_proxy_list(get))
def setUp(self) -> None:
self.reset_site_config()
self.config_mgr = ConfigMgr(self.site_name, PREFIX)
panel_path = "/www/server/panel"
_proxy_conf_file = "{}/data/mod_proxy_file.conf".format(panel_path)
if os.path.exists(_proxy_conf_file):
os.remove(_proxy_conf_file)
ng_proxy_dir = "/www/server/panel/vhost/nginx/proxy/" + self.site_name
ap_proxy_dir = "/www/server/panel/vhost/apache/proxy/" + self.site_name
if os.path.exists(ng_proxy_dir):
shutil.rmtree(ng_proxy_dir)
if os.path.exists(ap_proxy_dir):
shutil.rmtree(ap_proxy_dir)
def runTest(self):
# self.change_env_to_nginx()
# self.test_create_project_proxy()
# self.test_remove_project_proxy()
# self.check_web_server_config()
self.change_env_to_apache()
self.test_create_project_proxy()
self.test_remove_project_proxy()
self.test_modify_project_proxy()
self.test_get_project_proxy_list()
self.check_web_server_config()
# def tearDown(self):
# self.reset_site_config()
if __name__ == '__main__':
import unittest
s = unittest.TestSuite()
s.addTest(TestProxy())
unittest.TextTestRunner().run(s)
|