driver.rb

#!/usr/bin/env ruby

# 
# == Synopsis
#
# driver: used to generate tsung XML from api test branch and initiate load
#
# == Usage
#
# driver.rb [OPTIONS] ... SUITE_FILE_NAME PRIMARY_SERVER_CONTEXT SECONDARY_SERVER_CONTEXT
#
# -h, --help:
#     show help
#
# -d, --debug:
#     enable debug logging
#
# -t, --thinktime
#     enable thinktime sleeps (simulates user input delays)
#
# -v, --verbose:
#     will dump http traffic from tsung
#
# -x, --execute:
#     start the load after generating the XML. If not specified, only XML will be generated.
#
# -o, --output:
#     filename(path) that you want to use for output XML
#
# -s, --suite [suite file]:
#     suite file for performance run
#
# == Examples
#
#

require 'rubygems'
require 'optparse'
#require 'rdoc/usage'
require File.dirname(__FILE__) + '/lib/common.rb'
require File.dirname(__FILE__) + '/lib/config.rb'
require 'drb'
include Common

puts "patchlevel: #{RUBY_PATCHLEVEL}, release_date: #{RUBY_RELEASE_DATE}, ruby_version: #{RUBY_VERSION}, ruby_platform: #{RUBY_PLATFORM}"


########

errors = [] # Gather arg validation errors

# Initialize true/false vars
config = AutoConfig.new
option = false

# Default config values
config.tsung_log_level = 'notice'
config.execute = nil


# Get cmd line options
optparse = OptionParser.new do |opts|
    
  # Banner
  opts.banner = "Usage: driver.rb [OPTIONS]"
  
  # Definition of options
  opts.on('-h', '--help', 'Display help screen') do
    puts opts
    exit
  end
  
  # Config file
  opts.on('-c', '--config FILE', 'path to xml config file for everything after <tsung> and before <sessions> : NOT CURRENTLY SUPPORTED') do |file|
    config.intro_xml = file
    errors.push("#{opt} does not exist") unless(File.file?(config.intro_xml))
    option = true
  end
  
  # Product
  opts.on('-p', '--product PRODUCT', "supported products: #{config.products}" ) do |product|
    config.product = product.downcase
  end
  
  # Suite
  opts.on('-s', '--suite SUITE', 'suite file') do |suite|
    config.suite = suite
  end
  
  # Output xml file
  opts.on('-o', '--output FILE', 'path to xml output') do |file|
    config.output = file
  end
  
  # App contexts
  opts.on('--contexts PRIMARY_CONTEXT,SECONDARY_CONTEXT', Array, 'contexts for both primary and secondary servers') do |context|
    (config.context, config.secondary_context) = context
  end
  
  # Log file
  opts.on('-l', '--log FILE', 'path to log output from this framework') do |file|
    config.log_path = file
  end
  
  # SSL
  opts.on('--ssl', 'set ssl to true or false to set default ssl behavior') do |ssl|
    config.ssl = ssl
  end
  
  # Execute tests after generating xml
  opts.on('-x', '--execute', 'start the load after generating the XML') do
    config.execute = true
  end
  
  # Verbose option for tsung
  opts.on('-v', '--verbose', 'dumps http traffic from tsung') do
    config.verbose = true
  end
  
  # Log level for Tsung
  opts.on('-t', '--tsung_log_level LEVEL', 'sets the log level for tsung. Available levels: emergency, critical, error, warning, notice (default), info, debug') do |level|
    if(level != 'emergency' and level != 'critical' and level != 'error' and level != 'warning' and level != 'notice' and level != 'info' and level != 'debug')
      puts "Invalid setting for -t"
      puts opts
      exit
    else
      config.tsung_log_level = level
    end
  end

  # Enable debug
  opts.on('-d', '--debug', 'enable debug logging') do
    config.debug = true
  end
  
  # Enable thinktime
  opts.on('-t', '--thinktime', 'enable thinktime sleeps') do
    config.thinktime = true
  end
  
end

optparse.parse!

# Need to collect drb port
config.drb_port = ENV['DRB_PORT']
errors << "Must specify DRB_PORT in shell environment" unless(!config.drb_port.nil?)

# Exit if we have problems
if(!errors.empty?)
  errors.each { |err| puts "#{err}" }
  exit
end


# If no options are passed then we'll use interactive setup
config.initialize_logs
config.setup

# write initial config XML here before we get to tests
config.initialize_output_xml


config.log.debug_msg("----------CONFIG----------")
config.log.debug_msg(config.inspect)
config.log.debug_msg("--------------------------")

#
# EXECUTE TESTS
#

# Fire up service for tests to read config information from
DRb.start_service "druby://localhost:#{config.drb_port}", config
config.log.debug_msg("DRB service running at #{DRb.uri}")

begin
  
  # Write <sessions> container
  config.init_sessions
  
  config.tests.each_key do |test|
    puts `#{config.test_dir}/#{test}`
  end
  
end

config.xml_obj.write($stdout, 2) if(config.debug) # Dump out xml
config.xml_obj.write(config.xml_writer.file, 2) # Write xml to file
config.xml_writer.file.close # Need to close so Tsung can open the file


if(config.execute)
  # Launch tsung
  config.log.info_msg("Starting tsung with command [tsung -f #{config.xml_writer.file_path} start]")
  puts `tsung -f #{config.xml_writer.file_path} start`
end

DRb.stop_service