rspec 存根 current_company

Question

我有一个 Rails 3 项目，我想在其中存储在会话变量中选择的当前公司。

我正在使用员工控制器规范，并且想暂时删除 current_company，因为我正在隔离员工新控制器操作的规范示例。

it "should call current_company" do 
  company = mock_model(Company, :id => "1")
  controller.should_receive(:current_company).and_return(company)
  get :new
end

这是我对员工控制器的新操作，

  def new
    @staff = Staff.new
    @staff.company_id = current_company.id
  end

我不断收到错误，

Failure/Error: get :new
     NameError:
       undefined local variable or method `current_company' for #<StaffsController:0x000000028d6ad8>

我也尝试将其删除，而不是使用 should_receive

  controller.stub!(:current_company).and_return(company)

我收到相同的错误。

Answer 1

你的代码对我来说看起来很好，它应该可以工作。一定还有其他我们没有看到的问题。我注意到控制器名称是“StaffsController”——正确吗？仔细检查控制器的名称和相应的规格——它们应该是相同的。

Answer 2

我认为它在“应该成功”的示例/测试中被轰炸了，所以我把我的存根放在了之前的块中。

require 'spec_helper'

describe StaffsController do

  describe "GET 'new'" do
    let(:staff) { mock_model(Staff, :company_id= => nil)}
    let(:company) { mock_model(Company, :id => 1)}

    before do
      Staff.stub!(:new).and_return(staff)
      controller.stub!(:current_company).and_return(company)
    end

    it "should be successful" do
      get :new
      response.should be_success
    end

    it "should call current_company" do 
      controller.should_receive(:current_company).and_return(company)
      get :new
    end
  end
end

这适用于：

class StaffsController < ApplicationController
  def new
    @staff = Staff.new
    current_company.id
  end
end