test/ts_dbd.rb in dbd-mysql-0.4.2 vs test/ts_dbd.rb in dbd-mysql-0.4.3

@@ -1,5 +1,7 @@
+require 'rubygems'
+gem 'test-unit'
 # figure out what tests to run
 require 'yaml'
 require 'test/unit/testsuite'
 require 'test/unit/ui/console/testrunner'
 
@@ -15,10 +17,20 @@
         template &&= template.chomp
         return AssertionMessage.new(head, template, arguments)
     end
 end
 
+class Class
+    def name=(klass_name)
+        @name = klass_name
+    end
+
+    def name
+        return @name || super
+    end
+end
+
 module DBDConfig
     @testbase = { }
     @current_dbtype = nil
 
     def self.get_config
@@ -66,10 +78,11 @@
         return @testbase[klass_name]
     end
 
     def self.set_testbase(klass_name, klass)
         @testbase[klass_name] = klass
+        klass.name = klass_name.to_s
     end
 
     def self.suite
         @suite ||= []
     end
@@ -108,10 +121,10 @@
             # base.rb is special, see DBD_TESTS
             require "dbd/#{dbtype}/base.rb"
             Dir["dbd/#{dbtype}/test*.rb"].each { |file| require file }
             # run the general tests
             DBDConfig.current_dbtype = dbtype.to_sym
-            Dir["dbd/general/test*.rb"].each { |file| load file; DBDConfig.suite << @class }
+            Dir["dbd/general/test*.rb"].each { |file| load file; @class.name = file; DBDConfig.suite << @class }
         end
     elsif !config["dbtypes"]
         warn "Please see test/DBD_TESTS for information on configuring DBD tests."
     end
 end