cafe-grader-web

Location: cafe-grader-web/test/load/visitor.rb - annotation

Show More
Commit Description:
added load testing base library
Commit Description:
added load testing base library
References:
r313:0dcb692d14e0 default
File last commit:
r313:0dcb692d14e0 default
Show/Diff file:
Action:
test/load/visitor.rb | 172 lines | 3.7 KiB | text/x-ruby | RubyLexer |
History | Source | Raw | Download
Jittat Fakcharoenphol
added load testing base library
 r313 require 'rubygems'
require 'curb'
class Visitor
attr_accessor :talkative
class << self
attr_accessor :commands
attr_accessor :base_url
attr_accessor :cookies_stored
end
def initialize(id=0)
# initialize nil class variable
puts self.class.base_url
self.class.base_url = "" if (self.class.base_url) == nil
self.class.cookies_stored = false if self.class.cookies_stored == nil
@id = id
@curl = Curl::Easy.new
@curl.enable_cookies = self.class.cookies_stored
@curl.cookiejar = "mycookies"
@statistics = Array.new
@talkative = false
end
def self.site_url(url)
self.base_url = url
end
def self.stores_cookies
self.cookies_stored = true
end
def self.preprocess_param_hash(params)
return {} if params==nil
plist = {}
params.each do |key,val|
if key.is_a? Symbol
key_s = key.to_s
else
key_s = key
end
plist[key_s] = val
end
plist
end
def self.get(url,params=nil)
self.commands = [] if self.commands==nil
self.commands << {
:command => :get,
:url => url,
:params => Visitor.preprocess_param_hash(params) }
end
def self.post(url,params=nil,options=nil)
self.commands = [] if self.commands==nil
self.commands << { :command => :post,
:url => url,
:params => Visitor.preprocess_param_hash(params),
:options => options }
end
def get(url,params)
#build url
full_url = "#{self.class.base_url}#{url}"
if params!=nil and params.length!=0
full_url += '?'
params.each do |key,val|
if full_url.slice(-1..-1)!='?'
full_url += '&'
end
full_url += @curl.escape(key) + '=' + @curl.escape(val)
end
end
@curl.url = full_url
@curl.http_get
end
def post(url,params,options)
@curl.url = "#{self.class.base_url}#{url}"
if options!=nil and options[:multipart]==true
@curl.multipart_form_post = true
end
#build post fields
fields = []
params.each do |key,val|
if val.is_a? Hash
case val[:type]
when :file
fields << Curl::PostField.file(key,val[:data])
end
else
fields << Curl::PostField.content(key,val.to_s)
end
end
@curl.http_post *fields
end
def run(option=nil)
if (option==nil) or (option==:once)
times = 1
elsif (option==:forever)
times = -1
else
times = option
end
while times!=0
self.class.commands.each do |cmd|
puts "#{@id}: #{cmd[:command]} #{cmd[:url]}" if @talkative
start_time = Time.new
case cmd[:command]
when :get
get cmd[:url], cmd[:params]
when :post
post cmd[:url], cmd[:params], cmd[:options]
end
finish_time = Time.new
@statistics << {
:url => "#{cmd[:command]}:#{cmd[:url]}",
:time => finish_time - start_time }
end
if times!=-1 # infinity times
times -= 1
end
end
end
def show_raw_stat
@statistics.each do |stat|
puts "#{stat[:url]} => #{stat[:time]}"
end
end
def statistics
num_requested = @statistics.length
totaltime = 0.0
@statistics.each { |stat| totaltime += stat[:time] }
if num_requested>0
average_request_time = totaltime / num_requested
else
average_request_time = 0
end
sq_sum = 0.0
@statistics.each do |stat|
sq_sum += (stat[:time]-average_request_time) ** 2
end
if num_requested>1
sd = Math.sqrt(sq_sum/(num_requested-1))
else
sd = 0
end
return {
:num_requested => num_requested,
:avg_request_time => average_request_time,
:std_dev => sd
}
end
end