ruby-on-rails - stub ActiveRecord 结果数组

Question

使用 Ruby 1.8.6，当被测试的代码对结果调用“计数”方法时， stub 查找器方法以返回数组不起作用:

User.any_instance.stubs(:friends).returns([user1, user2])

那是因为 Array.count 只是在 Ruby 1.8.7 中添加的。 Rails 还动态添加了一个计数方法，我相信是通过 ActiveRecord::Calculations 模块实现的。

有什么方法可以将这个结果数组转换成与 Rails finder 方法返回的特殊数组类型完全一样的东西吗？对结果进行分页很简单:我可以简单地调用 [].paginate。但这不适用于正常的取景器结果。我尝试了 [].extend(ActiveRecord::Calculations) 但这也不起作用。

Answer 1

你也可以模拟计数:

arr = [user1, user2]
arr.stubs(:count).returns(arr.size)
User.any_instance.stubs(:friends).returns(arr)