File size: 3,445 Bytes
020c337 | 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 | import unittest
import sys
if "/www/server/panel" not in sys.path:
sys.path.insert(0, "/www/server/panel")
from mod.base import RealProcess
if "/www/server/panel/class" not in sys.path:
sys.path.insert(0, "/www/server/panel/class")
import public
real_process = RealProcess()
class TestRealProcess(unittest.TestCase):
def test_get_process_list(self):
res = real_process.get_process_list()
self.assertEqual(res['code'], 1)
def test_get_process_info_by_pid(self):
res = real_process.get_process_info_by_pid(1)
self.assertEqual(res['code'], 1)
def test_get_process_info_by_name(self):
res = real_process.get_process_info_by_name('system')
self.assertEqual(res['code'], 1)
def test_get_process_info_by_exec(self):
res = real_process.get_process_info_by_exec('/usr/sbin/sshd')
self.assertEqual(res['code'], 1)
def test_get_process_info_by_port(self):
res = real_process.get_process_info_by_port(22)
self.assertEqual(res['code'], 1)
def test_get_process_info_by_ip(self):
res = real_process.get_process_info_by_ip('192.168.168.66')
self.assertEqual(res['code'], 1)
def test_get_process_info_by_openfile(self):
res = real_process.get_process_info_by_openfile('/etc/passwd')
self.assertEqual(res['code'], 1)
def test_get_process_ps(self):
res = real_process.get_process_ps('grep')
self.assertEqual(res['code'], 1)
def test_get_process_tree(self):
res = real_process.get_process_tree(1)
self.assertEqual(res['code'], 1)
# def test_kill_pid(self):
# try:
# pid = real_process.get_process_info_by_exec('/www/server/panel/mod/test/process/test.py')['data'][0]['pid']
# res = real_process.kill_pid(pid)
# self.assertEqual(res['code'], 1)
# except:
# pass
# def test_kill_name(self):
# try:
# name = real_process.get_process_info_by_exec('/www/server/panel/mod/test/process/test.py')['data'][0]['name']
# res = real_process.kill_name(name)
# self.assertEqual(res['code'], 1)
# except:
# pass
#
# def test_kill_tree(self):
# try:
# pid = real_process.get_process_info_by_exec('/www/server/panel/mod/test/process/test.py')['data'][0]['pid']
# res = real_process.kill_tree(pid)
# self.assertEqual(res['code'], 1)
# except:
# pass
#
# def test_kill_proc_all(self):
# pid = real_process.get_process_info_by_exec('/www/server/panel/mod/test/process/test.py')['data'][0]['pid']
# res = real_process.kill_proc_all(pid)
# self.assertEqual(res['code'], 1)
#
# def test_kill_port(self):
# res = real_process.kill_port(22)
# self.assertEqual(res['code'], 1)
def test_add_black_ip(self):
res = real_process.add_black_ip(['1.2.3.4'])
self.assertEqual(res['code'], 1)
def test_del_black_ip(self):
res = real_process.del_black_ip(['1.2.3.4'])
self.assertEqual(res['code'], 1)
def test_firewall_reload(self):
res = real_process.firewall_reload()
self.assertEqual(res['code'], 1)
def test_get_run_list(self):
res = real_process.get_run_list()
self.assertEqual(res['code'], 1)
if __name__ == '__main__':
unittest.main()
|