ktonon/cog

require 'rspec'
require 'cog/spec_helpers/matchers/match_maker'

module Cog
  module SpecHelpers
    
    # Extra +should+ or +should_not+ matchers for RSpec.
    # Check out {#match_maker} for help writing new matchers.
    module Matchers
      
      # The target {Invocation} should write something to STDERR, indicating an error
      # @return [nil]
      def complain
        match_maker do
          message { "to [write something|not write anything] to STDERR" }
          test { !error.empty? }
        end
      end

      # The target {Invocation} should create a file at the given +path+
      # @param path [String] path to check for a file after the invocation
      # @return [nil]
      def make(path)
        match_maker do
          message { "to [create|not create] #{path}" }
          before do
            @existed = File.exists? path
          end
          test do
            !@existed && File.exists?(path)
          end
        end
      end
      
      # The target {Invocation} should do something, as determined by standard output
      # @return [nil]
      def do_something
        match_maker do
          message { "to [write something|not write anything] to STDOUT" }
          test { !lines.empty? }
        end
      end
      
      # The target {Invocation} should write the given list of lines to standard output
      # @param x [Array<String>, Regexp] a list of lines to match against standard output
      # @return [nil]
      def output(x)
        match_maker do
          message do
            if x.is_a? Regexp
              "to [write|not write] #{x.inspect} to STDOUT"
            else
              "to [write|not write] #{x.join "\n"} to STDOUT"
            end
          end
          test do
            if x.is_a? Regexp
              x =~ lines.join("\n")
            else
              lines.zip(x).all? {|a, b| a.strip == b.to_s.strip}
            end
          end
        end
      end
      
      # The target {Invocation} should output the default help text
      # @return [nil]
      def show_help
        match_maker do
          message { "to [show|not show] the default help text, got #{lines.first.inspect}" }
          test { (/help.*code gen/ =~ lines[1]) }
        end
      end
      
    end
  end
end

RSpec.configure do |config|
  config.include Cog::SpecHelpers::Matchers
end