Python Mock 学习笔记 - 文章教程

Python Mock 学习笔记

发布于 2021-03-30 字数 924 浏览 1097 评论 0

hello.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

__author__ = 'liuzhijun'

import os


def rm(filename):
    if os.path.isfile(filename):
        os.remove(filename)

test_hello.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import unittest

import mock

from hello import rm


class RmTestCase(unittest.TestCase):
    @mock.patch('hello.os.path')
    @mock.patch('hello.os')
    def test_rm(self, mock_os, mock_path):
        mock_path.isfile.return_value = False
        rm("any path")
        print mock_os
        self.assertFalse(mock_os.remove.called, "no call")
        mock_path.isfile.return_value = True
        rm("any path")
        mock_os.remove.assert_called_with("any path")


if __name__ == '__main__':
    unittest.main()

test_rm 函数的第一个参数有靠近该方法最近的那个装饰器提供。

如果你对这篇文章有疑问,欢迎到本站 社区 发帖提问或使用手Q扫描下方二维码加群参与讨论,获取更多帮助。

扫码加入群聊

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。

目前还没有任何评论,快来抢沙发吧!

关于作者

JSmiles

生命进入颠沛而奔忙的本质状态,并将以不断告别和相遇的陈旧方式继续下去。

2583 文章
29 评论
84935 人气
更多

推荐作者

清风夜微凉

文章 1 评论 0

为你鎻心

文章 2 评论 0

xxhui

文章 0 评论 0

1PKOH46yx8j0x

文章 0 评论 0

Arthur

文章 0 评论 0