teuton 2.3.5 → 2.3.7

data/lib/teuton/case_manager/case/dsl/expect.rb

@@ -1,50 +1,16 @@
 # frozen_string_literal: true
 
-require 'rainbow'
+require "colorize"
 
-# DSL module:
-# * expect_none
-# * expect_one
-# * expect_any
-# * expect
-# * expect2
-# * weight
 module DSL
-  def expect_none(input)
-    if input.class == Array
-      input.each { |i| result.find(i) }
-    else
-      result.find(input)
-    end
-    expect2 result.count.eq(0)
-  end
-
-  def expect_one(input)
-    if input.class == Array
-      input.each { |i| result.find(i) }
-    else
-      result.find(input)
-    end
-    expect2 result.count.eq(1)
-  end
-
-  def expect_any(input)
-    if input.class == Array
-      input.each { |i| result.find(i) }
-    else
-      result.find(input)
-    end
-    expect2 result.count.gt(0)
-  end
-
   # expect <condition>, :weight => <value>
   def expect(input, args = {})
-    if input.class == TrueClass || input.class == FalseClass
+    if input.instance_of?(TrueClass) || input.instance_of?(FalseClass)
       expect2(input, args)
-    elsif input.class == String || input.class == Regexp || input.class == Array
+    elsif input.instance_of?(String) || input.instance_of?(Regexp) || input.instance_of?(Array)
       expect_any input
     else
-      puts "[ERROR] expect #{input} (#{input.class})"
+      puts "[TypeError] expect #{input} (#{input.class})"
     end
   end
 
@@ -64,7 +30,48 @@ module DSL
     app = Application.instance
     c = app.letter[:bad]
     c = app.letter[:good] if cond
-    verbose Rainbow(c).yellow.bright
+    verbose c.colorize(:green)
+  end
+
+  def expect_any(input, args = {})
+    if input.instance_of? Array
+      input.each { |i| result.find(i) }
+    else
+      result.find(input)
+    end
+    expect2 result.count.gt(0), args
+  end
+
+  def expect_first(input, args = {})
+    @result.first
+    output = input
+    output = args[:expected] if args[:expected]
+    expect2 input, expected: output
+  end
+
+  def expect_last(input, args = {})
+    @result.last
+    output = input
+    output = args[:expected] if args[:expected]
+    expect2 input, expected: output
+  end
+
+  def expect_none(input, args = {})
+    if input.instance_of? Array
+      input.each { |i| result.find(i) }
+    else
+      result.find(input)
+    end
+    expect2 result.count.eq(0), args
+  end
+
+  def expect_one(input, args = {})
+    if input.instance_of? Array
+      input.each { |i| result.find(i) }
+    else
+      result.find(input)
+    end
+    expect2 result.count.eq(1), args
   end
 
   # Set weight value for the action

data/lib/teuton/case_manager/case/dsl/goto.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require_relative '../runner'
+require_relative "../runner"
 
 # Case class -> DSL module:
 # * goto
@@ -25,7 +25,7 @@ module DSL
     @action[:command] = args[:execute].to_s if args[:execute]
     @action[:command] = args[:exec].to_s if args[:exec]
     tempfile(args[:tempfile]) if args[:tempfile]
-    @action[:encoding] = args[:encoding] || 'UTF-8'
+    @action[:encoding] = args[:encoding] || "UTF-8"
 
     start_time = Time.now
     run_cmd_on(host)

data/lib/teuton/case_manager/case/dsl/log.rb

@@ -7,13 +7,12 @@ module DSL
   # Record log message
   # @param text (String)
   # @param type (Symbol) Values :info, :warn or :error
-  def log(text = '', type = :info)
-    s = ''
-    s = Rainbow('WARN!').color(:yellow) if type == :warn
-    s = Rainbow('ERROR').bg(:red) if type == :error
+  def log(text = "", type = :info)
+    s = ""
+    s = Rainbow("WARN!").color(:yellow) if type == :warn
+    s = Rainbow("ERROR").bg(:red) if type == :error
     t = Time.now
