Question

我是unittest.mock库的新手，无法解决我遇到的问题。我在下面的文件夹结构中有一个名为'function.py'的类

SRC _ init .py
- function.py
测试
- 的初始化的.py
- test_function.py

在test_function.py中我有一些这样的代码：

    import unittest
    from unittest import mock
    from ..src.function import get_subscriptions
    from ..src import function

class TestCheckOrder(unittest.TestCase):
    @mock.patch.object(function, 'table')
    def test_get_subscriptions_success(self, mocked_table):

        mocked_table.query.return_value = []
        user_id = "test_user"
        status = True

        get_subscriptions(user_id, status) 
        mocked_table.query.assert_called_with(
          KeyConditionExpression=conditions.Key('user_id').eq(user_id),
          FilterExpression=conditions.Attr('status').eq(int(status)))

在function.py中：

import boto3
from boto3.dynamodb import conditions
dynamodb = boto3.resource("dynamodb")
table = dynamodb.Table("Subscriptions")

def get_subscriptions(user_id, active=True):
    results = table.query(
        KeyConditionExpression=conditions.Key(
        'user_id').eq(user_id),
        FilterExpression=conditions.Attr('status').eq(int(active))
    )

return results['Items']

如果我运行此操作，我会收到以下异常：

**AssertionError: Expected call: query(FilterExpression=<boto3.dynamodb.conditions.Equals object at 0x1116011d0>, KeyConditionExpression=<boto3.dynamodb.conditions.Equals object at 0x111601160>)
Actual call: query(FilterExpression=<boto3.dynamodb.conditions.Equals object at 0x1116010f0>, KeyConditionExpression=<boto3.dynamodb.conditions.Equals object at 0x111601080>)**

先谢谢你帮助我。

Answer 1

问题在于，当您在测试中调用assert_called_with时，您正在创建conditions.Key和conditions.Attr的新实例。由于这些实例与实际调用中的实例不同，因此存在不匹配（检查回溯中显示的十六进制ID）。

您可以从函数调用本身获取kwargs并测试其属性，而不是这样：

name, args, kwargs = mocked_table.query.mock_calls[0]
assert kwargs['KeyConditionExpression'].get_expression()['values'][1] == user_id
assert kwargs['FilterExpression'].get_expression()['values'][1] == int(status)

Python mock：AssertionError：预期和实际调用不一样

1 个答案: