visitbench 0.2.0

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,22 @@
+## MAC OS
+.DS_Store
+*.swo
+*.swp
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+
+## PROJECT::SPECIFIC

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Jaap van der Meer
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,18 @@
+= visitbench
+
+Visitbench stresstests your server by simulating real user sessions.
+
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2010 Jaap van der Meer. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,49 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "visitbench"
+    gem.summary = %Q{Visitbench stress tests your server by manipulating real user sessions}
+    gem.description = %Q{}
+    gem.email = "jaapvandermeer@gmail.com"
+    gem.homepage = "http://github.com/japetheape/visitbench"
+    gem.authors = ["Jaap van der Meer"]
+    gem.add_development_dependency "rspec", ">= 1.2.9" 
+    gem.add_dependency "mechanize"
+    gem.bindir = 'bin'
+    gem.executables = ["visitbench"]
+     
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end
+
+require 'spec/rake/spectask'
+Spec::Rake::SpecTask.new(:spec) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.spec_files = FileList['spec/**/*_spec.rb']
+end
+
+Spec::Rake::SpecTask.new(:rcov) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.rcov = true
+end
+
+task :spec => :check_dependencies
+
+task :default => :spec
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "visitbench #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.2.0

data/bin/visitbench

@@ -0,0 +1,25 @@
+#!/usr/bin/env ruby
+require File.join(File.dirname(__FILE__), '..', 'lib', 'visit_bench')
+
+if ARGV.empty?
+  files_to_run = Dir["visitbench_tests/*.rb"]
+else 
+  files_to_run = ARGV
+end
+
+
+if files_to_run.empty? 
+  puts "No tests to run."
+  puts "Invoke visitbench with files to run or place them in ./visitbench_tests."
+end
+
+
+files_to_run.each { |file| 
+  100.times { print "-" }
+  puts ""
+  puts "Running: " + file
+  100.times { print "-" }
+  puts ""
+  require file 
+}
+

data/examples/request_example.rb

@@ -0,0 +1,25 @@
+require File.join(File.dirname(__FILE__), '../lib/visitbench.rb')
+
+host = "http://example.com"
+
+visit_bench = VisitBench.new(:host => host) do |vb|
+  vb.add_session do |session|
+    session.add_request '/', :time_on_page => 3
+    session.add_request '/somepage', :time_on_page => 2 
+    session.add_request '/otherpage'
+    session.add_request '/otherpage', :method => :post, :params => 100
+  end
+end
+
+benchmark = performance_engine.run(:amt_times => 2, :concurrent_users => 3)
+puts "Done benchmarking."
+puts ""
+puts ""
+100.times { print "-"}
+puts ""
+puts "Performance Report"
+100.times { print "-"}
+puts ""
+
+puts Report.new(benchmark)
+

data/lib/enumeration.rb

@@ -0,0 +1,27 @@
+#  Add methods to Enumerable, which makes them available to Array
+module Enumerable
+ 
+  #  sum of an array of numbers
+  def sum
+    return self.inject(0){|acc,i|acc +i}
+  end
+ 
+  #  average of an array of numbers
+  def average
+    return self.sum/self.length.to_f
+  end
+ 
+  #  variance of an array of numbers
+  def sample_variance
+    avg=self.average
+    sum=self.inject(0){|acc,i|acc +(i-avg)**2}
+    return(1/self.length.to_f*sum)
+  end
+ 
+  #  standard deviation of an array of numbers
+  def standard_deviation
+    return Math.sqrt(self.sample_variance)
+  end
+ 
+end
+

data/lib/report.rb

