标签: python-unittest

我正在尝试使用unittest.mock，但出现错误：

\n

AttributeError：没有属性“get_pledge_Frequency”

\n

我有以下文件结构：

pledges/views/\n\xe2\x94\x9c\xe2\x94\x80\xe2\x94\x80 __init__.py\n\xe2\x94\x9c\xe2\x94\x80\xe2\x94\x80 util.py\n\xe2\x94\x94\xe2\x94\x80\xe2\x94\x80 user_profile.py\npledges/tests/unit/profile\n\xe2\x94\x9c\xe2\x94\x80\xe2\x94\x80 __init__.py\n\xe2\x94\x94\xe2\x94\x80\xe2\x94\x80 test_user.py\n

里面pledges/views/__init___.py我有：

from .views import *\nfrom .account import account\nfrom .splash import splash\nfrom .preferences import preferences\nfrom .user_profile import user_profile\n

在里面，user_profile.py我有一个名为的函数user_profile，它调用内部的函数util.py，如下get_pledge_frequency所示：

def user_profile(request, user_id):\n    # some logic\n\n    # !!!!!!!!!!!!!!!!\n    a, b = get_pledge_frequency(parameter) # this is the function I want to mock\n\n    # more logic\n\n    return some_value\n

我里面有一个测试test_user.py如下：

def …

我想测试一些错误处理逻辑，所以我想在单元测试中模拟特定的异常类型。我正在模拟对 boto3 的调用，但我想进行该模拟以引发异常ParameterNotFound。我正在测试的代码遵循以下模式：

boto3_client = boto3.client('ssm')
try:
    temp_var = boto3_client.get_parameter(Name="Something not found")['Parameter']['Value']
except boto3_client.exceptions.ParameterNotFound:
    ... [logic I want to test]

我创建了一个单元测试模拟，但我不知道如何使其引发异常作为此 ParameterNotFound 异常。我尝试了以下方法，但它不起作用，因为在评估 except 子句时它得到“异常必须从基类派生”：

@patch('patching_config.boto3.client')
def test_sample(self, mock_boto3_client):
        mock_boto3_client.return_value = mock_boto3_client

        def get_parameter_side_effect(**kwargs):
            raise boto3.client.exceptions.ParameterNotFound()

        mock_boto3_client.get_parameter.side_effect = get_parameter_side_effect

如何在单元测试中模拟 ParameterNotFound boto3 异常？

我正在使用发现模式运行 Python 单元测试：

 % python -m unittest discover

系统为每个测试打印一个点，但我宁愿看到测试名称。

有没有一个选项可以实现这种情况？

请检查以下代码：

import unittest


CORRECT_MESSAGE = 'Correct message'
WRONG_MESSAGE = 'Wrong message'


def fn():
    raise KeyError(CORRECT_MESSAGE)


class Test(unittest.TestCase):
    def test(self):
        # I am expecting this test to fail as the msg I am
        # checking is WRONG_MESSAGE, and not CORRECT_MESSAGE.
        with self.assertRaises(KeyError, msg=WRONG_MESSAGE):
            fn()


unittest.main()

正如评论中提到的，我预计此测试会失败，因为我正在检查的消息 ( WRONG_MESSAGE) 不正确，但测试通过了。

我缺少什么？我已经检查过：assertRaises(exception, *, msg=None)。

我有一个名为的文件redis_db.py，其中包含连接到 redis 的代码

import os
import redis
import sys

class Database:
       def __init__(self, zset_name):
        redis_host = os.environ.get('REDIS_HOST', '127.0.0.1')
        redis_port = os.environ.get('REDIS_PORT', 6379)

        self.db = redis.StrictRedis(host=redis_host, port=redis_port)
        self.zset_name = zset_name

    def add(self, key):
        try:
            self.db.zadd(self.zset_name, {key: 0})
        except redis.exceptions.ConnectionError:
            print("Unable to connect to redis host.")
            sys.exit(0)

我有另一个名为 app.py 的文件，如下所示

from flask import Flask
from redis_db import Database

app = Flask(__name__)
db = Database('zset')

@app.route('/add_word/word=<word>')
def add_word(word):
   db.add(word)
   return ("{} added".format(word))

if __name__ == '__main__':
    app.run(host='0.0.0.0', port='8080')

现在我正在为 add_word 函数编写单元测试，如下所示 …

我希望无论传入什么内容，以下调用which_user都会返回，但它的行为就好像它根本没有被嘲笑一样。self.user

def test_user_can_retrieve_favs_using_impersonation(self):
        with mock.patch('impersonate.helpers.which_user', return_value=self.user):
            user = which_user(self.user2)

我在这里做错了什么？我which_user像这样导入：from impersonate.helpers import which_user如果有帮助的话。

我需要在整个测试套件运行后或在整个测试退出时执行命令。我在unittest中看到tearDownhook ，但在ruby中没有after-testsuite hook或类似的东西。at_exit