blitz 0.1.0

data/.document

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

data/Gemfile

@@ -0,0 +1,16 @@
+source "http://rubygems.org"
+# Add dependencies required to use your gem here.
+# Example:
+#   gem "activesupport", ">= 2.3.5"
+
+gem "rest-client", "~> 1.6.1"
+gem "json", "~> 1.4.6"
+gem "json_pure", "~> 1.4.6"
+gem "hexy", "~> 0.1.1"
+
+# Add dependencies to develop your gem here.
+# Include everything needed to run rake, tests, features, etc.
+group :development do
+  gem "bundler", "~> 1.0.0"
+  gem "jeweler", "~> 1.5.1"
+end

data/Gemfile.lock

@@ -0,0 +1,26 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    git (1.2.5)
+    hexy (0.1.1)
+    jeweler (1.5.1)
+      bundler (~> 1.0.0)
+      git (>= 1.2.5)
+      rake
+    json (1.4.6)
+    json_pure (1.4.6)
+    mime-types (1.16)
+    rake (0.8.7)
+    rest-client (1.6.1)
+      mime-types (>= 1.16)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  bundler (~> 1.0.0)
+  hexy (~> 0.1.1)
+  jeweler (~> 1.5.1)
+  json (~> 1.4.6)
+  json_pure (~> 1.4.6)
+  rest-client (~> 1.6.1)

data/LICENSE.txt

@@ -0,0 +1,20 @@
+Copyright (c) 2011 Mu Dynamics
+
+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,22 @@
+## ![blitz.io](http://blitz.io/images/logo2.png)
+
+Make load and performance a fun sport.
+
+* Run a sprint from around the world
+* Rush your API and website to scale it out
+* Condition your site around the clock
+
+## Getting started
+Login to [blitz.io](http://blitz.io) and in the blitz bar type:
+    --api-key
+
+Now
+    gem install blitz
+    
+and run a sprint like this:
+    blitz curl --region california http://blitz.io
+    
+and you can rush like this:
+    blitz curl --region california --pattern 1-100:60 http://blitz.io
+
+Copyright (c) 2011 Mu Dynamics. See LICENSE.txt for further details.

data/Rakefile

@@ -0,0 +1,51 @@
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), 'lib'))
+
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+
+require 'rake'
+require 'jeweler'
+require 'blitz'
+Jeweler::Tasks.new do |gem|
+  # gem is a Gem::Specification... see http://docs.rubygems.org/read/chapter/20 for more options
+  gem.name = "blitz"
+  gem.homepage = "http://blitz.io"
+  gem.license = "MIT"
+  gem.summary = %Q{Make load and performance testing a fun sport}
+  gem.description = %Q{Make load and performance testing a fun sport}
+  gem.email = "kowsik@gmail.com"
+  gem.authors = ["pcapr"]
+  gem.version = Blitz::Version
+  gem.executables = ['blitz']
+
+  # Include your dependencies below. Runtime dependencies are required when using your gem,
+  # and development dependencies are only needed for development (ie running rake tasks, tests, etc)
+  #  gem.add_runtime_dependency 'jabber4r', '> 0.1'
+  #  gem.add_development_dependency 'rspec', '> 1.2.3'
+end
+Jeweler::RubygemsDotOrgTasks.new
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+task :default => :test
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "blitz #{Blitz::Version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+

data/bin/blitz

@@ -0,0 +1,11 @@
+#!/usr/bin/env ruby
+
+lib = File.expand_path(File.dirname(__FILE__) + '/../lib')
+$LOAD_PATH.unshift(lib) if File.directory?(lib) && !$LOAD_PATH.include?(lib)
+
+require 'blitz'
+
+argv = ARGV.dup
+ARGV.clear
+cmd = argv.shift.strip rescue 'help'
+Blitz.run cmd, argv

data/blitz.gemspec

