草庐IT

python mock side_effect 或 return_value 取决于 call_count

coder 2023-05-22 原文

为了测试一个轮询函数，我想模拟一个子函数的调用，这样第一次调用它就会失败，第二次调用它就会成功。这是它的一个非常简化的版本:

poll_function(var1):
    value = sub_function(var1)  # First call will return None
    while not value:
        time.sleep(POLLING_INTERVAL)  
        value = sub_function(var1) # A subsequent call will return a string, e.g "data"
    return value

这可能与 mock 框架中的 Mock 对象有关吗？我知道 Mock 对象有一个 call_count 属性，我应该能够以某种方式使用。

现在我已经通过创建一个自定义模拟对象来解决这个问题，我用它来猴子补丁 sub_function()，但我觉得应该有一个更好的不那么冗长的方法:

def test_poll():
    class MyMock(object):                                                      

        def __init__(self, *args):                                             
            self.call_count = 0                                                

        def sub_function(self, *args, **kwargs):                             
            if self.call_count > 1:                                            
                return "data"            
            else:                                                              
                self.call_count += 1                                           
                return None  

    my_mock = MyMock()                                                         
    with patch('sub_function', my_mock.sub_function):           
        ok_(poll_function())

最佳答案

如果我正确理解了您的问题，请通过 setting side_effect to an iterable 进行操作.对于您的简单案例:

>>> mock_poll = Mock(side_effect=[None, 'data'])
>>> mock_poll()
None
>>> mock_poll()
'data'

如果您想允许无限次调用，请使用 itertools cycle 和 chain 函数:

>>> mock_poll = Mock(side_effect=chain(['first'], cycle(['others'])))

关于python mock side_effect 或 return_value 取决于 call_count，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/13358939/

return_valueside_effectcodefunctionsectionpythonmocking

