lib/puffer/extensions/mapper.rb in puffer-0.0.15 vs lib/puffer/extensions/mapper.rb in puffer-0.0.16

@@ -19,15 +19,17 @@
         def resources_with_puffer *resources, &block
           puffer_resources(*Marshal.load(Marshal.dump(resources)), &block) || resources_without_puffer(*resources, &block)
         end
 
         def puffer_controller controller
-          puffer = Rails.application.routes.puffer
-          prefix = @scope[:module]
-          puffer[prefix] ||= ActiveSupport::OrderedHash.new
-          puffer[prefix][controller.configuration.group] ||= []
-          puffer[prefix][controller.configuration.group] << controller
+          if controller.configuration.group
+            puffer = Rails.application.routes.puffer
+            namespace = @scope[:module]
+            puffer[namespace] ||= ActiveSupport::OrderedHash.new
+            puffer[namespace][controller.configuration.group] ||= []
+            puffer[namespace][controller.configuration.group] << controller
+          end
         end
 
         def puffer_resource(*resources, &block)
           options = resources.extract_options!
 
@@ -156,17 +158,17 @@
       def self.included base
         base.class_eval do
           include InstanceMethods
 
           alias_method_chain :clear!, :puffer
-          attr_accessor_with_default :puffer, {}
+          attr_accessor_with_default :puffer, ActiveSupport::OrderedHash.new
         end
       end
 
       module InstanceMethods
 
         def clear_with_puffer!
-          self.puffer = {}
+          self.puffer = ActiveSupport::OrderedHash.new
           clear_without_puffer!
         end
 
       end