Initial YakPanel commit

mod/test/process/test_user.py

@@ -0,0 +1,149 @@
+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()