rtsung 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: c6e00ba6e47dd13bfbc4ab175c9bb1c4b94d391e
+  data.tar.gz: 90783de29e11f74d50ec57e07447931de6d155e3
+SHA512:
+  metadata.gz: 4bd07c4d1c5051e331d0b8dd4681f8019cc8bd4e74f961dafe3a30fc40ed34dc1215f651121974e04fcea8418cc203e06ea1f21d99c770316cab5e7a20f8d7bf
+  data.tar.gz: 1d4fc1ae3a6db35d91e26b99c87dff0e7fa3809e4a723a386b6c1cd5612cd369e54ae668c039eb9f6d66f7f4a5a72c43b0ac05bd815a7b7f3e97b084f72e5ca6

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in rtsung.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Maxim Pechnikov
+
+MIT License
+
+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

@@ -0,0 +1,36 @@
+require 'rtsung'
+
+rtsung = RTsung.new do
+  client 'localhost', :vm => true
+
+  client 'router', :max_users => 32
+
+  client 'db1', :cpu => 2, :ip => '192.168.2.1', :weight => 1
+  client 'db2', :cpu => 2, :ip => ['192.168.2.3', '192.168.2.4'], :weight => 2
+  
+  server 'example.com'
+  server 'www.example.com', :port => 8080
+
+  phase
+  phase 2, :rate => 30
+  phase 3, 30, :second, { :interval => 2 }
+
+  user_agents do
+    name 'Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.7.8) Gecko/20050513 Galeon/1.3.21', :probability => 30
+    name 'Mozilla/5.0 (Windows; U; Windows NT 5.2; fr-FR; rv:1.7.8) Gecko/20050511 Firefox/1.0.4', :probability => 70
+  end
+
+  session :search do
+    request '/'
+
+    think_time 1..5
+
+    request '/signin', :params => { :uniq => true }
+
+    think 5
+
+    request '/signin', :method => :POST, :params => { :email => 'test@example.com', :password => 'ov7Feift' }
+  end
+end
+
+print rtsung.to_xml

data/README.md

@@ -0,0 +1,29 @@
+# Rtsung
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'rtsung'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install rtsung
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/examples/google.rb

@@ -0,0 +1,13 @@
+$: << '../lib'
+
+require 'rtsung'
+
+rtsung = RTsung.new do
+  server 'google.com'
+
+  session :search do
+    request '/'
+  end
+end
+
+print rtsung.to_xml

data/lib/rtsung.rb

@@ -0,0 +1,93 @@
+require 'builder'
+
+require 'rtsung/client'
+require 'rtsung/server'
+require 'rtsung/phase'
+require 'rtsung/option'
+require 'rtsung/session'
+
+class RTsung
+  LOG_LEVEL = :notice
+  DUMP_TRAFFIC = false
+
+  TSUNG_DTD = '/usr/share/tsung/tsung-1.0.dtd'
+
+  def initialize(options = {}, &block)
+    @log_level = options[:log_level] || LOG_LEVEL
+    @dump_traffic = options[:dump_traffic] || DUMP_TRAFFIC
+
+    @clients, @servers = [], []
+    @phases, @options = [], []
+    @sessions = []
+
+    instance_eval(&block) if block_given?
+  end
+
+  def client(host, options = {})
+    @clients << Client.new(host, options)
+  end
+
+  def server(host, options = {})
+    @servers << Server.new(host, options)
+  end
+
+  def phase(name = 1, duration = nil, unit = nil, options = {})
+    if duration.is_a? Hash
+      options = duration
+      duration, unit = nil, nil
+    end
+
+    if unit.is_a? Hash
+      options = unit
+      unit = nil
+    end
+
+    @phases << Phase.new(name, duration, unit, options)
+  end
+
+  def option(name, options = {}, &block)
+    @options << Option.new(name, options, &block)
+  end
+
+  def user_agents(&block)
+    @options << Option.new('user_agent', { :type => :ts_http }, &block)
+  end
+
+  def session(name, options = {}, &block)
+    @sessions << Session.new(name, options, &block)
+  end
+
+  def to_xml
+    output = ''
+
+    xml = ::Builder::XmlMarkup.new(:target => output, :indent => 2)
+
+    xml.instruct!
+    xml.declare! :DOCTYPE, :tsung, :SYSTEM, TSUNG_DTD, :'[]'
+
+    xml.tsung(:loglevel => @log_level) do
+      xml.clients do
+        if @clients.empty?
+          Client.new.to_xml(xml)
+        else
+          @clients.each { |client| client.to_xml(xml) }
+        end
+      end
+
+      xml.servers { @servers.each { |server| server.to_xml(xml) } }
+
+      xml.load do
+        if @phases.empty?
+          Phase.new.to_xml(xml)
+        else
+          @phases.each { |phase| phase.to_xml(xml) }
+        end
+      end
+
+      xml.options { @options.each { |o| o.to_xml(xml) } }
+      xml.sessions { @sessions.each { |s| s.to_xml(xml) } }
+    end
+
+    output
+  end
+end

