Brad
Brad

Reputation: 5488

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.

Upvotes: 0

Views: 1583

Answers (2)

Jesse Wolgamott
Jesse Wolgamott

Reputation: 40277

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.

Upvotes: 1

fuzzyalej
fuzzyalej

Reputation: 5973

Have you tried something similar to:

ApplicationController.stub!(:current_brand).and_return(brand)

Upvotes: 0

Related Questions