Rspec 测试迭代答案

【问题标题】：Rsepc testing iterationRspec 测试迭代
【发布时间】：2021-05-02 03:07:47
【问题描述】：

我正在为这样的方法编写测试：

  def get_team_members(project_id)
    team_members = []

    Result.where(project_id: project_id).each do |result|
      user = User.find_by(id: result.user_id)
      team_members.push(user) unless user.nil?
    end

    team_members
  end

当我在测试只找到 1 条 result 记录的情况下，我可以这样做

allow(Result).to receive(:where).with(project_id:project_id).and_return([result1])

allow(User).to receive(:find_by).with(id: valid_result[:user_id]).and_return(nil)

或

allow(Result).to receive(:where).with(project_id:project_id).and_return([result1])

allow(User).to receive(:find_by).with(id: valid_result[:user_id]).and_return(user1)

问题1：如果我想在找到2条result记录时测试案例

allow(Result).to receive(:where).with(project_id:project_id).and_return([result1, result2])

对于result1，找到匹配的user (user1)，对于result2，未找到匹配的user (nil)。

在这种情况下，预期的返回值应该是team_members = [user1]。

如何使用不同的返回值 (user1, nil) 模拟两次 User.find_by()？

问题2：我真的需要测试我上面提到的案例吗？

【问题讨论】：

这与您提出的测试问题无关，但您的查询效率非常低 - 您将其构建为 N+1 查询 - 也就是说，对于包含 N 个团队成员的结果，您将访问数据库一次（对于您的where 语句），然后每个团队成员一次。这感觉就像你应该能够利用 Rails 的内置 has_many :through 选项：如果一个 Project has_many :results，并且每个 Result 属于一个 :user，然后声明一个项目 @ 987654341@ 意味着project.users 会给你你想要的收藏。
是的，斯科特你是对的。数据库设计不是很好。它是由以前的团队成员构建的，我正在开发它。我可能需要稍后更新架构。

标签： ruby-on-rails unit-testing rspec mocking iteration

【解决方案1】：

关于您的问题 2，是的，最好测试有多个记录的场景，因为您有一个循环。你如何模拟你的数据：

let(:user_id1) { 1 }
let(:user_id2) { 2 }
let(:user1) { double(:user, id: user_id1) }
let(:user2) { double(:user, id: user_id2) }

before do
  allow(Result).to receive(:where).with(project_id: project_id).and_return([result1, result2])
  allow(result1).to receive(:user_id).and_return(user_id1)
  allow(result1).to receive(:user_id).and_return(user_id2)
  allow(User).to receive(:find_by).with(id: user_id1).and_return(user1)
  allow(User).to receive(:find_by).with(id: user_id2).and_return(user2)
end

【讨论】：

非常感谢萨钦！您的答案在我的代码中有效。