gpt4 book ai didi

python - 检查使用运行时对象创建调用的模拟

转载 作者:太空宇宙 更新时间:2023-11-04 05:27:17 25 4
gpt4 key购买 nike

我有类似 cassandra 集群的代码

cluster = Cluster(
config.CASS_CLUSTER,
load_balancing_policy=policies.DCAwareRoundRobinPolicy(
config.CASS_D_CENTER))

当我为此编写 UT 并尝试使用我的参数检查 Cluster 调用时。

mock_cluster.assert_called_with(
config.CASS_CLUSTER,
load_balancing_policy=policies.DCAwareRoundRobinPolicy(
config.CASS_D_CENTER))

它给出了错误。

AssertionError: Expected call: Cluster(['192.168.1.1'], load_balancing_policy=<cassandra.policies.DCAwareRoundRobinPolicy object at 0x106fa2ed0>)
Actual call: Cluster(['192.168.1.1'], load_balancing_policy=<cassandra.policies.DCAwareRoundRobinPolicy object at 0x106fa2cd0>)

我得到这个是因为两个对象不同,有什么方法可以检查 mock 中的 called_with 吗?

最佳答案

mock_cluster.assert_called_with(
config.CASS_CLUSTER,
load_balancing_policy=mock.ANY)

关于python - 检查使用运行时对象创建调用的模拟,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/38319505/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com