diff --git a/lib/draper/base.rb b/lib/draper/base.rb index 1999513..25558ec 100644 --- a/lib/draper/base.rb +++ b/lib/draper/base.rb @@ -43,8 +43,12 @@ module Draper end private + def model + @model + end + def select_methods - self.allowed || (@model.public_methods - denied) + self.allowed || (model.public_methods - denied) end def build_methods @@ -52,7 +56,7 @@ module Draper unless self.respond_to?(method) (class << self; self; end).class_eval do define_method method do |*args, &block| - @model.send method, *args, &block + model.send method, *args, &block end end end diff --git a/spec/samples/decorator_application_helper.rb b/spec/samples/decorator_application_helper.rb deleted file mode 100644 index daae87c..0000000 --- a/spec/samples/decorator_application_helper.rb +++ /dev/null @@ -1,5 +0,0 @@ -class DecoratorApplicationHelper < Draper::Base - def uses_hello - self.hello - end -end