@@ -0,0 +1,56 @@
+require File.join(File.dirname(__FILE__), 'report.rb')
+
+require File.join(File.dirname(__FILE__), 'enumeration.rb')
+
+class Report
+  def initialize(run_times)
+    @run_times = run_times
+  end
+  
+  def to_s
+    out = []
+    out << "Run times per user: "
+    out << "\t - values: " + run_times_per_user.inspect
+    out << "\t - averages per user: " + run_times_per_user.map(&:average).inspect
+    out << "\t - total average: " + run_times_per_user.flatten.average.to_s
+    out << ""
+    out << "Run times per request: "
+    out << "\t - values: "
+    run_times_per_request.each do |k,v|
+      out << "\t\t -" + k + " \t\t" + v.inspect
+    end
+    out << "\t - averages per request:"
+    
+    run_times_per_request.each do |k,v|
+      out << "\t\t-" + k
+      out << "\t\t avg: " + v.average.to_s + " \tstd:" + v.standard_deviation.to_s
+      out << ""
+    end
+    out.join("\n")
+  end
+
+
+  def run_times_per_user
+    all = []
+    @run_times.each do |session|
+      all << session.map { |x| x[:time] }
+    end
+    all
+  end
+
+  def run_times_per_request
+    request_times = {}
+    @run_times.flatten.each do |rt|
+      key = rt[:request].url + "(" + rt[:request].method.to_s + ")"
+      request_times[key] ||= []
+      request_times[key] << rt[:time]
+    end
+    request_times
+  end
+  
+
+
+
+end
+
+

data/lib/request.rb

@@ -0,0 +1,51 @@
+require 'net/http'
+require 'uri'
+require 'mechanize'
+class WWW::Mechanize
+def set_headers(uri, request, cur_page)
+super
+raise "JO"
+request.add_field('Foo', 'bar')
+end
+end
+
+
+class Request
+  attr_reader :time_on_page, :url, :method
+
+
+  def initialize(url, options = {})
+    @time_on_page = options[:time_on_page] || 4
+    @url = (options[:host] || "") + url
+    @method = options[:method] || :get
+    @params = options[:params] || {}
+    @browser = options[:browser] #|| Mechanize.new
+  end
+
+
+  def do_request!
+    uri = URI.parse(@url) 
+    if @method == :post
+      Net::HTTP.post_form uri, @params
+    else
+      #@browser.get(:url => @url, :headers => {'Authorization' => 'Basic ZGV2X3RlYW06RW5lcmd5Mi4w'})
+      @browser.get(@url)
+      #http = Net::HTTP.new("testing.et-model.com")
+      
+      
+      #@req = req = Net::HTTP::Get.new(
+      #Net::HTTP.get_print uri
+      #req = Net::HTTP::Get.new(uri)
+      #req.basic_auth "dev_team", "Energy2.0"
+      #response = http.request(req)
+      #puts response.body
+
+    end
+
+  end
+
+  
+
+end
+
+

data/lib/session.rb

@@ -0,0 +1,30 @@
+# Simulation of a user that browses a site
+class Session
+  
+  # requests is an array of requests that will be run
+  def initialize(options = {})
+    @requests = []
+    @options = options
+  end
+
+  def add_request(*args)
+    args[1] ||= {}
+    args[1].merge!(@options[:request_options] || {})
+    @requests << Request.new(*args)
+  end
+
+  def run
+    puts "Started session."
+    run_times = []
+    @requests.each do |r|
+      puts "Doing request to %s with a time on page of %0.02f..." % [r.url, r.time_on_page]
+      run_times << { :request => r, :time => VisitBench.benchmark { r.do_request! } }
+      sleep r.time_on_page
+    end
+    return run_times
+  end
+
+
+
+    
+end

data/lib/visit_bench.rb

@@ -0,0 +1,61 @@
+require File.join(File.dirname(__FILE__), 'request.rb')
+require File.join(File.dirname(__FILE__), 'session.rb')
+require File.join(File.dirname(__FILE__), 'report.rb')
+require "rubygems"
+require "mechanize"
+require "logger"
+
+# Visitbench stresstests your server
+# by simulation real user sessions. 
+class VisitBench 
+  def initialize(options = {})
+    @host = options[:host] || 'localhost'
+    @browser = Mechanize.new 
+    @browser.log = Logger.new(STDOUT) if options[:debug]
+    options[:headers] ||= []
+    options[:headers].each do |k,v|
+      @browser.request_headers[k] = v
+    end
+    
+    @browser.basic_auth(options[:username], options[:password]) if options[:username]
+    @sessions = options[:sessions] || []
+    yield self
+  end
+
+  # add a session with preinitialized values
+  def add_session
+    s = Session.new(:request_options => {:host => @host, :browser => @browser})
+    yield s
+    @sessions << s
+  end
+
+  # run the sessions multiple times and run them in parallel
+  def run(options = {})
+    amt_times = options[:amt_times] || 10
+    concurrent_users = options[:concurrent_users] || 3
+    run_times = []
+    amt_times.times do |i|
+        trs = []
+        concurrent_users.times do
+          puts "Starting thread"
+          trs << Thread.new do      
+            session = @sessions[(@sessions.length - 1) * rand]
+            run_times << session.run
+          end
+        end
+        trs.each {|x| x.join }
+    end
+    run_times
+  end
+
+  
+  class << self
+    def benchmark(&block)
+      begin_time = Time.now
+      yield
+      return Time.now - begin_time 
+    end
+  end
+
+
+end

