Rspec的。测试后,测试代码自动启动

时间:2016-10-26 07:25:58

标签: ruby rspec mocking sensu

我在测试Sensu插件时遇到问题。 每当我启动rspec测试插件时,它都会测试它,但无论如何,在测试结束时,原始插件会自动启动。所以我在我的控制台中:

Finished in 0 seconds (files took 0.1513 seconds to load) 
1 example, 0 failures 
CheckDisk OK:       # This comes from the plugin

简短说明我的系统如何工作: 插件调用系统'wmic'命令,处理它,检查有关磁盘参数的条件并返回退出状态(ok,critical等) Rspec模拟来自系统的响应并设置为插件的输入。最后,当给出模拟输入时,rspec会检查插件退出状态。

我的插件看起来像这样:

require 'rubygems' if RUBY_VERSION < '1.9.0'
require 'sensu-plugin/check/cli'

class CheckDisk < Sensu::Plugin::Check::CLI
  def initialize
    super
    @crit_fs = []
  end

  def get_wmic
    `wmic volume where DriveType=3 list brief`
  end

  def read_wmic
    get_wmic
    # do something, fill the class variables with system response  
  end

  def run
   severity = "ok"
   msg = ""
   read_wmic
   unless @crit_fs.empty?
     severity = "critical"  
  end  
  case severity
    when /ok/
      ok msg
    when /warning/
      warning msg
    when /critical/
      critical msg
    end
  end
end

这是我在Rspec中的测试:

require_relative '../check-disk.rb'
require 'rspec'

  def loadFile
    #Load template of system output when ask 'wmic volume(...)
  end

  def fillParametersInTemplate (template, parameters)
    #set mocked disk parameters in template
  end

  def initializeMocks (options)
    mockedSysOutput = fillParametersInTemplate @loadedTemplate, options 
    po = String.new(mockedSysOutput)
    allow(checker).to receive(:get_wmic).and_return(po) #mock system call here
  end

  describe CheckDisk do
    let(:checker) { described_class.new }
    before(:each) do   
       @loadedTemplate = loadFile   
       def checker.critical(*_args)
          exit 2
       end    
     end

  context "When % of free disk space = 10 >" do 
    options = {:diskName => 'C:\\', :diskSize => 1000, :diskFreeSpace => 100}       
    it 'Returns ok exit status ' do      
      begin                   
        initializeMocks options
        checker.run 
      rescue SystemExit => e 
        exit_code = e.status   
      end 
      expect(exit_code).to eq 0
    end  
  end 
end

我知道我可以在最后一个示例之后放置“exit 0”,但这不是一个解决方案,因为当我尝试启动许多spec文件时,它将在第一个之后退出。如何在不运行插件的情况下开始测试?也许有人可以帮助我并展示如何处理这样的问题? 谢谢。

1 个答案:

答案 0 :(得分:0)

您可以存根原始插件调用,并可选择返回虚拟对象:

allow(SomeObject).to receive(:method) # .and_return(double)

您可以将其放在before块中,以确保所有断言都能共享代码。

另一件事是,当您的代码因错误而中止时,您正在使用rescue块来捕获这种情况。您应该使用raise_error匹配器代替:

expect { run }.to raise_error(SystemExit)