Changeset 31404

Timestamp:

03/20/09 17:40:03 (4 years ago)

Author:

frsyuki

Message:

lang/ruby/chukan: Chukan::Test experimental impl.

Files:

• lang/ruby/chukan/trunk/chukan.rb (modified) (1 diff)

lang/ruby/chukan/trunk/chukan.rb

@@ -290 +290 @@
                 Remote.new(host, user, key)
         end
+
+
+        module Test
+                @@start = nil
+                @@cases = Hash.new {|hash,key| hash[key] = [0,0,0] }
+                @@count = 0
+                @@data  = nil
+
+                if ENV["TERM"] =~ /color/i && $stdout.stat.chardev?
+                        module Color
+                                SUCCESS = "\e[0;32m"
+                                FAIL    = "\e[1;33m"
+                                ERROR   = "\e[0;31m"
+                                NORMAL  = "\e[00m"
+                        end
+                else
+                        module Color
+                                SUCCESS = ""
+                                FAIL    = ""
+                                ERROR   = ""
+                                NORMAL  = ""
+                        end
+                end
+        
+                def self.report
+                        proc {
+                                finish = Time.now
+                                puts "\n1..#{@@count}"
+                                puts "Finished in #{finish - @@start} seconds."
+                                puts ""
+                                succes  = @@cases.to_a.inject(0) {|r,(n,c)| r + c[0] }
+                                failure = @@cases.to_a.inject(0) {|r,(n,c)| r + c[1] }
+                                error   = @@cases.to_a.inject(0) {|r,(n,c)| r + c[2] }
+                                # FIXME color
+                                puts "#{@@cases.size} tests, " +
+                                        "#{succes+failure+error} assertions, " +
+                                        "#{failure} failures, " +
+                                        "#{error} errors"
+                        }
+                end
+        
+                def self.included(*mod)
+                        ObjectSpace.define_finalizer(mod, report)
+                        @@start = Time.now
+                end
+        
+                def test(name, directive = nil, &block)
+                        @@count += 1
+                        if yield
+                                tap(Color::SUCCESS, "ok", @@count, name, directive)
+                                @@cases[name][0] += 1
+                        else
+                                tap(Color::FAIL, "not ok", @@count, name, directive)
+                                print_backtrace(caller, "test failed")
+                                @@cases[name][1] += 1
+                        end
+                rescue
+                        tap(Color::ERROR, "not ok", @@count, name, directive)
+                        print_backtrace($!.backtrace, "#{$!} (#{$!.class})")
+                        @@cases[name][2] += 1
+                end
+
+                def data
+                        require 'yaml'
+                        @@data ||= YAML.load_stream(DATA.read.gsub(/(^\t+)/) {
+                                '  ' * $+.length
+                        }).documents.map {|obj|
+                                obj.each_pair {|k,v|
+                                        (class<<obj; self; end).instance_eval do
+                                                define_method(k) { v }
+                                        end rescue nil
+                                } rescue obj
+                        }
+                end
+
+                def run(&block)
+                        data.each {|d|
+                                yield d
+                        }
+                end
+
+                private
+                def tap(color, stat, count, name, directive = nil)
+                        if directive
+                                directive = "  # #{directive.to_s.upcase}"
+                        end
+                        print "#{color}\n#{stat} #{count} #{name}#{directive}\n#{Color::NORMAL}"
+                end
+
+                def print_backtrace(trace, msg)
+                        puts "#{trace.shift}: #{msg}"
+                        trace.each {|c|
+                                unless c.to_s.include?(__FILE__)
+                                        puts "\tfrom  #{c}"
+                                end
+                        }
+                end
+        end
+
 end