File size: 4,376 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 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 | import unittest
import sys
if "/www/server/panel" not in sys.path:
sys.path.insert(0, "/www/server/panel")
if "/www/server/panel/class" not in sys.path:
sys.path.insert(0, "/www/server/panel/class")
from mod.base import RealUser
realuser = RealUser()
class TestRealUser(unittest.TestCase):
def setUp(self):
realuser.add_user('test1', 'test1', 'test1')
realuser.add_group('test')
def tearDown(self):
realuser.remove_user('test1')
realuser.remove_user('test')
realuser.remove_group('test1')
realuser.remove_group('test')
def test_get_user_list(self):
print('get_user_list:')
res = realuser.get_user_list()
print(res)
self.assertEqual(res['code'], 1)
def test_get_group_list(self):
print('get_group_list:')
res = realuser.get_group_list()
print(res)
self.assertEqual(res['code'], 1)
def test_add_user(self):
print('add_user:')
res = realuser.add_user('test', 'test', 'test')
print(res)
self.assertEqual(res['code'], 1)
def test_remove_user(self):
print('remove_user:')
res = realuser.remove_user('test1')
print(res)
self.assertEqual(res['code'], 1)
def test_edit_user_pwd(self):
print('edit_user_pwd:')
res = realuser.edit_user_pwd('test1', 'test111')
print(res)
self.assertEqual(res['code'], 1)
def test_edit_user_group(self):
print('edit_user_group:')
res = realuser.edit_user_group('test1', 'test')
print(res)
self.assertEqual(res['code'], 1)
def test_edit_user_ps(self):
print('edit_user_ps:')
res = realuser.edit_user_ps('test1', 'test')
print(res)
self.assertEqual(res['code'], 1)
def test_edit_user_status(self):
print('edit_user_status:')
res = realuser.edit_user_status('test1', '0')
print(res)
self.assertEqual(res['code'], 1)
def test_edit_user_login_shell(self):
print('edit_user_login_shell:')
res = realuser.edit_user_login_shell('test1', '/bin/bash')
print(res)
self.assertEqual(res['code'], 1)
def test_edit_user_home(self):
print('edit_user_home:')
res = realuser.edit_user_home('test1', '/home/test1')
print(res)
self.assertEqual(res['code'], 1)
def test_get_user_info(self):
print('get_user_info:')
res = realuser.get_user_info('test1')
print(res)
self.assertEqual(res['code'], 1)
def test_add_group(self):
print('add_group:')
res = realuser.add_group('test2')
print(res)
realuser.remove_group('test2')
self.assertEqual(res['code'], 1)
def test_remove_group(self):
realuser.add_group('test')
print('remove_group:')
res = realuser.remove_group('test')
print(res)
self.assertEqual(res['code'], 1)
def test_edit_group_name(self):
print('edit_group_name:')
res = realuser.edit_group_name('test', 'test5')
print(res)
print(realuser.remove_group('test5'))
self.assertEqual(res['code'], 1)
def test_get_group_info(self):
print('get_group_info:')
res = realuser.get_group_info('test1')
print(res)
self.assertEqual(res['code'], 1)
def test_get_group_user(self):
print('get_group_user:')
res = realuser.get_group_user('test1')
print(res)
self.assertEqual(res['code'], 1)
def test_get_user_group(self):
print('get_user_group:')
res = realuser.get_user_group('test1')
print(res)
self.assertEqual(res['code'], 1)
def test_edit_user_permission(self):
print('edit_user_permission:')
res = realuser.edit_user_permission('test1', '777')
print(res)
self.assertEqual(res['code'], 1)
def test_edit_group_permission(self):
print('edit_group_permission:')
res = realuser.edit_group_permission('test1', '777')
print(res)
self.assertEqual(res['code'], 1)
def test_edit_user_name(self):
print('edit_user_name:')
res = realuser.edit_user_name('test1', 'test')
print(res)
self.assertEqual(res['code'], 1)
if __name__ == '__main__':
unittest.main()
|