hulk_smash 1.1.0 → 1.2.0

data/lib/hulk_smash/request.rb

@@ -0,0 +1,62 @@
+require_relative 'smasher'
+require_relative 'url_data_converter'
+
+module HulkSmash
+  class Request
+    attr_reader :url, :concurrent_users, :method, :url_data
+    attr_accessor :benchmark, :duration
+
+    def initialize(url, options={})
+      @url = url
+      @concurrent_users = options[:concurrent_users]
+      @duration = options[:duration]
+      @method = options[:method] || :get
+      @url_data = UrlDataConverter.from_hash(options[:data]) if options[:data]
+      self.benchmark = options[:benchmark]
+    end
+
+    def command
+      if post?
+        write_to_urls_file("#{url} POST #{url_data}")
+        "siege #{cmd_options} -f #{urls_file}"
+      elsif put?
+        write_to_urls_file("#{url} POST _method=put&#{url_data}")
+        "siege #{cmd_options} -f #{urls_file}"
+      else
+        "siege #{cmd_options} #{url}"
+      end
+    end
+
+    def cmd_options
+      ary = []
+      ary << '-b' if benchmark
+      ary << "-c#{concurrent_users}" if concurrent_users
+      ary << "-t#{duration}" if duration
+      ary.join(" ")
+    end
+
+    def post?
+      method == :post
+    end
+
+    def put?
+      method == :put
+    end
+
+    def urls_file
+      File.expand_path("urls_file.txt", Smasher.cache_dir)
+    end
+
+    private
+
+    def urls_file_base_path
+      File.expand_path("../../../cache", __FILE__)
+    end
+
+    def write_to_urls_file(content)
+      file = File.open(urls_file, "w+")
+      file.write(content)
+      file.close
+    end
+  end
+end

data/lib/hulk_smash/result.rb

@@ -2,11 +2,14 @@ require_relative 'validator'
 
 module HulkSmash
   class Result
-    attr_reader :avg_response_time, :requests_per_second, :validator
+    attr_reader :avg_response_time, :requests_per_second, :validator, :availability
 
     def initialize(siege_result)
       @siege_result = siege_result
       @validator = Validator.new siege_result
+      @avg_response_time = 'N/A'
+      @requests_per_second = 'N/A'
+      @availability = 'N/A'
       parse_result
     end
 
@@ -28,9 +31,15 @@ module HulkSmash
       if valid?
         parse_avg_response_time
         parse_requests_per_second
+        parse_availability
       end
     end
 
+    def parse_availability
+      regex = /Availability:\s*([\d\.]*) %/
+      @availability = regex.match(siege_result)[1] + " %"
+    end
+
     def parse_avg_response_time
       regex = /Response time:\s*([\d\.]*) secs/
       @avg_response_time = regex.match(siege_result)[1].to_f*1000

data/lib/hulk_smash/smasher.rb

@@ -1,34 +1,37 @@
 require_relative 'result'
-require 'stringio'
+require_relative 'request'
 
 module HulkSmash
   class Smasher
-    attr_reader :url, :duration, :concurrent_users, :result
+    attr_reader :request, :result
 
     def initialize(url='http://localhost', options={})
-      @duration = options[:duration] || self.class.default_duration
-      @concurrent_users = options[:concurrent_users] || self.class.default_concurrent_users
-      @url = url
+      options = default_options.merge(options)
+      options[:method] = options[:method].downcase.to_sym if options[:method]
+      @request = Request.new url, options
+      setup_cache_dir
     end
 
-    def run_load_test
-      @result = execute_siege_command "siege -b -t#{duration} -c#{concurrent_users} #{url}"
+    def setup_cache_dir
+      Dir.mkdir self.class.cache_dir unless Dir.exists?(self.class.cache_dir)
     end
 
-    def run_scalability_test
-      @result = execute_siege_command "siege -t#{duration} -c#{concurrent_users} #{url}"
+    def load_test=(val)
+      request.benchmark = val
     end
 
-    private
+    def run_load_test
+      self.load_test = true
+      run
+    end
 
-    def execute_siege_command(command)
-      `#{command} > #{results_file} 2>&1`
-      results = File.read(results_file)
-      HulkSmash::Result.new(results)
+    def run_scalability_test
+      run
     end
 
-    def results_file
-      @results_file ||= File.expand_path('../../../log/results.log', __FILE__)
+    def run
+      `#{request.command} > #{self.class.results_file} 2>&1`
+      @result = HulkSmash::Result.new(self.class.results_file_contents)
     end
 
     def self.default_duration
@@ -38,5 +41,23 @@ module HulkSmash
     def self.default_concurrent_users
       15
     end
+
+    def self.cache_dir
+      File.expand_path('../../../cache', __FILE__)
+    end
+
+    def self.results_file
+      File.expand_path('results.txt', cache_dir)
+    end
+
+    def self.results_file_contents
+      File.read(results_file) 
+    end
+
+    private
+
+    def default_options
+      { duration: '5s', concurrent_users: 15 }
+    end
   end
 end

data/lib/hulk_smash/url_data_converter.rb

@@ -0,0 +1,28 @@
+module HulkSmash
+  class UrlDataConverter
+    def self.from_hash(hash)
+      encode(hash)
+    end
+
+    def self.encode(value, key = nil, out_hash = {})
+      case value
+      when Hash  then
+        value.each { |k,v| encode(v, append_key(key,k), out_hash) }
+        out_hash
+      when Array then
+        value.each { |v| encode(v, "#{key}[]", out_hash) }
+        out_hash
+      when nil   then ''
+      else
+        out_hash[key] = value
+      end
+      out_hash.map{|k,v| "#{k}=#{v}" }.join("&")
+    end
+
+    private
+
+    def self.append_key(root_key, key)
+      root_key.nil? ? "#{key}" : "#{root_key}[#{key.to_s}]"
+    end
+  end
+end

data/lib/hulk_smash/validator.rb

@@ -9,6 +9,7 @@ module HulkSmash
 
     def valid?
       validate_version_is_supported
+      validate_successful_siege
       reasons_for_failure.empty?
     end
 
@@ -19,6 +20,14 @@ module HulkSmash
       @version ||= regex.match(siege_result)[1].to_f
     end
 
+    def unsuccessful_siege?
+      siege_result.include?('error') && !siege_result.include?('Shortest transaction')
+    end
+
+    def validate_successful_siege
+      @reasons_for_failure << "Unable to connect" if unsuccessful_siege?
+    end
+
     def validate_version_is_supported
       @reasons_for_failure << "Siege version must be 2.x" if version < 2 || version >= 3
     end

data/lib/hulk_smash/version.rb

@@ -5,7 +5,7 @@ module HulkSmash
     end
 
     def self.minor
-      1
+      2
     end
 
     def self.patch

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: hulk_smash
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 1.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-08-02 00:00:00.000000000 Z
+date: 2012-08-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -81,12 +81,13 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- lib/hulk_smash/version.rb
 - lib/hulk_smash.rb
+- lib/hulk_smash/version.rb
 - lib/hulk_smash/smasher.rb
 - lib/hulk_smash/result.rb
 - lib/hulk_smash/validator.rb
-- log/.gitkeep
+- lib/hulk_smash/request.rb
+- lib/hulk_smash/url_data_converter.rb
 homepage: http://asynchrony.com
 licenses: []
 post_install_message: