Lazy collection model

gabynaiman · Sep 30, 2016 · a69b68a · a69b68a
1 parent 1505169
commit a69b68a
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/lib/rasti/db/collection.rb b/lib/rasti/db/collection.rb
@@ -15,7 +15,13 @@ def primary_key
         end
 
         def model
-          @model ||= Consty.get(demodulize(singularize(name)), self)
+          if @model.is_a? Class
+            @model
+          elsif @model
+            @model = Consty.get @model, self
+          else
+            @model = Consty.get demodulize(singularize(name)), self
+          end
         end
 
         def relations

diff --git a/spec/collection_spec.rb b/spec/collection_spec.rb
@@ -25,6 +25,14 @@
       collection_class.primary_key.must_equal :code
     end
 
+    it 'Lazy model name' do
+      collection_class = Class.new(Rasti::DB::Collection) do
+        set_model :User
+      end
+
+      collection_class.model.must_equal User
+    end
+
   end
 
   describe 'Insert, Update and Delete' do