从其他方法调用的测试方法

Question

我有模块数据库，其中包含方法generate_from_database，用于为循环旋转并调用方法get_length。如何通过使用rspec或mocha来测试get_length是否被调用了n次？

module Database 
class Length < ActiveRecord::Base 
    def get_length(i,j) 
    ... 
    end 
end 
def Database.generate_from_database 
    ... 
for i in 0...size 
    for j in 0...size 
    Length.new.get_length(i+1,j+1)) 
end 
end 

Answer 1

此：

mock_length = mock("length") 
Length.should_receive(:new).exactly(n).times.and_return(mock_length) 
mock_length.should_receive(:get_length).exactly(n).times 

应该工作。

Answer 2

你可以这样类似：

Length.should_receive(:get_length).exactly(n).times

更多信息：http://rspec.info/documentation/mocks/message_expectations.html