-    f = format('%<hour>02d:%<min>02d:%<sec>02d',
-               { hour: t.hour, min: t.min, sec: t.sec })
+    f = format("%<hour>02d:%<min>02d:%<sec>02d", {hour: t.hour, min: t.min, sec: t.sec})
     @report.lines << "[#{f}] #{s}: #{text}"
   end
   alias msg log

data/lib/teuton/case_manager/case/dsl/macro.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
-require_relative '../../../application'
-require 'os'
+
+require_relative "../../../application"
 
 # DSL module methods: assert, missing_method
 module DSL
@@ -17,10 +17,10 @@ module DSL
     input.each_pair { |k, v| set(k, v) }
     errors = []
     macros[name][:args].each do |i|
-      errors << i if get(i) == 'NODATA'
+      errors << i if get(i) == "NODATA"
     end
     if errors.count > 0
-      log("Macro #{name} => required params #{errors.join(',')}",:error)
+      log("Macro #{name} => required params #{errors.join(",")}", :error)
     else
       instance_eval(&macros[name][:block])
     end
@@ -32,10 +32,14 @@ module DSL
   # * Invoke macro (assert)
   def method_missing(method, args = {})
     a = method.to_s
-    if a.start_with?('_') && a.end_with?('_')
-      return instance_eval("get(:#{a[1, a.size - 2]})")
+    if a.start_with?("_") && a.end_with?("_")
+      return instance_eval("get(:#{a[1, a.size - 2]})", __FILE__, __LINE__)
     end
-    return macro a[6, a.size], args if a[0,6]=='macro_'
+    return macro a[6, a.size], args if a[0, 6] == "macro_"
     macro a, args
   end
+
+  def respond_to_missing?(method, *)
+    true
+  end
 end

data/lib/teuton/case_manager/case/dsl/main.rb

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
-require_relative 'expect'
-require_relative 'getset'
-require_relative 'goto'
-require_relative 'log'
-require_relative 'macro'
-require_relative 'send'
-require_relative 'target'
-require_relative 'unique'
+require_relative "expect"
+require_relative "getset"
+require_relative "goto"
+require_relative "log"
+require_relative "macro"
+require_relative "send"
+require_relative "target"
+require_relative "unique"

data/lib/teuton/case_manager/case/dsl/send.rb

@@ -15,21 +15,21 @@ module DSL
     host = args[:copy_to].to_s
     return unless @conn_status[host].nil?
 
-    ip = get((host + '_ip').to_sym)
-    username = get((host + '_username').to_sym).to_s
-    password = get((host + '_password').to_sym).to_s
-    port = get((host + '_port').to_sym).to_i
+    ip = get((host + "_ip").to_sym)
+    username = get((host + "_username").to_sym).to_s
+    password = get((host + "_password").to_sym).to_s
+    port = get((host + "_port").to_sym).to_i
     port = 22 if port.zero?
 
-    filename = @report.filename + '.' + @report.format.to_s
+    filename = "#{@report.filename}.#{@report.format}"
+    filename = "#{@report.filename}.txt" if @report.format == :colored_text
     localfilepath = File.join(@report.output_dir, filename)
     filename = args[:prefix].to_s + filename if args[:prefix]
 
-    if args[:remote_dir]
-      remotefilepath = File.join(args[:remote_dir], filename)
+    remotefilepath = if args[:remote_dir]
+      File.join(args[:remote_dir], filename)
     else
-      # remotefilepath = File.join(remote_tempdir, filename)
-      remotefilepath = File.join('.', filename)
+      File.join(".", filename)
     end
 
     # Upload a file or directory to the remote host
@@ -37,9 +37,9 @@ module DSL
       Net::SFTP.start(ip, username, password: password, port: port) do |sftp|
         sftp.upload!(localfilepath, remotefilepath)
       end
-      verboseln("=> [ OK  ] #{(get(:tt_members)[0,15]).ljust(16)} : #{remotefilepath}")
+      verboseln("=> [ OK  ] #{(get(:tt_members)[0, 15]).ljust(16)} : #{remotefilepath}")
     rescue
