存根ActiveRecord結果數組

Question

使用Ruby 1.8.6，當被測代碼對結果調用'count'方法時，對finder方法進行存根返回不起作用：

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

這是因為Array.count僅在Ruby 1.8.7中添加。 我相信Rails還會通過ActiveRecord :: Calculations模塊動態添加一個count方法。

有什么方法可以將結果數組轉換為與Rails finder方法返回的特殊Array類型完全一樣的行為嗎？ 對結果進行分頁時很容易：我可以簡單地調用[] .paginate。 但這不適用於普通的取景器結果。 我試過[].extend(ActiveRecord::Calculations)但這也不起作用。

Answer 1

您也可以模擬計數：

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