need to add extension_module after cloning in has_set

ninjudd · Sep 12, 2012 · 20b3c0d · 20b3c0d
1 parent d2f259f
commit 20b3c0d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/lib/model_set.rb b/lib/model_set.rb
@@ -761,7 +761,9 @@ def set.parent_model
       if options[:clone] == false or args.include?(:no_clone)
         @model_set_cache[name]
       else
-        @model_set_cache[name].clone
+        set = @model_set_cache[name].clone
+        set.extend(extension_module) if extension_module
+        set
       end
     end
 

diff --git a/test/model_set_test.rb b/test/model_set_test.rb
@@ -199,6 +199,7 @@ def add_birthday!
       underdog     = Superpet.create(:name => 'Underdog', :owner_id => hero.id, :species => 'dog')
 
       set = hero.pets
+
       assert_equal ['mouse', 'dog', 'dog'], set.collect {|pet| pet.species}
 
       assert_equal [sammy.id, underdog.id], set.dogs!.ids