如何对具有给定权限的文档进行单元测试？

import unittest
from unittest.mock import patch

class DocumentTestCase(unittest.TestCase):
    @patch('your_module.check_permission')
    def test_document_access(self, mock_check_permission):
        # 模拟管理员权限
        mock_check_permission.return_value = True
        self.assertTrue(your_module.access_document('admin'))

        # 模拟普通用户权限
        mock_check_permission.return_value = False
        self.assertFalse(your_module.access_document('user'))

class PermissionTestCase(unittest.TestCase):
    def test_admin_permissions(self):
        self.assertTrue(check_permission('admin', 'read'))
        self.assertTrue(check_permission('admin', 'write'))
        self.assertTrue(check_permission('admin', 'delete'))

    def test_user_permissions(self):
        self.assertTrue(check_permission('user', 'read'))
        self.assertFalse(check_permission('user', 'write'))
        self.assertFalse(check_permission('user', 'delete'))

def check_permission(user, action):
    if user == 'admin':
        return True
    elif user == 'user' and action == 'read':
        return True
    return False