Ruby元编程：动态添加类属性及其实际应用

上个星期测试道的Monkey老师和我聊到测试用例参数过多的问题，其实这样的问题在我这里也同样经历过。比如我的测试用例必须面对不同的测试环境，每个环境有无数的参数，开发的最初阶段，因为参数少，所以就放在执行的命令行里，随着测试用例的不断增长，参数从4-5个增长到30多个，而且每个用例使用的参数也不完全相同，有使用ABCD的，有使用ADHJ的。另外有些参数想传一个数组进去，用命令行参数的方法就很难处理。

经过考虑，果断的使用配置文件来解决问题。选择配置文件当时有两个方案，一个是直接写成Ruby代码，但是考虑到要和自动化测试框架共享配置文件，所以最后决定使用YAML格式。Ruby对YAML的支持非常好，你可以用如下方式访问某个配置参数Configs['Host']['IP']，考虑到让我们的测试代码更直观，我们希望能用Configs.Host_IP的方式访问环境参数。

YAML配置文件例子：

Storage:
  IP: 192.168.1.1
  Port: 80
  SSL_Port: 443
  Service_Port: 2000

DB:
  IP: 192.168.1.2
  User: Tom
  Password: test

Manifests:
  - IP: 192.168.2.1
  - IP: 192.168.2.2
  - IP: 192.168.2.3
  - IP: 192.168.2.4

我们希望能用Configs.Storage_IP，Configs.DB_Password这样的方法拿到参数，对于Manifests，我们的输入是数组，则希望能用Configs.Manifests[i]['IP']的方式访问。利用Ruby的元编程特性，在我们读取YAML文件解析生成数组后，我们就可以遍历每个元素然后动态为Configs添加属性。代码如下：

 require 'yaml'

 module ConfigParser
   class << self
     def add_attribute(klass, symbol)
       codes = %Q{
         def #{symbol}
           return @#{symbol}
         end

         def #{symbol}=(value)
           @#{symbol} = value
         end
       }

       klass.instance_eval(codes)
     end

     def expand_configs(configs = {})
       configs.each do |key, value|
         expand_sub_configs(key, value)
       end
     end

     def expand_sub_configs(prefix, configs)
       if configs.class != Hash
         add_attribute(Configs, prefix)
         eval("Configs.#{prefix} = configs")
       else
         configs.each do |key, value|
           expand_sub_configs(prefix + '_' + key, value)
         end
       end
     end
   end
 end

 if ARGV.size != 1
   puts "Usage: ..."
   exit(-1)
 end

 Configs = Class.new

 ConfigParser.expand_configs(YAML.load(File.open(ARGV[0])))

 puts 'Storage information:'
 puts " #{Configs.Storage_IP}"
 puts "  #{Configs.Storage_Port}"
 puts "  #{Configs.Storage_SSL_Port}"
 puts "  #{Configs.Storage_Service_Port}\n"

 puts 'DB information: '
 puts "  #{Configs.DB_IP}"
 puts "  #{Configs.DB_User}"
 puts "  #{Configs.DB_Password}\n"

 puts 'Manifests information: '
 Configs.Manifests.each do |info|
   info.each do |k, v|
     puts "  #{k}: #{v}"
   end
 end

执行命令ruby test.rb config.yml得到输出如下：

Storage information:
 192.168.1.1
  80
  443
  2000
DB information:
  192.168.1.2
  Tom
  test
Manifests information:
  IP: 192.168.2.1
  IP: 192.168.2.2
  IP: 192.168.2.3
  IP: 192.168.2.4