使用Enumerable模块实现简单的测试框架并进行数据统计

Ruby核心库中的Enumerable模块可能听起来很陌生，但实际上大家是应该经常接触到的。这里简单总结一下Enumerable模块的常用方法。

The Enumerable mixin provides collection classes with several traversal and searching methods, and with the ability to sort. The class must provide a method each, which yields successive members of the collection. If Enumerable#max, min, or sort is used, the objects in the collection must also implement a meaningful <=> operator, as these methods rely on an ordering between members of the collection.

Enumerable模块提供了一系列的迭代器方法。使用Enumerable模块很简单，只需要如下步骤：

• mixin Enumerable模块；

• 在使用Enumerable模块的类中定义each方法，保证each方法能够迭代该类中的所有元素；

• 如果需要使用max min sort等方法的话，那么需要定义<=>方法；

<=>方法一般有1个参数且称之为other，如果self > other，则返回1；self == other则返回0，否则返回-1。（这样解释实际上是很不严谨的，但是为了方便可以如此简单理解,高手请尽情鄙视吧...）

下面是简单的用法。下面的代码实现了简单的测试框架——测试用例及测试用例集，我们可以看到Enumerable模块中定义的方法在统计结果时发挥了重要的作用：

require 'pp'
Object.method_undefined(:assert_true) if Object.method_defined?(:assert_true)
Object.method_undefined(:assert_false) if Object.method_defined?(:assert_false)
 
Object.module_eval do
  def assert_true 
    self == true
  end
 
  def assert_false
    self == false
  end  
end
 
class TestCase
  attr_accessor :priority 
  attr_reader :execute_time, :name, :test_result
 
  def initialize(name, priority, &case_step)
    @name = name
    @priority = priority
    @case_step = case_step
  end
 
  def <=>(other)
    @priority <=> other.priority
  end
 
  def execute    
    @test_result = @case_step.call
    @execute_time = @priority
  end
end
 
class TestCaseGroup
  include Enumerable
 
  attr_reader :test_cases, :test_result
 
  def initialize 
    @test_cases = []
    @test_result = {}
  end
 
  def add_case *test_case
    test_case.each do |c|
      @test_cases << c
    end
  end
 
  def delete_case test_cases
    @test_cases.delete test_cases
  end
 
  def each(&block)
    @test_cases.each &block
  end
 
  def execute
    @test_cases.each do |c|
      @test_result[c.name.to_sym] = c.execute
    end    
  end    
end
 
#定义具体用例
case1 = TestCase.new('case1', 1) do
  (1 > 1).assert_false
end
 
case2 = TestCase.new('case2', 2) do
  (1 > 1).assert_true
end
 
case3 = TestCase.new('case3', 3) do
  (1 == 1).assert_true
end
 
#创建用例组并添加用例
test_group = TestCaseGroup.new
test_group.add_case case1, case2, case3
 
#执行用例并输出结果
test_group.execute
p test_group.test_result
 
#是否有case的执行时间大于2？
#from Enumerable
pp test_group.any? {|c| c.execute_time > 2}
 
#是否有case失败
#from Enumerable
pp test_group.any? {|c| c.test_result.eql? false}
 
#统计总共执行时间
#from Enumerable
puts test_group.inject(0){|total, c|total + c.execute_time}
 
#按执行时间长短分组
#小于等于2的一组，其他的一组
pp test_group.partition{|c| c.execute_time < 2}

Enumerable模块中提供了大量的实用迭代方法。这里用到到any? all? inject partition方法只是冰山一角而已。

其他的如each_with_index，collect等方法应用也非常广泛。总而言之Enumerable模块是需要下功夫掌握的1个模块。