data/lib/rtsung/client.rb

@@ -0,0 +1,32 @@
+class RTsung
+  class Client
+    HOST = 'localhost'
+    OPTIONS = { :vm => true }
+
+    def initialize(host = HOST, options = OPTIONS)
+      @attrs = { :host => host }
+
+      @attrs[:use_controller_vm] = options[:vm] if options[:vm]
+      @attrs[:maxusers] = options[:max_users] if options[:max_users]
+      @attrs[:weight] = options[:weight] if options[:weight]
+      @attrs[:cpu] = options[:cpu] if options[:cpu]
+
+      if ip = options[:ip]
+        @ips = ip.is_a?(Array) ? ip : [ip]
+      else
+        @ips = []
+      end
+    end
+
+    def to_xml(xml)
+      if @ips.empty?
+        xml.client @attrs
+      else
+        xml.client(@attrs) do
+          @ips.each { |i| xml.ip({ :value => i }) }
+        end
+      end
+    end
+
+  end
+end

data/lib/rtsung/option.rb

@@ -0,0 +1,44 @@
+class RTsung
+  class Option
+    TYPE = :ts_http
+
+    USER_AGENT_PROBABILITY = 100
+
+    def initialize(name, options = {}, &block)
+      @attrs = {
+        :name => name,
+        :type => options[:type] || TYPE
+      }
+
+      @user_agents = []
+
+      instance_eval(&block) if block_given?
+    end
+
+    def user_agent(name, options = {})
+      @user_agents << {
+        :name => name,
+        :probability => options[:probability] || USER_AGENT_PROBABILITY
+      }
+    end
+
+    def name(name, options = {})
+      user_agent(name, options)
+    end
+
+    def to_xml(xml)
+      if @user_agents.empty?
+        xml.option @attrs
+      else
+        xml.option(@attrs) do
+          @user_agents.each { |u|
+            xml.user_agent({ :probability => u[:probability] }) do
+              xml.text! u[:name]
+            end
+          }
+        end
+      end
+    end
+
+  end
+end

data/lib/rtsung/phase.rb

@@ -0,0 +1,39 @@
+class RTsung
+  class Phase
+    OPTIONS = { :interval => 1 }
+
+    PHASE_DURATION = 1
+    PHASE_UNIT = :minute
+
+    USERS_UNIT = :second
+
+    def initialize(name = 1, duration = nil, unit = nil, options = {})
+      @attrs = {
+        :phase => name,
+        :duration => duration || PHASE_DURATION,
+        :unit => unit || PHASE_UNIT
+      }
+
+      options = OPTIONS if options.empty?
+
+      if options[:rate]
+        @users = {
+          :arrivalrate => options[:rate],
+          :unit => options[:unit] || USERS_UNIT
+        }
+      else
+        @users = {
+          :interarrival => options[:interval],
+          :unit => options[:unit] || USERS_UNIT
+        }
+      end
+    end
+
+    def to_xml(xml)
+      xml.arrivalphase(@attrs) do
+        xml.users @users
+      end
+    end
+
+  end
+end