有关python mock side_effect 或 return_value 取决于 call_count的更多相关文章

  1. ruby - 怎么来的(a_method || :other) returns :other only when assigning to a var called a_method? - 2

    给定以下方法:defsome_method:valueend以下语句按我的预期工作:some_method||:other#=>:valuex=some_method||:other#=>:value但是下面语句的行为让我感到困惑:some_method=some_method||:other#=>:other它按预期创建了一个名为some_method的局部变量，随后对some_method的调用返回该局部变量的值。但为什么它分配:other而不是:value呢？我知道这可能不是一件明智的事情，并且可以看出它可能有多么模棱两可，但我认为应该在考虑作业之前评估作业的右侧...我已经在R

  2. ruby - Ruby 是否使用 $stdout 来写入 puts 和 return 的输出？ - 2

    我想知道Ruby用来在命令行打印这些东西的输出流:irb(main):001:0>a="test"=>"test"irb(main):002:0>putsatest=>nilirb(main):003:0>a=>"test"$stdout是否用于irb(main):002:0>和irb(main):003:0>？而且，在这两次调用之间，$stdout的值是否有任何变化？另外，有人能告诉我打印/写入这些内容的Ruby源代码吗？ 最佳答案 是的。而且很容易向自己测试/证明。在命令行试试这个:ruby-e'puts"foo"'>test.

  3. ruby - 为什么 return 关键字会导致我的 'if block' 出现问题？ - 2

    下面的代码工作正常:person={:a=>:A,:b=>:B,:c=>:C}berson={:a=>:A1,:b=>:B1,:c=>:C1}kerson=person.merge(berson)do|key,oldv,newv|ifkey==:aoldvelsifkey==:bnewvelsekeyendendputskerson.inspect但是如果我在“ifblock”中添加return，我会得到一个错误:person={:a=>:A,:b=>:B,:c=>:C}berson={:a=>:A1,:b=>:B1,:c=>:C1}kerson=person.merge(berson

  4. ruby-on-rails - 在 Ruby 或 Rails 中，hash.merge({ :order => 'asc' }) can return a new hash with a new key. 什么可以返回带有已删除键的新散列？ - 2

    在Ruby(或Rails)中，我们可以做到new_params=params.merge({:order=>'asc'})现在new_params是一个带有添加键:order的散列。但是是否有一行可以返回带有已删除key的散列？线路new_params=params.delete(:order)不会工作，因为delete方法返回值，仅此而已。我们必须分3步完成吗？tmp_params=paramstmp_params.delete(:order)returntmp_params有没有更好的方法？因为我想做一个new_params=(params[:order].blank?||para

  5. ruby - setter 方法 return self 不起作用，这是一个错误？ - 2

    这个问题在这里已经有了答案:Isitpossibletohaveclass.property=xreturnsomethingotherthanx?(3个答案)关闭8年前。我想迭代一个字符串数组，并将它们中的每一个分配给类User的一个新实例，我希望我会得到一个User对象数组:classUserdefname=(name)@name=nameselfendendoriginal_array=["aaa","bbb","bbb"]result=original_array.collect{|str|User.new.name=str}但结果是一个字符串数组!putsresult.ins

  6. ruby-on-rails - Unicorn.rb 配置取决于环境 - 2

    我在config/unicorn.rb中有一行看起来像这样:working_directory"/SomePath/Web\Development/Rails/learning"但是在生产环境中，需要不同的路径。我想尽可能多地保留源代码控制(还没有放弃任何东西)。有没有办法根据环境设置working_directory？ 最佳答案 您尝试如何访问RAILS_ENV或RACK_ENV？它应该通过“ENV”就像...ENV['RAILS_ENV']……或者……ENV['RACK_ENV']当您将“-E”标志传递给unicorn时，一个或

  7. ruby-on-rails - 删除 rspec 中的测试 - 更改(模型，:count) failing - Why is reload needed? - 2

    TLDR:App.count需要重新加载才能看到创建的记录。为什么？我找到了很多关于测试DELETE方法的引用资料，如下所示:expect{delete_request}.tochange(App,:count).by(-1)这是有道理的，并且适用于一些类似的场景。但是，我在测试不应该执行的删除时遇到了一个问题，例如当没有用户登录时。我从这里开始，用两种方法来测试同一件事:require'rails_helper'RSpec.describeV1::AppsController,type::controllerdolet(:user){create(:user)}let(:app){c

  8. sql - 如何模拟ActiveRecord Model.count.to_sql - 2

    我想显示一个计数中使用的SQL。但是，Model.count.to_sql将不起作用，因为count返回一个没有to_sql方法的FixNum。我认为最简单的解决方案是这样做:Model.where(nil).to_sql.sub(/SELECT.*FROM/,"SELECTCOUNT(*)FROM")这会创建与Model.count中使用的SQL相同的SQL，但它是否会导致进一步的问题？例如，如果我添加一个复杂的where子句和一些连接。有更好的方法吗？ 最佳答案 你可以试试Model.select("count(*)asmode

  9. ruby-on-rails - 为什么 Array.count 在开发模式下可用但在生产模式下不可用？ - 2

    对于最近的一个项目，我有几个View是这样的代码:这在开发模式下工作得很好......我将它推出到生产模式并且它爆炸了，说count不是Array的有效方法。我将每个实例都改为使用Array#length，它似​​乎可以正常工作。1)这种行为差异的原因是什么？2)我应该注意开发模式和生产模式之间的任何其他令人兴奋的差异吗？道德:确保您的生产托管环境使用与本地开发环境相同的Ruby版本。:)谢谢汤姆 最佳答案 count方法仅在Ruby1.9及更高版本中可用。我建议您使用与服务器相同版本的Ruby以避免此类问题-1.9中发生了很多变化

  10. ruby-on-rails - rails : Prevent will_paginate from calling #count of ActiveRelation - 2

    当我向will_paginate传递一个ActiveRelation时，它总是调用它的#count方法并访问数据库以找出项目的总数。但是这个操作需要时间，而且我已经缓存并准备好了总数。我可以将这个预先计算的计数传递给will_paginate并阻止它访问数据库吗？我尝试了:count选项，但它作为选项传递给了ActiveRecord:active_relation.paginate(page:2,per_page:100,count:total_count)谢谢!:) 最佳答案 通过:total_entries传递缓存计数解决了问题

随机推荐