ansible_spec_plus 1.0.0

data/lib/ansiblespec_helper.rb

@@ -0,0 +1,34 @@
+module AnsibleSpecHelper
+  def self.get_properties
+    playbook, inventoryfile = AnsibleSpec.load_ansiblespec
+
+    hosts = AnsibleSpec.load_targets(inventoryfile)
+    properties = AnsibleSpec.load_playbook(playbook)
+
+    properties.each do |var|
+      var["hosts_childrens"] = hosts["hosts_childrens"]
+      var["group"] = var["hosts"]
+      if var["hosts"].to_s == "all"
+        var["hosts"] = hosts.values.flatten
+      elsif hosts.has_key?("#{var["hosts"]}")
+        var["hosts"] = hosts["#{var["hosts"]}"]
+      elsif var["hosts"].instance_of?(Array)
+        tmp_host = var["hosts"]
+        var["hosts"] = []
+        tmp_host.each do |v|
+          if hosts.has_key?("#{v}")
+            hosts["#{v}"].map {|target_server| target_server["hosts"] = v}
+            var["hosts"].concat hosts["#{v}"]
+          end
+        end
+        if var["hosts"].size == 0
+          properties = properties.compact.reject{|e| e["hosts"].length == 0}
+        end
+      else
+        var["hosts"] = []
+      end
+    end
+
+    return properties
+  end
+end

data/lib/helpers/buffered_logger.rb

@@ -0,0 +1,110 @@
+require 'logger'
+require_relative 'ring_buffer'
+
+module Helpers
+
+  class BufferedLogger < Logger
+
+    #
+    # :call-seq:
+    #   Logger.new(name, shift_age = 7, shift_size = 1048576)
+    #   Logger.new(name, shift_age = 'weekly')
+    #
+    # === Args
+    #
+    # +logdev+::
+    #   The log device.  This is a filename (String) or IO object (typically
+    #   +STDOUT+, +STDERR+, or an open file).
+    # +shift_age+::
+    #   Number of old log files to keep, *or* frequency of rotation (+daily+,
+    #   +weekly+ or +monthly+).
+    # +shift_size+::
+    #   Maximum logfile size (only applies when +shift_age+ is a number).
+    #
+    # === Description
+    #
+    # Create an instance.
+    #
+    def initialize(logdev, shift_age = 0, shift_size = 1048576)
+      super
+      if logdev == STDOUT
+        @stdout = STDOUT
+      end
+
+      @last_messages = RingBuffer.new(200)
+    end
+
+    #
+    # :call-seq:
+    #   Logger#add(severity, message = nil, progname = nil) { ... }
+    #
+    # === Args
+    #
+    # +severity+::
+    #   Severity.  Constants are defined in Logger namespace: +DEBUG+, +INFO+,
+    #   +WARN+, +ERROR+, +FATAL+, or +UNKNOWN+.
+    # +message+::
+    #   The log message.  A String or Exception.
+    # +progname+::
+    #   Program name string.  Can be omitted.  Treated as a message if no
+    #   +message+ and +block+ are given.
+    # +block+::
+    #   Can be omitted.  Called to get a message string if +message+ is nil.
+    #
+    # === Return
+    #
+    # When the given severity is not high enough (for this particular logger),
+    # log no message, and return +true+. Even if the severity is not high enough
+    # we save each message in a ring buffer.
+    #
+    # === Description
+    #
+    # Log a message if the given severity is high enough. This is the generic
+    # logging method.  Users will be more inclined to use #debug, #info, #warn,
+    # #error, and #fatal.
+    #
+    # <b>Message format</b>: +message+ can be any object, but it has to be
+    # converted to a String in order to log it.  Generally, +inspect+ is used
+    # if the given object is not a String.
+    # A special case is an +Exception+ object, which will be printed in detail,
+    # including message, class, and backtrace.  See #msg2str for the
+    # implementation if required.
+    #
+    # === Bugs
+    #
+    # * Logfile is not locked.
+    # * Append open does not need to lock file.
+    # * If the OS supports multi I/O, records possibly may be mixed.
+    #
+    def add(severity, message = nil, progname = nil, &block)
+      super
+      severity ||= UNKNOWN
+      progname ||= @progname
+      if message.nil?
+        if block_given?
+          message = yield
+        else
+          message = progname
+          progname = @progname
+        end
+      end
+      @last_messages.push(format_message(format_severity(severity), Time.now, progname, message))
+      @stdout.flush if @stdout
+      true
+    end
+
+    #
+    # === Return
+    #
+    # Get array of last log messages. Even log messages with low severity are included.
+    # The number of returned messages is limited by the size of the underlying ring buffer.
+    #
+    # === Description
+    #
+    # After calling this method the underling ring buffer is empty.
+    #
+    def last_messages
+      @last_messages.flush
+    end
+  end
+end

