rspec 中的测试循环返回值

Question

我需要测试如下方法

class Something
  def test_method
   count = 0
   arr = []
   max = 500
    puts "test_method"
    loop do
      arr += count if count.even?
      count += 1
    end break if count > max
    return arr
  end
end

所以基本上我需要测试 test_method() 是否返回一个数组实例并且 arr 大小是否大于 3。 但我不想每次都进入这个循环并返回结果。那么在 rspec 中有什么方法可以存根最大值并返回数组而不循环 500 次。

Answer 1

您可以使用class_variable_set 方法。

class Something
  @@max = 500

  def test_method
   count = 0
   arr = []

    puts "test_method"
    loop do
      arr += count if count.even?
      count += 1
    end break if count > @@max
    return arr
  end
end

在您的情况下，rspec before_each 块可能如下所示：

before(:each) { Something.class_variable_set :@@max, 20 }

您也可以使用instance_variable，如this answer 中所述

Answer 2

也许可以将max 作为可选参数传递给该方法？

class Something
  def test_method(max = 500)
    count = 0
    arr = []
    puts "test_method"
    # ...
  end
end