如何定义用于操作Puppet目录对象的辅助方法

时间:2017-05-07 04:43:19

标签: rspec puppet rspec-puppet

我有一堆重复的代码来操作Puppet目录对象(注意重复的content =行):

require 'nokogiri'

describe 'role::jenkins' do
  before(:each) do
    @jobs = catalogue.resource_keys.select{|k,v| k == 'Jenkins::Jobs'}.map{|k,v| v}
  end

  it 'Jenkins jobs should be valid XML' do
    @jobs.each do |j|
      content = catalogue.resource('file', "/tmp/#{j}.xml").send(:parameters)[:content]
      result = Nokogiri::XML(content).errors.empty?
      if ! result
        puts "    Job #{j} does NOT have valid XML"
      end
      expect(result).to be true
    end
  end 

  it "XML should contain a variables.json snippet that is valid JSON" do
    @jobs.each do |j|
      content = catalogue.resource('file', "/tmp/#{j}.xml").send(:parameters)[:content]
      if content.match(/cat << EOF > #{json_file}.*?EOF/m)
        json_snippet = content.match(/#{json_file}(.*?)EOF/m)[1]
        expect { JSON.parse(json_snippet) }.to_not raise_error
      end
    end
  end
end

可以看出,我已将长查询移动到before(:each)块并将其保存在实例变量中。这使它在it块中可用。

我不明白的是如何为content =行定义方法,例如:

def content(file_name)
  catalogue.resource('file', file_name).send(:parameters)[:content]
end

如果我知道该怎么做,我可以大大清理这段代码。我无法弄清楚的是我可以放置这个def块,如果有可能的话。

1 个答案:

答案 0 :(得分:0)

我犯了一个愚蠢的错误(以为我已经尝试了一些事实,但实际上我没有)。

答案就是将def放在before块中:

require 'nokogiri'

describe 'role::jenkins' do
  before(:each) do
    @jobs = catalogue.resource_keys.select{|k,v| k == 'Jenkins::Jobs'}.map{|k,v| v}

    def content(file_name)
      catalogue.resource('file', file_name).send(:parameters)[:content]
    end
  end

  it 'Jenkins jobs should be valid XML' do
    @jobs.each do |j|
      result = Nokogiri::XML(content("/tmp/#{j}.xml").errors.empty?
      if ! result
        puts "    Job #{j} does NOT have valid XML"
      end
      expect(result).to be true
    end
  end 

  it 'XML should contain a variables.json snippet that is valid JSON' do
    @jobs.each do |j|
      content = content("/tmp/#{j}.xml")
      if content.match(/cat << EOF > #{json_file}.*?EOF/m)
        json_snippet = content.match(/#{json_file}(.*?)EOF/m)[1]
        expect { JSON.parse(json_snippet) }.to_not raise_error
      end
    end
  end
end

如果有人能看到任何进一步的改进,请告诉我!