data/lib/helpers/color_formatter.rb

@@ -0,0 +1,18 @@
+require_relative 'colorize'
+require 'logger'
+
+module Helpers
+  # colorful log formatter, also logs thread object id instead of process id
+  class ColorFormatter < Logger::Formatter
+
+    SCHEMA = %w(nothing green yellow red purple cyan)
+
+    def call(severity, time, progname, msg)
+      level = ::Logger::Severity.const_get(severity)
+      color = SCHEMA[level]
+      text = Format % [severity[0..0], format_datetime(time), Thread.current.object_id, severity, progname,
+        msg2str(msg)]
+      color ? Colorize.colorize(color, text) : text
+    end
+  end
+end

data/lib/helpers/colorize.rb

@@ -0,0 +1,52 @@
+module Helpers
+  module Colorize
+    module Colors
+      NOTHING      = '0;0'
+      BLACK        = '0;30'
+      DARK_RED     = '0;31'
+      DARK_GREEN   = '0;32'
+      BROWN        = '0;33'
+      BLUE         = '0;34'
+      PURPLE       = '0;35'
+      CYAN         = '0;36'
+      GRAY         = '0;37'
+      LIGHT_GRAY   = '0;37'
+      DARK_GRAY    = '1;30'
+      RED          = '1;31'
+      GREEN        = '1;32'
+      YELLOW       = '1;33'
+      LIGHT_BLUE   = '1;34'
+      LIGHT_PURPLE = '1;35'
+      LIGHT_CYAN   = '1;36'
+      WHITE        = '1;37'
+    end
+
+    class << self
+      def colorize(color, text)
+        color = Colorize::Colors.const_get(color.to_s.upcase)
+        color ? colorful_text(color, text) : text
+      rescue
+#        puts color
+#        puts $?
+        text
+      end
+
+      def method_missing(symbol, *args)
+        color = Colorize::Colors.const_get(color.to_s.upcase)
+        return colorful_text(color, *args) if color
+        super
+      rescue
+        super
+#        puts symbol
+#        puts $?
+#        args
+      end
+
+      private
+
+      def colorful_text(color, text)
+        "\e[#{color}m#{text}\e[0;0m"
+      end
+    end
+  end
+end

data/lib/helpers/log.rb

@@ -0,0 +1,80 @@
+require 'logger'
+require_relative 'color_formatter'
+require_relative 'buffered_logger'
+
+module Helpers
+  module Log
+    class << self
+      def log
+        unless @logger
+          @logger = BufferedLogger.new($stdout)
+          @logger.formatter = ColorFormatter.new if $stdout.tty?
+        end
+        @logger
+      end
+
+      def log=(logger)
+        @logger = logger
+        log.info('changed logger')
+      end
+
+      def debug?
+        !@no_debug
+      end
+
+      def set_debug(debug, level = ::Logger::INFO)
+        @no_debug = !debug
+        log.level = level
+      end
+
+      def last_messages
+        return [] unless @logger
+        @logger.last_messages
+      end
+    end
+
+    def self.included(base)
+      class << base
+        def log
+          Log.log
+        end
+
+        def log=(logger)
+          Log.log=(logger)
+        end
+
+        def debug?
+          Log.debug?
+        end
+
+        def set_debug(debug, level = Logger::INFO)
+          Log.set_debug(debug, level)
+        end
+
+        def last_messages
+          Log.last_messages
+        end
+      end
+    end
+
+    def log
+      Log.log
+    end
+
+    def log=(logger)
+      Log.log=(logger)
+    end
+
+    def debug?
+      Log.debug?
+    end
+
+    def set_debug(debug, level = ::Logger::INFO)
+      Log.set_debug(debug, level)
+    end
+
+    def last_messages
+      Log.last_messages
+    end
+  end
+end

data/lib/helpers/ring_buffer.rb

