metasploit-framework/test/lib/msf_matchers.rb

$:.unshift(File.join((File.dirname(__FILE__))))
require 'regexr'

module MsfTest

module MsfMatchers

  class ContainACompleteTest

    def initialize()
      @r = Regexr.new(true)
    end

    def matches?(data)
      @data = data
      return @r.verify_start_and_end(@data,"meterpreter_functional_test_start", "meterpreter_functional_test_end")
    end

    def failure_message
      "Beginning or end was incorrect."
    end

    def negative_failure_message
      "Expected to find a no beginning or end, but it matched."
    end

  end
  
  def contain_a_complete_test
    ContainACompleteTest.new
  end

  class ContainAllSuccesses

    def initialize(successes=[])
      @successes = successes
      @r = Regexr.new(true)
    end

    def matches?(data)
      @data = data
      @string = @r.find_strings_that_dont_exist_in_data(@data,@successes)
      return true if !@string
      nil
    end

    def failure_message
      "expected all successes, but didn't find '#{@string}'"
    end

    def negative_failure_message
      "expected to miss successes but found'm all :("
    end

    #alias :have_all_successes :contain_all_successes
  end
  
  def contain_all_successes(successes=[])
    ContainAllSuccesses.new(successes)
  end
  
  class ContainNoFailuresExcept

    def initialize(failures=[],exceptions=[])
      @failures = failures
      @exceptions = exceptions
      @r = Regexr.new(true)
    end

    def matches?(data)
      @data = data
      @string = @r.find_strings_that_exist_in_data_except(@data,@failures,@exceptions)
      return true if !@string
      nil
    end

    def failure_message
      "expected no failure to be found, but found this: '#{@string}'"
    end

    def negative_falure_message
      "expected to find failures, but didn't find any :("
    end

    #alias :have_no_failures :contain_no_failures
  end

  def contain_no_failures_except(failures=[],exceptions=[])
    ContainNoFailuresExcept.new(failures,exceptions)
  end

  
end
end