yakpanel-core/mod/test/process/test_user.py

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()