@@ -0,0 +1,65 @@
+module Helpers
+  class RingBuffer
+    def initialize(size)
+      @size = size
+      @start = 0
+      @count = 0
+      @buffer = Array.new(size)
+      @mutex = Mutex.new
+    end
+
+    def full?
+      @count == @size
+    end
+
+    def empty?
+      @count == 0
+    end
+
+    def push(value)
+      @mutex.synchronize do
+        stop = (@start + @count) % @size
+        @buffer[stop] = value
+        if full?
+          @start = (@start + 1) % @size
+        else
+          @count += 1
+        end
+        value
+      end
+    end
+    alias :<< :push
+
+    def shift
+      @mutex.synchronize do
+        remove_element
+      end
+    end
+
+    def flush
+      values = []
+      @mutex.synchronize do
+        until empty?
+          values << remove_element
+        end
+      end
+      values
+    end
+
+    def clear
+      @buffer = Array.new(@size)
+      @start = 0
+      @count = 0
+    end
+
+    private
+
+    def remove_element
+      return nil if empty?
+      value, @buffer[@start] = @buffer[@start], nil
+      @start = (@start + 1) % @size
+      @count -= 1
+      value
+    end
+  end
+end

data/lib/src/.ansiblespec

@@ -0,0 +1,5 @@
+---
+
+- playbook: site.yml
+  inventory: hosts
+  hash_behaviour: merge

data/lib/src/.rspec

@@ -0,0 +1,3 @@
+--color
+--format documentation
+--format json --out report.json

data/lib/src/spec/spec_helper.rb

@@ -0,0 +1,92 @@
+require 'serverspec'
+require 'net/ssh'
+require 'ansible_spec'
+require 'winrm'
+
+ENV['BACKEND'] = 'ssh' if ! ENV['BACKEND'] || ENV['BACKEND'] == ''
+
+#
+# Set ansible variables to serverspec property
+#
+
+host = ENV['TARGET_HOST']
+hosts = ENV["TARGET_HOSTS"]
+
+group_idx = ENV['TARGET_GROUP_INDEX'].to_i
+ssh_config_file = AnsibleSpec.get_ssh_config_file
+
+connection = ENV['TARGET_CONNECTION']
+
+if connection != 'winrm'
+#
+# OS type: UN*X
+#
+  set :backend, :"#{ENV['BACKEND']}"
+
+  if ENV['ASK_SUDO_PASSWORD']
+    begin
+      require 'highline/import'
+    rescue LoadError
+      fail "highline is not available. Try installing it."
+    end
+    set :sudo_password, ask("Enter sudo password: ") { |q| q.echo = false }
+  else
+    set :sudo_password, ENV['SUDO_PASSWORD']
+  end
+
+  unless ssh_config_file
+    options = Net::SSH::Config.for(host)
+  else
+    options = Net::SSH::Config.for(host,files=[ssh_config_file])
+  end
+
+  options[:user] ||= ENV['TARGET_USER']
+  options[:port] ||= ENV['TARGET_PORT']
+  options[:keys] ||= ENV['TARGET_PRIVATE_KEY']
+
+  set :host,        options[:host_name] || host
+  set :ssh_options, options
+
+  # Disable sudo
+  # set :disable_sudo, true
+
+  # Set environment variables
+  # set :env, :LANG => 'C', :LC_MESSAGES => 'C'
+
+  # Set PATH
+  # set :path, '/sbin:/usr/local/sbin:$PATH'
+else
+#
+# OS type: Windows
+#
+  set :backend, :winrm
+  set :os, :family => 'windows'
+
+  user = ENV['TARGET_USER']
+  port = ENV['TARGET_PORT']
+  pass = ENV['TARGET_PASSWORD']
+
+  if user.nil?
+    begin
+      require 'highline/import'
+    rescue LoadError
+      fail "highline is not available. Try installing it."
+    end
+    user = ask("\nEnter #{host}'s login user: ") { |q| q.echo = true }
+  end
+  if pass.nil?
+    begin
+      require 'highline/import'
+    rescue LoadError
+      fail "highline is not available. Try installing it."
+    end
+    pass = ask("\nEnter #{user}@#{host}'s login password: ") { |q| q.echo = false }
+  end
+
+  endpoint = "http://#{host}:#{port}/wsman"
+
+  winrm = ::WinRM::WinRMWebService.new(endpoint, :ssl, :user => user, :pass => pass, :basic_auth_only => true)
+  winrm.set_timeout 300 # 5 minutes max timeout for any operation
+  Specinfra.configuration.winrm = winrm
+
+end