From 5dedd370e668aa86faac560718b455c3942fe83f Mon Sep 17 00:00:00 2001 From: Jeff Casimir Date: Sat, 23 Jul 2011 07:33:08 -0700 Subject: [PATCH] Add private accessor .model method --- lib/draper/base.rb | 8 ++++++-- spec/samples/decorator_application_helper.rb | 5 ----- 2 files changed, 6 insertions(+), 7 deletions(-) delete mode 100644 spec/samples/decorator_application_helper.rb 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