Merge pull request #3 from Burgestrand/fix_ref_mock_gc

Ref::Mock.gc with no arguments does not trigger fake garbage collect for all objects

Author: bdurand

lib/ref/mock.rb

@@ -107,11 +107,16 @@ def object_space # :nodoc:
       # Simulate garbage collection of the objects passed in as arguments. If no objects
       # are specified, all objects will be reclaimed.
       def gc(*objects)
-        objects = object_space.keys if objects.empty?
-        objects.each do |obj|
-          finalizers = object_space.delete(obj.__id__)
+        objects = if objects.empty?
+          object_space.keys
+        else
+          objects.map { |obj| obj.__id__ }
+        end
+
+        objects.each do |id|
+          finalizers = object_space.delete(id)
           if finalizers
-            finalizers.each{|finalizer| finalizer.call(obj.__id__)}
+            finalizers.each{|finalizer| finalizer.call(id)}
           end
         end
       end

test/mock_test.rb

@@ -0,0 +1,33 @@
+require File.expand_path("../test_helper", __FILE__)
+
+class TestMock < Test::Unit::TestCase
+  def test_gc_with_argument
+    Ref::Mock.use do
+      obj_1 = Object.new
+      obj_2 = Object.new
+
+      ref_1 = Ref::WeakReference.new(obj_1)
+      ref_2 = Ref::WeakReference.new(obj_2)
+
+      Ref::Mock.gc(obj_1)
+
+      assert_nil ref_1.object
+      assert_equal ref_2.object, obj_2
+    end
+  end
+
+  def test_gc_with_no_argument
+    Ref::Mock.use do
+      obj_1 = Object.new
+      obj_2 = Object.new
+
+      ref_1 = Ref::WeakReference.new(obj_1)
+      ref_2 = Ref::WeakReference.new(obj_2)
+
+      Ref::Mock.gc
+
+      assert_nil ref_1.object
+      assert_nil ref_2.object
+    end
+  end
+end