@@ -0,0 +1,80 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{blitz}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["pcapr"]
+  s.date = %q{2011-03-10}
+  s.default_executable = %q{blitz}
+  s.description = %q{Make load and performance testing a fun sport}
+  s.email = %q{kowsik@gmail.com}
+  s.executables = ["blitz"]
+  s.extra_rdoc_files = [
+    "LICENSE.txt",
+    "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+    "Gemfile",
+    "Gemfile.lock",
+    "LICENSE.txt",
+    "README.rdoc",
+    "Rakefile",
+    "bin/blitz",
+    "blitz.gemspec",
+    "lib/blitz.rb",
+    "lib/blitz/client.rb",
+    "lib/blitz/command.rb",
+    "lib/blitz/command/api.rb",
+    "lib/blitz/command/curl.rb",
+    "lib/blitz/command/help.rb",
+    "lib/blitz/curl/error.rb",
+    "lib/blitz/curl/sprint.rb",
+    "lib/blitz/helper.rb",
+    "test/helper.rb",
+    "test/test_blitz.rb"
+  ]
+  s.homepage = %q{http://blitz.io}
+  s.licenses = ["MIT"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.7}
+  s.summary = %q{Make load and performance testing a fun sport}
+  s.test_files = [
+    "test/helper.rb",
+    "test/test_blitz.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_runtime_dependency(%q<rest-client>, ["~> 1.6.1"])
+      s.add_runtime_dependency(%q<json>, ["~> 1.4.6"])
+      s.add_runtime_dependency(%q<json_pure>, ["~> 1.4.6"])
+      s.add_runtime_dependency(%q<hexy>, ["~> 0.1.1"])
+      s.add_development_dependency(%q<bundler>, ["~> 1.0.0"])
+      s.add_development_dependency(%q<jeweler>, ["~> 1.5.1"])
+    else
+      s.add_dependency(%q<rest-client>, ["~> 1.6.1"])
+      s.add_dependency(%q<json>, ["~> 1.4.6"])
+      s.add_dependency(%q<json_pure>, ["~> 1.4.6"])
+      s.add_dependency(%q<hexy>, ["~> 0.1.1"])
+      s.add_dependency(%q<bundler>, ["~> 1.0.0"])
+      s.add_dependency(%q<jeweler>, ["~> 1.5.1"])
+    end
+  else
+    s.add_dependency(%q<rest-client>, ["~> 1.6.1"])
+    s.add_dependency(%q<json>, ["~> 1.4.6"])
+    s.add_dependency(%q<json_pure>, ["~> 1.4.6"])
+    s.add_dependency(%q<hexy>, ["~> 0.1.1"])
+    s.add_dependency(%q<bundler>, ["~> 1.0.0"])
+    s.add_dependency(%q<jeweler>, ["~> 1.5.1"])
+  end
+end
+

data/lib/blitz.rb

@@ -0,0 +1,31 @@
+require 'rubygems'
+require 'hexy'
+
+class Blitz
+    require 'blitz/helper'
+    Version = "0.1.0".freeze
+        
+    extend Blitz::Helper
+    
+    def self.run cmd, argv
+        kname, mname = cmd.split(':', 2)
+        klass = Blitz::Command.const_get kname.capitalize rescue nil
+        mname ||= 'default'
+        mname = "cmd_#{mname}".to_sym
+        if klass and klass < Blitz::Command and klass.method_defined? mname
+            command = klass.new
+            begin
+                command.send mname, argv
+            rescue Test::Unit::AssertionFailedError, ArgumentError => e
+                error e.message.chomp('.')
+            end
+        else
+            error "Unknown command #{cmd}"
+        end        
+    end    
+end
+
+require 'blitz/client'
+require 'blitz/curl/error'
+require 'blitz/curl/sprint'
+require 'blitz/command'

data/lib/blitz/client.rb

@@ -0,0 +1,35 @@
+require 'rubygems'
+require 'json/pure'
+require 'restclient'
+
+class Blitz
+class Client
+    attr_reader :blitz
+    
+    def initialize user, apik, host='blitz.io'
+        scheme = host.index('localhost') ? 'http' : 'https'
+        @blitz = RestClient::Resource.new "#{scheme}://#{host}", \
+            :headers => {
+                :x_api_user => user,
+                :x_api_key => apik,
+                :x_gem_version => ::Blitz::Version
+            }
+    end
+    
+    def curl_execute data
+        JSON.parse blitz['/api/1/curl/execute'].post(data.to_json)
+    end
+    
+    def login
+        JSON.parse blitz['/login/api'].get
+    end
+    
+    def job_status job_id
+        JSON.parse blitz["/api/1/jobs/#{job_id}/status"].get
+    end
+    
+    def abort_job job_id
+        JSON.parse blitz["/api/1/jobs/#{job_id}/abort"].put ''
+    end
+end
+end

data/lib/blitz/command.rb

@@ -0,0 +1,22 @@
+require 'test/unit/assertions'
+
+# The default template string contains what was sent and received. Strip 
+# these out since we don't need them
+class Test::Unit::Assertions::AssertionMessage
+    alias :old_template :template
+    
+    def template
+        @template_string = ''
+        @parameters = []
+        old_template
+    end
+end
+
+class Blitz
+class Command
+    include Test::Unit::Assertions
+    include Helper
+end
+end # Blitz
+
+Dir["#{File.dirname(__FILE__)}/command/*.rb"].each { |c| require c }

data/lib/blitz/command/api.rb

@@ -0,0 +1,96 @@
+class Blitz
+class Command
+class API < Command
+    attr_accessor :credentials
+        
+    def cmd_init argv
+        FileUtils.rm credentials_file rescue nil
+        API.client
+        
+        msg "You are now ready to blitz!"
+        msg "Try blitz help to learn more about the commands."
+    end
+    
+    def client
+        get_credentials
+        Blitz::Client.new(user, password, host)
+    end
+    
+    def user
+        get_credentials
+        @credentials[0]
+    end
+    
+    def password
+        get_credentials
+        @credentials[1]
+    end
+    
+    def host
+        ENV['BLITZ_HOST'] || 'blitz.io'
+    end
+    
+    def credentials_file
+        ENV['HOME'] + '/.blitz/credentials'
+    end
+    
+    def get_credentials
+        return if @credentials
+        unless @credentials = read_credentials
+            @credentials = ask_for_credentials
+            save_credentials
+        end
+        @credentials
+    end
+    
+    def read_credentials
+        File.exists?(credentials_file) and File.read(credentials_file).split("\n")        
+    end
+    
+    def ask_for_credentials
+        msg "Enter your blitz credentials."
+        print "User-ID: "
+        user = ask
+        print "API-Key: "
+        apik = ask
+        apik2 = Blitz::Client.new(user, apik, host).login['api_key']
+        if not apik2
+            error "Authentication failed"
+            exit 1
+        end
+        
+        [ user, apik2 ]
+    end
+    
+    def save_credentials
+        write_credentials
+    end
+    
+    def write_credentials
+        FileUtils.mkdir_p(File.dirname(credentials_file))
+        File.open(credentials_file, 'w') do |f|
+          f.puts self.credentials
+        end
+        set_credentials_permissions
+    end
+    
+    def set_credentials_permissions
+        FileUtils.chmod 0700, File.dirname(credentials_file)
+        FileUtils.chmod 0600, credentials_file        
+    end
+    
+    def self.instance
+        @instance ||= API.new
+    end
+    
+    def self.client
+        self.instance.client
+    end
+    
+    private
+    def initialize
+    end
+end
+Api = API
+end # Command
+end # Blitz

data/lib/blitz/command/curl.rb

@@ -0,0 +1,272 @@
+class Blitz
+class Command
+class Curl < Command
+    def cmd_help argv
+        help
+    end
+
+    def cmd_run argv
+        args = parse_cli argv
+        if args['help']
+            return help
+        end
+
+        if not args['pattern']
+            sprint args
+            return
+        else
+            error "Rushing not supported yet, but coming soon"
+        end
+    end
+    
+    alias_method :cmd_default, :cmd_run
+
+    private
+    def authorize_error e
+        base_url = "#{e.scheme}://#{e.host}:#{e.port}"
+        puts
+        error "You haven't verified that you are the devops dude for #{e.host}"
+        error ""
+        error "Make sure the following URL is reachable by us and returns the value 42."
+        error "#{base_url}/#{e.uuid}"
+        error ""
+        error "If your app is RESTfully built with sinatra or rails, simply"
+        error "add a route like so:"
+        error ""
+        error "get '/#{e.uuid}' do"
+        error "    '42'"
+        error "end"
+        error ""
+        error "And we'll be on our merry way to blitz #{e.host}"
+        puts
+    end
+
+    def sprint args
+        begin
+            job = ::Blitz::Curl::Sprint.execute args
+            result = job.result
+            print_sprint_result args, result
+        rescue ::Blitz::Curl::Error::Authorize => e
+            authorize_error e
+        rescue ::Blitz::Curl::Error::Region => e
+            error "#{e.region}: #{e.message}"
+        rescue ::Blitz::Curl::Error => e
+            error e.message
+        end
+    end
+
+    def print_sprint_result args, result
+        rtt = result.duration
+		if rtt < 1.0
+			rtt = (rtt * 1000).floor.to_s + ' ms';
+		else
+			rtt = ("%.2f" % rtt) + ' sec';
+		end
+
+		puts "-" * 70
+        msg "#{result.region}: Response time of #{rtt}"
+        unless args['dump-header'] or args['verbose']
+            msg "Try --verbose to see the request/response headers"
+        end
+        puts "-" * 70
+        puts
+
+        if args['dump-header'] or args['verbose']
+            puts "> " + result.request.line
+            result.request.headers.each_pair { |k, v| puts "> #{k}: #{v}\r\n" }
+            puts
+        
+            content = result.request.content
+            if not content.empty?
+                if /^[[:print:]]+$/ =~ content
+                    puts content
+                else
+                    puts Hexy.new(content).to_s
+                end
+                puts
+            end
+        
+            puts "< " + result.response.line
+            result.response.headers.each_pair { |k, v| puts "> #{k}: #{v}\r\n" }
+            puts
+        end
+        
+        content = result.response.content
+        if not content.empty?
+            if /^[[:print:]]+$/ =~ content
+                puts content
+            else
+                puts Hexy.new(content).to_s
+            end
+        end
+    end
+
+    def help
+        helps = [
+            { :short => '-A', :long => '--user-agent', :value => '<string>', :help => 'User-Agent to send to server' },
+            { :short => '-b', :long => '--cookie', :value => 'name=<string>', :help => 'Cookie to send to the server (multiple)' },
+            { :short => '-d', :long => '--data', :value => '<string>', :help => 'Data to send in a PUT or POST request' },
+            { :short => '-D', :long => '--dump-header', :value => '<file>', :help => 'Print the request/response headers' },
+            { :short => '-e', :long => '--referer', :value => '<string>', :help => 'Referer URL' },
+            { :short => '-h', :long => '--help', :value => '', :help => 'Help on command line options' },
+            { :short => '-H', :long => '--header', :value => '<string>', :help => 'Custom header to pass to server' },
+            { :short => '-p', :long => '--pattern', :value => '<s>-<e>:<d>', :help => 'Ramp from s to e concurrent requests in d secs' },
+            { :short => '-r', :long => '--region', :value => '<string>', :help => 'Choose one of california, virginia, singapore or ireland' },
+            { :short => '-s', :long => '--status', :value => '<number>', :help => 'Assert on the HTTP response status code' },
+            { :short => '-T', :long => '--timeout', :value => '<ms>', :help => 'Wait time for both connect and responses' },
+            { :short => '-u', :long => '--user', :value => '<user[:pass]>', :help => 'User and password for authentication' },
+            { :short => '-X', :long => '--request', :value => '<string>', :help => 'Request method to use (GET, HEAD, PUT, etc.)' },
+            { :short => '-v', :long => '--verbose', :value => '', :help => 'Print the request/response headers' },
+            { :short => '-1', :long => '--tlsv1', :value => '', :help => 'Use TLSv1 (SSL)' },
+            { :short => '-2', :long => '--sslv2', :value => '', :help => 'Use SSLv2 (SSL)' },
+            { :short => '-3', :long => '--sslv3', :value => '', :help => 'Use SSLv3 (SSL)' }
+        ]
+
+        max_long_size = helps.inject(0) { |memo, obj| [ obj[:long].size, memo ].max }
+        max_value_size = helps.inject(0) { |memo, obj| [ obj[:value].size, memo ].max }
+        puts
+        msg "Usage: blitz curl <options> <url>"
+        puts
+        helps.each do |h|
+            msg "%-*s %*s %-*s %s" % [max_long_size, h[:long], 2, h[:short], max_value_size, h[:value], h[:help]]
+        end
+        puts
+    end
+
+    def parse_cli argv
+        hash = Hash.new
+        while not argv.empty?
+            break if argv.first[0,1] != '-'
+
+            k = argv.shift
+            if [ '-A', '--user-agent' ].member? k
+                hash['user-agent'] = shift(k, argv)
+                next
+            end
+
+            if [ '-b', '--cookie' ].member? k
+                # TODO: support cookie jars
+                hash['cookies'] ||= []
+                hash['cookies'] << shift(k, argv)
+                next
+            end
+
+            if [ '-d', '--data' ].member? k
+                hash['content'] ||= Hash.new
+                hash['content']['data'] ||= []
+                v = shift(k, argv)
+                v = File.read v[1..-1] if v =~ /^@/
+                hash['content']['data'] << v
+                next
+            end
+
+            if [ '-D', '--dump-header' ].member? k
+                hash['dump-header'] = shift(k, argv)
+                next
+            end
+
+            if [ '-e', '--referer'].member? k
+                hash['referer'] = shift(k, argv)
+                next
+            end
+
+            if [ '-h', '--help' ].member? k
+                hash['help'] = true
+                next
+            end
+
+            if [ '-H', '--header' ].member? k
+                hash['headers'] ||= []
+                hash['headers'].push shift(k, argv)
+                next
+            end
+
+            if [ '-p', '--pattern' ].member? k
+                v = shift(k, argv)
+                if not /^(\d+)-(\d+):(\d+)$/ =~ v
+                    raise Test::Unit::AssertionFailedError, "invalid ramp pattern"
+                end
+                hash['pattern'] = {
+                    'iterations' => 1,
+                    'intervals' => [{
+                        'iterations' => 1,
+                        'start' => $1.to_i,
+                        'end' => $2.to_i,
+                        'duration' => $3.to_i
+                    }]
+                }
+                next
+            end
+
+            if [ '-r', '--region' ].member? k
+                v = shift(k, argv)
+                assert_match(/^california|virginia|singapore|ireland|japan$/, v, 'region must be one of california, virginia, singapore, japan or ireland')
+                hash['region'] = v
+                next
+            end
+
+            if [ '-s', '--status' ].member? k
+                hash['status'] = shift(k, argv).to_i
+                next
+            end
+
+            if [ '-T', '--timeout' ].member? k
+                hash['timeout'] = shift(k, argv).to_i
+                next
+            end
+
+            if [ '-u', '--user' ].member? k
+                hash['user'] = shift(k, argv)
+                next
+            end
+
+            if [ '-X', '--request' ].member? k
+                hash['request'] = shift(k, argv)
+                next
+            end
+            
+            if [ '-v', '--verbose' ].member? k
+                hash['verbose'] = true
+                next
+            end
+
+            if [ '-1', '--tlsv1' ].member? k
+                hash['ssl'] = 'tlsv1'
+                next
+            end
+
+            if [ '-2', '--sslv2' ].member? k
+                hash['ssl'] = 'sslv2'
+                next
+            end
+
+            if [ '-3', '--sslv3' ].member? k
+                hash['ssl'] = 'sslv2'
+                next
+            end
+
+            raise ArgumentError, "Unknown option #{k}"
+        end
+
+        if not hash['help']
+            url = argv.shift
+            assert_not_nil(url, 'no URL specified!')
+            hash['url'] = url
+        end
+        
+        if hash.member? 'content'
+            data_size = hash['content']['data'].inject(0) { |m, v| m + v.size }
+            assert(data_size < 10*1024, "POST content must be < 10K")
+        end
+
+        hash
+    end
+
+    def shift key, argv
+        val = argv.shift
+        assert_not_nil(val, "missing value for #{key}")
+        val
+    end
+end # Curl
+end # Command
+end # Blitz

data/lib/blitz/command/help.rb

@@ -0,0 +1,22 @@
+class Blitz
+class Command
+class Help < Command
+    def cmd_default argv
+        puts
+        msg "Usage: blitz <command> <options>"
+        helps = [
+            { :cmd => 'help', :help => "Display this help" },
+            { :cmd => 'api:init', :help => 'Validate your API key' },
+            { :cmd => 'curl', :help => 'Run a sprint or a rush' },
+            { :cmd => 'curl:help', :help => 'Show help on sprint and rushing' }
+        ]
+        
+        max_cmd_size = helps.inject(0) { |memo, obj| [ obj[:cmd].size, memo ].max } + 4
+        helps.each do |h|
+            msg "%*s - %s" % [max_cmd_size, h[:cmd], h[:help]]
+        end
+        puts
+    end
+end
+end # Command
+end # Blitz

data/lib/blitz/curl/error.rb

@@ -0,0 +1,45 @@
+class Blitz
+module Curl
+class Error < StandardError
+    def initialize json={}
+        super json['reason'] || "Hmmm, something went wrong. Try again in a little bit?"
+    end
+    
+    class Authorize < Error
+        attr_reader :scheme, :host, :port, :uuid
+
+        def initialize json
+            @scheme = json['scheme']
+            @host   = json['host']
+            @port   = json['port']
+            @uuid   = json['uuid']
+            super
+        end
+    end
+    
+    class Region < Error
+        attr_reader :region
+        
+        def initialize json
+            @region = json['region']
+            super
+        end
+    end
+    
+    class DNS < Region
+    end
+    
+    class Connect < Region
+    end
+    
+    class Timeout < Region
+    end
+
+    class Parse < Region
+    end
+    
+    class Status < Region
+    end
+end
+end # Curl
+end # Blitz

data/lib/blitz/curl/sprint.rb

@@ -0,0 +1,108 @@
+class Blitz
+module Curl 
+class Sprint
+    class Request
+        attr_reader :line
+        attr_reader :method
+        attr_reader :url
+        attr_reader :headers
+        attr_reader :content
+        
+        def initialize json
+            @line = json['line']
+            @method = json['method']
+            @url = json['url']
+            @content = json['content'].unpack('m')[0]
+            @headers = json['headers']
+        end
+    end
+    
+    class Response
+        attr_reader :line
+        attr_reader :status
+        attr_reader :message
+        attr_reader :headers
+        attr_reader :content
+        
+        def initialize json
+            @line = json['line']
+            @status = json['status']
+            @message = json['message']
+            @content = json['content'].unpack('m')[0]
+            @headers = json['headers']
+        end        
+    end
+    
+    class Result
+        attr_reader :region
+        attr_reader :duration
+        attr_reader :connect
+        attr_reader :request
+        attr_reader :response
+        
+        def initialize json
+            result = json['result']
+            @region = result['region']
+            @duration = result['duration']
+            @connect = result['connect']
+            @request = Request.new result['request']
+            @response = Response.new result['response']
+        end        
+    end
+    
+    def self.execute args
+        args.delete 'pattern'
+
+        res = Command::API.client.curl_execute args
+        raise Error.new(res) if res['error']
+        return self.new res['job_id']
+    end
+    
+    attr_reader :job_id
+    
+    def initialize job_id
+        @job_id = job_id
+    end
+    
+    def result
+        while true
+            sleep 2.0
+
+            job = Command::API.client.job_status job_id
+            if job['error']
+                raise Error
+            end
+
+            result = job['result']
+            next if job['status'] == 'queued'
+            next if job['status'] == 'running' and not result
+
+            raise Error if not result
+
+            error = result['error']
+            if error
+                if error == 'dns'
+                    raise Error::DNS.new(result)
+                elsif error == 'connect'
+                    raise Error::Connect.new(result)
+                elsif error == 'timeout'
+                    raise Error::Timeout.new(result)
+                elsif error == 'parse'
+                    raise Error::Parse.new(result)
+                elsif result['assert'] == 0
+                    raise Error::Status.new(result)
+                else
+                    raise Error
+                end
+            end
+            
+            return Result.new(job)
+        end
+    end
+    
+    def abort
+        Command::API.client.abort_job job_id rescue nil
+    end    
+end
+end # Curl
+end # Blitz

data/lib/blitz/helper.rb

@@ -0,0 +1,15 @@
+class Blitz
+module Helper
+    def error msg
+        $stderr.puts "!! #{msg}"
+    end
+    
+    def msg msg
+        puts msg
+    end
+    
+    def ask
+        gets.strip
+    end
+end
+end # Blitz

data/test/helper.rb

@@ -0,0 +1,17 @@
+require 'rubygems'
+require 'bundler'
+begin
+    Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+    $stderr.puts e.message
+    $stderr.puts "Run `bundle install` to install missing gems"
+    exit e.status_code
+end
+require 'test/unit'
+
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'blitz'
+
+class Test::Unit::TestCase
+end

data/test/test_blitz.rb

@@ -0,0 +1,6 @@
+require 'helper'
+
+class TestBlitz < Test::Unit::TestCase
+    def test_something_for_real
+    end
+end

metadata

@@ -0,0 +1,182 @@
+--- !ruby/object:Gem::Specification 
+name: blitz
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- pcapr
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-03-10 00:00:00 -08:00
+default_executable: blitz
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  type: :runtime
+  prerelease: false
+  name: rest-client
+  version_requirements: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 13
+        segments: 
+        - 1
+        - 6
+        - 1
+        version: 1.6.1
+  requirement: *id001
+- !ruby/object:Gem::Dependency 
+  type: :runtime
+  prerelease: false
+  name: json
+  version_requirements: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 11
+        segments: 
+        - 1
+        - 4
+        - 6
+        version: 1.4.6
+  requirement: *id002
+- !ruby/object:Gem::Dependency 
+  type: :runtime
+  prerelease: false
+  name: json_pure
+  version_requirements: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 11
+        segments: 
+        - 1
+        - 4
+        - 6
+        version: 1.4.6
+  requirement: *id003
+- !ruby/object:Gem::Dependency 
+  type: :runtime
+  prerelease: false
+  name: hexy
+  version_requirements: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 25
+        segments: 
+        - 0
+        - 1
+        - 1
+        version: 0.1.1
+  requirement: *id004
+- !ruby/object:Gem::Dependency 
+  type: :development
+  prerelease: false
+  name: bundler
+  version_requirements: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 23
+        segments: 
+        - 1
+        - 0
+        - 0
+        version: 1.0.0
+  requirement: *id005
+- !ruby/object:Gem::Dependency 
+  type: :development
+  prerelease: false
+  name: jeweler
+  version_requirements: &id006 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 1
+        segments: 
+        - 1
+        - 5
+        - 1
+        version: 1.5.1
+  requirement: *id006
+description: Make load and performance testing a fun sport
+email: kowsik@gmail.com
+executables: 
+- blitz
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE.txt
+- README.rdoc
+files: 
+- .document
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.rdoc
+- Rakefile
+- bin/blitz
+- blitz.gemspec
+- lib/blitz.rb
+- lib/blitz/client.rb
+- lib/blitz/command.rb
+- lib/blitz/command/api.rb
+- lib/blitz/command/curl.rb
+- lib/blitz/command/help.rb
+- lib/blitz/curl/error.rb
+- lib/blitz/curl/sprint.rb
+- lib/blitz/helper.rb
+- test/helper.rb
+- test/test_blitz.rb
+has_rdoc: true
+homepage: http://blitz.io
+licenses: 
+- MIT
+post_install_message: 
+rdoc_options: []
+
+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: Make load and performance testing a fun sport
+test_files: 
+- test/helper.rb
+- test/test_blitz.rb