-      verboseln("=> [ERROR] #{(get(:tt_members)[0,15]).ljust(16)} : scp #{localfilepath} => #{remotefilepath}")
+      verboseln("=> [ERROR] #{(get(:tt_members)[0, 15]).ljust(16)} : scp #{localfilepath} => #{remotefilepath}")
     end
   end
 
@@ -47,7 +47,7 @@ module DSL
     return @action[:tempfile] if input.nil?
 
     name = input
-    name = 'teuton.tmp' if input == :default
+    name = "teuton.tmp" if input == :default
 
     @action[:tempfile] = File.join(@tmpdir, name)
     @action[:remote_tempfile] = File.join(@remote_tmpdir, name)
@@ -56,12 +56,10 @@ module DSL
   end
 
   def tempdir
-    # puts '[WARN] Using DSL.tempdir'
     @tmpdir
   end
 
   def remote_tempfile
-    # puts '[WARN] Using DSL.tempfile'
     @action[:remote_tempfile]
   end
 

data/lib/teuton/case_manager/case/dsl/unique.rb

@@ -1,11 +1,9 @@
 # frozen_string_literal: true
 
-# Case#DSL#unique
 module DSL
   def unique(key, value)
     return if value.nil?
 
-    k = (key.to_s + '=' + value.to_s).to_sym
-    @uniques << k
+    @uniques << "#{key}=#{value}".to_sym
   end
 end

data/lib/teuton/case_manager/case/main.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require_relative 'dsl/main'
-require_relative 'config'
-require_relative 'close'
-require_relative 'play'
-require_relative 'runner'
+require_relative "dsl/main"
+require_relative "config"
+require_relative "close"
+require_relative "play"
+require_relative "runner"

data/lib/teuton/case_manager/case/play.rb

@@ -28,7 +28,7 @@ class Case
 
   def close_opened_sessions
     @sessions.each_value do |s|
-      s.close if s.class == Net::SSH::Connection::Session
+      s.close if s.instance_of? Net::SSH::Connection::Session
     end
   end
 

data/lib/teuton/case_manager/case/result/ext_array.rb

@@ -3,12 +3,12 @@
 # This is an extension of Result class
 class Result
   def count
-    @alterations << 'count'
-    if @content.class == Array
+    @alterations << "count"
+    if @content.instance_of? Array
       @content = [@content.count]
       self
     elsif @content.nil?
-      @content = ['0']
+      @content = ["0"]
     else
       @content = [@content.to_i.to_s]
     end
@@ -31,13 +31,13 @@ class Result
   end
 
   def empty
-    @expected = 'Empty!'
+    @expected = "Empty!"
     @content.empty
   end
 
-  alias count!     count
-  alias length     count
-  alias len        count
-  alias size       count
-  alias empty?     empty
+  alias count! count
+  alias length count
+  alias len count
+  alias size count
+  alias empty? empty
 end

data/lib/teuton/case_manager/case/result/ext_compare.rb

@@ -1,104 +1,93 @@
 # frozen_string_literal: true
 
 # This is an extension of Result class
-# rubocop:disable Metrics/ClassLength
 class Result
-  # rubocop:disable Metrics/MethodLength
-  # Return true when content is equal than input
-  # @param input (Object)
   def eq(input)
+    # Return true when content is equal than input
     @expected = input
 
     case input.class.to_s
-    when 'Fixnum'
+    when "Fixnum"
       value = @content[0].to_i
-      puts '[WARN] Fixnum class is deprecated!'
-      puts '       Upgrade your Ruby version.'
-    when 'Float'
+      puts "[WARN] Fixnum is a Ruby deprecated class!"
+      puts "       (Upgrade your Ruby version)"
+    when "Float"
       value = @content[0].to_f
-    when 'Integer'
+    when "Integer"
       value = @content[0].to_i
-    when 'String'
+    when "String"
       value = @content[0].to_s
     else
       value = @content[0]
     end
     value == input
   end
