1 files changed, 44 insertions, 0 deletions
diff --git a/lib/test/unit/collector/objectspace.rb b/lib/test/unit/collector/objectspace.rb
new file mode 100644
index 0000000000..32b6332689
--- /dev/null
+++ b/lib/test/unit/collector/objectspace.rb
@@ -0,0 +1,44 @@
+# Author:: Nathaniel Talbott.
+# Copyright:: Copyright (c) 2000-2003 Nathaniel Talbott. All rights reserved.
+# License:: Ruby license.
+
+module Test
+  module Unit
+    module Collector
+      class ObjectSpace
+        NAME = 'collected from the ObjectSpace'
+        
+        def initialize(source=::ObjectSpace)
+          @source = source
+        end
+        
+        def collect(name=NAME)
+          suite = TestSuite.new(name)
+          tests = []
+          @source.each_object(Class) do |klass|
+            tests.concat(klass.suite.tests) if(Test::Unit::TestCase > klass)
+          end
+          tests.each{|test| suite << test if(include(test))}
+          suite
+        end
+        
+        def include(test)
+          return true unless(@filters)
+          @filters.each do |filter|
+            return true if(filter.call(test))
+          end
+          false
+        end
+        
+        def filter=(filters)
+          @filters = case(filters)
+            when Proc
+              [filters]
+            when Array
+              filters
+          end
+        end
+      end
+    end
+  end
+end