Search code examples
ruby-on-railsrspecrspec2rspec-rails

Stub a controller helper method in a helper spec


In my application_controller.rb:

helper_method :current_brand
def current_brand
  @brand ||= Brand.find_by_organization_id(current_user.organization_id)
end

In my helper something_helper.rb

def brands
  return [] unless can? :read, Brand
  # current_brand is called
end

I am writing a spec for something_helper and wish to stub current_brand

describe SomethingHelper do
  before :each do
    helper.stub!(:can?).and_return(true) # This stub works
  end

  it "does the extraordinary" do
    brand = Factory.create(:brand)
    helper.stub!(:current_brand).and_return(brand) # This stub doesnt work
    helper.brands.should_not be_empty
  end
end

Results in NameError: undefined local variable or method 'current_brand' for #<#<Class:0x000001068fd188>:0x0000010316f6f8>

I have tried doing the stub! on self and controller as well. Strangely, when I stub on self, the helper.stub!(:can?).and_return(true) gets unregistered.


Solution

  • OK, how about something else... You're really asking Brand.for_user

    So:

    class Brand
      ...
      def self.for_user(user)
        find_by_organization_id(user.organization_id)
      end
    end
    

    Then, you'd just:

    brand = mock(Brand)
    Brand.stub(:for_user => brand)
    

    Or something similar... If you extract that logic out to something that is easily stubbable, it'll make things easier. A Presenter class, perhaps, or this static method.