-  # rubocop:enable Metrics/MethodLength
-  alias eq?        eq
-  alias equal      eq
-  alias equal?     eq
-  alias is_equal?  eq
+  alias eq? eq
+  alias equal eq
+  alias equal? eq
+  alias is_equal? eq
 
-  # rubocop:disable Metrics/MethodLength
   def neq(external)
     @expected = "Not equal to #{external}"
 
     case external.class.to_s
-    when 'Fixnum'
+    when "Fixnum"
       internal = @content[0].to_i
-      puts '[WARN] Fixnum class is deprecated!'
-      puts '       Upgrade your Ruby version.'
-    when 'Float'
+      puts "[WARN] Fixnum class is deprecated!"
+      puts "       Upgrade your Ruby version."
+    when "Float"
       internal = @content[0].to_f
-    when 'Integer'
+    when "Integer"
       internal = @content[0].to_i
     else
       internal = @content[0]
     end
     internal != external
   end
-  # rubocop:enable Metrics/MethodLength
-  alias neq?       neq
-  alias not_equal  neq
+  alias neq? neq
+  alias not_equal neq
   alias not_equal? neq
 
-  # rubocop:disable Metrics/MethodLength
   def ge(input)
     @expected = "Greater or equal to #{input}"
     return false if @content.nil? || @content[0].nil?
 
     value = @content[0]
     case input.class.to_s
-    when 'Fixnum'
+    when "Fixnum"
       value = @content[0].to_i
-      puts '[WARN] Fixnum class is deprecated!'
-      puts '       Upgrade your Ruby version.'
-    when 'Float'
+      puts "[WARN] Fixnum class is deprecated!"
+      puts "       Upgrade your Ruby version."
+    when "Float"
       value = @content[0].to_f
-    when 'Integer'
+    when "Integer"
       value = @content[0].to_i
     end
     value >= input
   end
-  # rubocop:enable Metrics/MethodLength
-  alias greater_or_equal  ge
+  alias greater_or_equal ge
   alias greater_or_equal? ge
 
-  # rubocop:disable Metrics/MethodLength
   def gt(input)
     @expected = "Greater than #{input}"
     return false if @content.nil? || @content[0].nil?
 
     value = @content[0]
     case input.class.to_s
-    when 'Fixnum'
+    when "Fixnum"
       value = @content[0].to_i
-      puts '[WARN] Fixnum class is deprecated!'
-      puts '       Upgrade your Ruby version.'
-    when 'Float'
+      puts "[WARN] Fixnum class is deprecated!"
+      puts "       Upgrade your Ruby version."
+    when "Float"
       value = @content[0].to_f
-    when 'Integer'
+    when "Integer"
       value = @content[0].to_i
     end
     value > input
   end
-  # rubocop:enable Metrics/MethodLength
-  alias greater          gt
-  alias greater_than     gt
+  alias greater gt
+  alias greater_than gt
 
-  # rubocop:disable Metrics/MethodLength
   def le(input)
     @expected = "Lesser or equal to #{input}"
 
@@ -106,41 +95,38 @@ class Result
 
     value = @content[0]
     case input.class.to_s
-    when 'Fixnum'
+    when "Fixnum"
       value = @content[0].to_i
-      puts '[WARN] Fixnum class is deprecated!'
-      puts '       Upgrade your Ruby version.'
-    when 'Float'
+      puts "[WARN] Fixnum class is deprecated!"
+      puts "       Upgrade your Ruby version."
+    when "Float"
       value = @content[0].to_f
-    when 'Integer'
+    when "Integer"
       value = @content[0].to_i
     end
     value <= input
   end
-  # rubocop:enable Metrics/MethodLength
-  alias lesser_or_equal  le
+  alias lesser_or_equal le
   alias lesser_or_equal? le
 
-  # rubocop:disable Metrics/MethodLength
   def lt(input)
     @expected = "Lesser than #{input}"
     return false if @content.nil? || @content[0].nil?
 
     value = @content[0]
     case input.class.to_s
-    when 'Fixnum'
+    when "Fixnum"
       value = @content[0].to_i