data/spec/spec.opts

@@ -0,0 +1 @@
+--color

data/spec/spec_helper.rb

@@ -0,0 +1,9 @@
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'visitbench'
+require 'spec'
+require 'spec/autorun'
+
+Spec::Runner.configure do |config|
+  
+end

data/spec/visitbench_spec.rb

@@ -0,0 +1,6 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "Visitbench" do
+  it "fails" do
+  end
+end

data/tasks/visit_bench.rake

@@ -0,0 +1,9 @@
+namespace :visitbench do
+  task :run do
+    
+  end
+end
+
+
+
+

data/visitbench.gemspec

@@ -0,0 +1,68 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{visitbench}
+  s.version = "0.2.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Jaap van der Meer"]
+  s.date = %q{2010-09-23}
+  s.default_executable = %q{visitbench}
+  s.description = %q{}
+  s.email = %q{jaapvandermeer@gmail.com}
+  s.executables = ["visitbench"]
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+     ".gitignore",
+     "LICENSE",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "bin/visitbench",
+     "examples/request_example.rb",
+     "lib/enumeration.rb",
+     "lib/report.rb",
+     "lib/request.rb",
+     "lib/session.rb",
+     "lib/visit_bench.rb",
+     "spec/spec.opts",
+     "spec/spec_helper.rb",
+     "spec/visitbench_spec.rb",
+     "tasks/visit_bench.rake",
+     "visitbench.gemspec"
+  ]
+  s.homepage = %q{http://github.com/japetheape/visitbench}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.7}
+  s.summary = %q{Visitbench stress tests your server by manipulating real user sessions}
+  s.test_files = [
+    "spec/spec_helper.rb",
+     "spec/visitbench_spec.rb",
+     "examples/request_example.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<rspec>, [">= 1.2.9"])
+      s.add_runtime_dependency(%q<mechanize>, [">= 0"])
+    else
+      s.add_dependency(%q<rspec>, [">= 1.2.9"])
+      s.add_dependency(%q<mechanize>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<rspec>, [">= 1.2.9"])
+    s.add_dependency(%q<mechanize>, [">= 0"])
+  end
+end
+

metadata

@@ -0,0 +1,116 @@
+--- !ruby/object:Gem::Specification 
+name: visitbench
+version: !ruby/object:Gem::Version 
+  hash: 23
+  prerelease: false
+  segments: 
+  - 0
+  - 2
+  - 0
+  version: 0.2.0
+platform: ruby
+authors: 
+- Jaap van der Meer
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-09-23 00:00:00 +02:00
+default_executable: visitbench
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 13
+        segments: 
+        - 1
+        - 2
+        - 9
+        version: 1.2.9
+  type: :development
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: mechanize
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id002
+description: ""
+email: jaapvandermeer@gmail.com
+executables: 
+- visitbench
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .document
+- .gitignore
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- bin/visitbench
+- examples/request_example.rb
+- lib/enumeration.rb
+- lib/report.rb
+- lib/request.rb
+- lib/session.rb
+- lib/visit_bench.rb
+- spec/spec.opts
+- spec/spec_helper.rb
+- spec/visitbench_spec.rb
+- tasks/visit_bench.rake
+- visitbench.gemspec
+has_rdoc: true
+homepage: http://github.com/japetheape/visitbench
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Visitbench stress tests your server by manipulating real user sessions
+test_files: 
+- spec/spec_helper.rb
+- spec/visitbench_spec.rb
+- examples/request_example.rb