lib/volt/models/root_models/root_models.rb in volt-0.9.3 vs lib/volt/models/root_models/root_models.rb in volt-0.9.4.pre1

@@ -17,10 +17,15 @@
   Object.const_set("#{base_name}Root", Class.new(BaseRootModel))
 end
 
 module Volt
   class RootModels
+    class_attribute :model_classes
+    self.model_classes = []
+
     def self.add_model_class(klass)
+      self.model_classes << klass
+
       method_name = klass.to_s.underscore.pluralize
 
       # Create a getter for each model class off of root.
       BaseRootModel.send(:define_method, method_name) do
         get(method_name)