-      puts '[WARN] Fixnum class is deprecated!'
-      puts '       Upgrade your Ruby version.'
-    when 'Float'
+      puts "[WARN] Fixnum class is deprecated!"
+      puts "       Upgrade your Ruby version."
+    when "Float"
       value = @content[0].to_f
-    when 'Integer'
+    when "Integer"
       value = @content[0].to_i
     end
     value < input
   end
-  # rubocop:enable Metrics/MethodLength
-  alias lesser  lt
+  alias lesser lt
   alias smaller lt
   alias lesser_than lt
 
@@ -151,7 +137,7 @@ class Result
     return false if @content.nil?
 
     target = @content[0].to_f
-    desv   = (target * 10.0) / 100.0
+    desv = (target * 10.0) / 100.0
     return true if (target - input.to_f).abs.to_f <= desv
 
     false

data/lib/teuton/case_manager/case/result/ext_filter.rb

@@ -6,18 +6,24 @@ class Result
   def find(filter)
     @alterations << "find(#{filter})"
     case filter.class.to_s
-    when 'Array'
+    when "Array"
       find_when_array(filter)
-    when 'String' || 'Integer'
+    when "String" || "Integer"
       @content.select! { |i| i.include?(filter.to_s) }
-    when 'Regexp'
+    when "Regexp"
       @content.select! { |i| filter.match(i) }
     end
     self
   end
-  alias grep   find
-  alias grep!  find
-  alias find!  find
+  alias grep find
+  alias grep! find
+  alias find! find
+
+  def first
+    @alterations << "first"
+    @content = [@content.first]
+    self
+  end
 
   def not_find(p_filter)
     @alterations << "not_find(#{p_filter})"
@@ -32,7 +38,7 @@ class Result
     @alterations << "since(#{filter})"
     return self if @content.size.zero?
 
-    if filter.class == String
+    if filter.instance_of? String
       flag = false
       @content.select! do |i|
         flag = true if i.include?(filter.to_s)
@@ -42,11 +48,17 @@ class Result
     self
   end
 
+  def last
+    @alterations << "last"
+    @content = [@content.last]
+    self
+  end
+
   def until(filter)
     @alterations << "until(#{filter})"
     return self if @content.size.zero?
 
-    if filter.class == String
+    if filter.instance_of? String
       flag = true
       @content.select! do |i|
         flag = false if i.include?(filter.to_s)

data/lib/teuton/case_manager/case/result/result.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require_relative 'ext_array'
-require_relative 'ext_compare'
-require_relative 'ext_filter'
+require_relative "ext_array"
+require_relative "ext_compare"
+require_relative "ext_filter"
 
 # This object contains data returned by remote/local execution
 # * initialize
@@ -22,7 +22,7 @@ class Result
   end
 
   def alterations
-    @alterations.join(' & ')
+    @alterations.join(" & ")
   end
 
   def content=(content)
@@ -30,30 +30,31 @@ class Result
     @content = content.clone
   end
 
-  def reset
-    @content_backup = []
-    @content        = []
-    @exitstatus     = nil
-    @value          = nil
-    @expected       = nil
-    @alterations    = []
-  end
-
   def debug
-    print "\n" + '*' * 20
+    print "\n" + "*" * 20
     print " [DEBUG] count=#{@content.count} "
-    puts '*' * 20
+    puts "*" * 20
     @content.each_with_index do |item, index|
-      puts format('%<index>2d: %<item>s', { index: index, item: item })
+      puts format("%<index>2d: %<item>s", {index: index, item: item})
     end
-    puts '*' * 57
+    puts "*" * 57
   end
 
   def expected
     @expected.to_s
   end
 
+  def reset
+    @content_backup = []
+    @content = []
+    @exitstatus = nil
+    @value = nil
+    @expected = nil
+    @alterations = []
+  end
+
   def ok?
+    # REVISE THIS
     return false if @exitstatus.nil?
 
     @exitstatus.zero?
@@ -63,7 +64,7 @@ class Result
     temp = @content_backup.clone
     reset
     @content_backup = temp
-    @content        = temp.clone
+    @content = temp.clone
   end
   alias restore! restore