lib/bourgeois/view_helper.rb in bourgeois-0.1.1 vs lib/bourgeois/view_helper.rb in bourgeois-0.1.2

@@ -1,17 +1,25 @@
 module Bourgeois
   module ViewHelper
-    # Wrap a resource or a collection into its related delegator
+    # Wrap a resource or a collection into its related presenter
     #
     # @example
     #   present User.new(name: 'Remi') do |user|
     #     puts user.inspect # => #<UserPresenter object=#<User name="Remi>>
     #     puts user.name # => Remi
     #   end
     def present(object, klass = nil, &blk)
       return object.map { |o| present(o, klass, &blk) } if object.respond_to?(:to_a)
 
-      klass ||= "#{object.class}Presenter".constantize
+      if klass.blank?
+        begin
+          klass_name = "#{object.class}Presenter"
+          klass = klass_name.constantize
+        rescue ::NameError
+          raise UnknownPresenter.new(klass_name)
+        end
+      end
+
       presenter = klass.new(object, self)
       yield presenter if block_given?
 
       presenter
     end