dm-unlazy 0.0.1 → 0.0.1.1

data/lib/dm-unlazy/model.rb

@@ -22,9 +22,18 @@ class Model
 	def method_missing (id, *args, &block)
 		result = @model.__send__ id, *args, &block
 
-		if [Associations::OneToMany::Collection, Associations::ManyToMany::Collection].any? { |k| k === result }
-			result.source.define_singleton_method(:saved?) { true }
+		if Associations::OneToMany::Collection === result
+			class << result.source
+				def saved?
+					true
+				end
+			end
+
 			result.reload
+
+			class << result.source
+				remove_method :saved?
+			end
 		end
 
 		result

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: dm-unlazy
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.1.1
   prerelease: 
 platform: ruby
 authors:
@@ -13,7 +13,7 @@ date: 2012-02-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: dm-core
-  requirement: &14330280 !ruby/object:Gem::Requirement
+  requirement: &19433700 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *14330280
+  version_requirements: *19433700
 - !ruby/object:Gem::Dependency
   name: rake
-  requirement: &14329060 !ruby/object:Gem::Requirement
+  requirement: &19432600 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,10 +32,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *14329060
+  version_requirements: *19432600
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &14327500 !ruby/object:Gem::Requirement
+  requirement: &19431160 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -43,7 +43,7 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *14327500
+  version_requirements: *19431160
 description: 
 email: meh@paranoici.org
 executables: []