data/lib/rtsung/server.rb

@@ -0,0 +1,19 @@
+class RTsung
+  class Server
+    PORT = 80
+    TYPE = :tcp
+
+    def initialize(host, options = {})
+      @attrs = {
+        :host => host,
+        :port => options[:port] || PORT,
+        :type => options[:type] || TYPE
+      }
+    end
+
+    def to_xml(xml)
+      xml.server @attrs
+    end
+
+  end
+end

data/lib/rtsung/session.rb

@@ -0,0 +1,90 @@
+class RTsung
+  class Session
+    PROBABILITY = 100
+    TYPE = :ts_http
+
+    HTTP_VERSION = '1.1'
+    HTTP_METHOD = :GET
+
+    THINK_TIME_RANDOM = true
+
+
+    def initialize(name, options = {}, &block)
+      @attrs = {
+        :name => name,
+        :probability => options[:probability] || PROBABILITY,
+        :type => options[:type] || TYPE
+      }
+
+      @steps = []
+
+      instance_eval(&block) if block_given?
+    end
+
+    def request(url, options = {})
+      attrs = {
+        :url => url,
+        :version => options[:version] || HTTP_VERSION,
+        :method => options[:method] || HTTP_METHOD
+      }
+
+      if options[:params]
+        params = []
+        options[:params].keys.each { |k| params << "#{k}=#{options[:params][k]}" }
+
+        params = params.join('&amp;')
+
+        if attrs[:method] == :GET
+          attrs[:url] = "#{attrs[:url]}?#{params}"
+        elsif attrs[:method] == :POST
+          attrs[:contents] = params
+        end
+      end
+
+      attrs[:content_type] = options[:content_type] if options[:content_type]
+
+      @steps << {
+        :type => :request,
+        :attrs => attrs
+      }
+    end
+
+    def think_time(value, options = {})
+      if value.is_a?(Range)
+        attrs = {
+          :min => value.min,
+          :max => value.max
+        }
+      else
+        attrs = { :value => value }
+      end
+      
+      attrs[:random] = options[:random] || THINK_TIME_RANDOM
+      
+      @steps << {
+        :type => :think_time,
+        :attrs => attrs
+      }
+    end
+    alias :think :think_time
+
+    def to_xml xml
+      if @steps.empty?
+        xml.session @attrs
+      else
+        xml.session(@attrs) do
+          @steps.each { |s|
+            if s[:type] == :request
+              xml.request do
+                xml.http(s[:attrs]) 
+              end
+            elsif s[:type] == :think_time
+              xml.thinktime(s[:attrs])
+            end
+          }
+        end
+      end
+    end
+
+  end
+end

data/lib/rtsung/version.rb

@@ -0,0 +1,5 @@
+class RTsung
+  class Version
+    VERSION = "0.0.1"
+  end
+end

data/rtsung.gemspec

@@ -0,0 +1,25 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'rtsung/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "rtsung"
+  spec.version       = RTsung::Version::VERSION
+  spec.authors       = ["Tema Ivanov"]
+  spec.email         = []
+  spec.description   = %q{XML generator for Tsung}
+  spec.summary       = %q{XML generator for Tsung}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+
+  spec.add_dependency "builder", "~> 3.2.2"
+end

metadata

@@ -0,0 +1,100 @@
+--- !ruby/object:Gem::Specification
+name: rtsung
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Tema Ivanov
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-12-03 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: builder
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.2.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.2.2
+description: XML generator for Tsung
+email: []
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README
+- README.md
+- Rakefile
+- examples/google.rb
+- lib/rtsung.rb
+- lib/rtsung/client.rb
+- lib/rtsung/option.rb
+- lib/rtsung/phase.rb
+- lib/rtsung/server.rb
+- lib/rtsung/session.rb
+- lib/rtsung/version.rb
+- rtsung.gemspec
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.14
+signing_key: 
+specification_version: 4
+summary: XML generator for Tsung
+test_files: []