blitz 0.1.1 → 0.1.2

data/Gemfile

@@ -3,6 +3,7 @@ source "http://rubygems.org"
 # Example:
 #   gem "activesupport", ">= 2.3.5"
 
+gem "couchrest", "~> 1.0.1"
 gem "rest-client", "~> 1.6.1"
 gem "json", "~> 1.4.6"
 gem "json_pure", "~> 1.4.6"

data/Gemfile.lock

@@ -1,6 +1,10 @@
 GEM
   remote: http://rubygems.org/
   specs:
+    couchrest (1.0.1)
+      json (>= 1.4.6)
+      mime-types (>= 1.15)
+      rest-client (>= 1.5.1)
     git (1.2.5)
     hexy (0.1.1)
     jeweler (1.5.1)
@@ -19,6 +23,7 @@ PLATFORMS
 
 DEPENDENCIES
   bundler (~> 1.0.0)
+  couchrest (~> 1.0.1)
   hexy (~> 0.1.1)
   jeweler (~> 1.5.1)
   json (~> 1.4.6)

data/README.md

@@ -0,0 +1,36 @@
+## ![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
+    
+## Using the couch:fuzz command
+Simply point this blitz gem to your CouchDB URL and we'll auto generate
+full parameterized tests that you can use to measure view performance.
+
+    blitz http://localhost:5984 my_database
+    
+will generate tests like this:
+
+    -v:l number[1,10] -v:gl number[1,5] http://dell-5:5984/pcapr_local_root/_design/pcaps/_view/by_service?group=true&limit=#{l}&group_level=#{gl}
+    -v:sk alpha[4,12] -v:ek alpha[4,12] -v:l number[1,10] -v:id [true,false] http://dell-5:5984/pcapr_local_root/_design/pcaps/_view/indexed?startkey=%22am#{sk}%22&endkey=%22ykz#{ek}%22&include_docs=#{id}&limit=#{l}
+    
+which you can simply copy/paste to the [blitz.io](http://blitz.io). Your
+CouchDB must be on the public cloud though.
+
+Copyright (c) 2011 Mu Dynamics. See LICENSE.txt for further details.

data/blitz.gemspec

@@ -5,25 +5,25 @@
 
 Gem::Specification.new do |s|
   s.name = %q{blitz}
-  s.version = "0.1.1"
+  s.version = "0.1.2"
 
   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.date = %q{2011-03-15}
   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"
+    "README.md"
   ]
   s.files = [
     ".document",
     "Gemfile",
     "Gemfile.lock",
     "LICENSE.txt",
-    "README.rdoc",
+    "README.md",
     "Rakefile",
     "bin/blitz",
     "blitz.gemspec",
@@ -31,6 +31,7 @@ Gem::Specification.new do |s|
     "lib/blitz/client.rb",
     "lib/blitz/command.rb",
     "lib/blitz/command/api.rb",
+    "lib/blitz/command/couch.rb",
     "lib/blitz/command/curl.rb",
     "lib/blitz/command/help.rb",
     "lib/blitz/curl/error.rb",
@@ -55,6 +56,7 @@ Gem::Specification.new do |s|
     s.specification_version = 3
 
     if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<couchrest>, ["~> 1.0.1"])
       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"])
@@ -62,6 +64,7 @@ Gem::Specification.new do |s|
       s.add_development_dependency(%q<bundler>, ["~> 1.0.0"])
       s.add_development_dependency(%q<jeweler>, ["~> 1.5.1"])
     else
+      s.add_dependency(%q<couchrest>, ["~> 1.0.1"])
       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"])
@@ -70,6 +73,7 @@ Gem::Specification.new do |s|
       s.add_dependency(%q<jeweler>, ["~> 1.5.1"])
     end
   else
+    s.add_dependency(%q<couchrest>, ["~> 1.0.1"])
     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"])

data/lib/blitz.rb

@@ -1,10 +1,11 @@
 require 'rubygems'
+require 'couchrest'
 require 'hexy'
 require 'pp'
 
 class Blitz
     require 'blitz/helper'
-    Version = "0.1.1".freeze
+    Version = "0.1.2".freeze
         
     extend Blitz::Helper
     

data/lib/blitz/client.rb

@@ -1,7 +1,3 @@
-require 'rubygems'
-require 'json/pure'
-require 'restclient'
-
 class Blitz
 class Client
     attr_reader :blitz

data/lib/blitz/command/couch.rb

@@ -0,0 +1,136 @@
+require 'couchrest'
+
+class Blitz
+class Command
+class Couch < Command
+    attr_reader :urls
+
+    def initialize
+        @urls = []
+    end
+
+    def cmd_fuzz argv
+        assert argv.size != 0, "need a CouchDB URL (like http://localhost:5984)"
+
+        url = argv.shift
+        database = argv.shift
+
+        uri = URI.parse url rescue nil
+        assert_not_nil uri, "Can't parse #{url} - Is it valid?"
+
+        couch = CouchRest.new(uri)
+        couch.databases.reject { |n| n[0,1] == '_' }.each do |name|
+            next if database and database != name
+            msg "-------------- processing #{name}... ----------------------"
+            db = couch.database(name)
+            designs = db.documents :startkey => "_design", :endkey => "_design0", :include_docs => true
+            designs['rows'].each do |row|
+                design = row['doc']
+
+                # Ignore design documents that don't have views
+                next unless design['views']
+                analyze_design db, design
+            end
+
+            if urls.empty?
+                error "Nothing interesting found. Sorry."
+                return
+            end
+            
+            output_urls db
+            urls.clear
+        end
+    end
+
+    def analyze_design db, design
+        _, dname = design['_id'].split('/', 2)
+        views = design['views']
+        views.each_pair do |vname, view|
+            analyze_view db, dname, vname, view['reduce']
+        end
+    end
+
+    def analyze_view db, design, view, reduce
+        msg "  analyzing #{design}/#{view}..."
+
+        opts = {}
+        opts[:group] = true if reduce
+        first = db.view "#{design}/#{view}", opts.merge(:limit => 1)
+        last = db.view "#{design}/#{view}", opts.merge(:limit => 1, :descending => true)
+
+        return if first['rows'].empty?
+
+        first_row = first['rows'][0]
+        first_key = first_row['key']
+        last_row = last['rows'][0]
+        last_key = last_row['key']
+
+        # Randomize the limit so it's anywhere from 1 to 10 docs maximum
+        variables = { :l => 'number[1,10]' }
+        query = { :limit => '#{l}' }
+        if reduce
+            query.merge! :group => true
+        else
+            variables.merge! :id => '[true,false]'
+            query.merge! :include_docs => '#{id}'
+        end
+
+        case first_key
+        when Array
+            # If the key is an array and this view is reducing, randomize the 
+            # group level to have the query server work harder
+            if reduce
+                variables[:gl] = 'number[1,5]'
+                query[:group_level] = '#{gl}'
+            end
+        when Integer
+            # Randomize start/end keys, but within the valid range
+            variables[:sk] = "number[#{first_key},#{last_key}]"
+            variables[:ek] = "number[#{first_key},#{last_key}]"
+            query[:startkey] = '#{sk}'
+            query[:endkey] = '#{ek}'
+        when String
+            # Randomize start/end keys, but within the valid range
+            variables[:sk] = "alpha[4,12]"
+            variables[:ek] = "alpha[4,12]"
+            query[:startkey] = "%22#{first_key[0,2]}\#{sk}%22"
+            query[:endkey] = "%22#{last_key[0,2]}z\#{ek}%22"
+        when TrueClass, FalseClass
+            variables[:sk] = "[true,false]"
+            variables[:ek] = "[true,false,false,true]"
+            query[:startkey] = '#{sk}'
+            query[:endkey] = '#{ek}'
+        # TODO: Handle the case when the key is a Hash
+        end
+
+        vs = variables.to_a.map do |k, v|
+            if v =~ /\s/
+                "-v:#{k} '#{v}'"
+            else
+                "-v:#{k} #{v}"
+            end
+        end.join(' ')
+        qs = query.to_a.map { |k, v| "#{k}=#{v}" }.join('&')
+
+        # Generate blitz query URL's with all sorts of randomization
+        @urls << {
+            :reduce => reduce ? true : false,
+            :design => design,
+            :view => view,
+            :url => "#{vs} #{db.to_s}/_design/#{design}/_view/#{view}?#{qs}"
+        }
+    end
+
+    def output_urls db
+        puts
+        msg "Here are full parameterized blitz URL's for each view. Simply copy "
+        msg "paste these into the blitz bar @ http://blitz.io to generate load "
+        msg "on your couch."
+        puts
+        puts urls.map { |u| u[:url] }.join("\n\n")
+        puts
+        puts
+    end
+end
+end # Command
+end # Blitz

data/lib/blitz/command/help.rb

@@ -8,7 +8,8 @@ class Help < Command
             { :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' }
+            { :cmd => 'curl:help', :help => 'Show help on sprint and rushing' },
+            { :cmd => 'couch:fuzz', :help => 'Auto generate blitz tests from CouchDB' }
         ]
         
         max_cmd_size = helps.inject(0) { |memo, obj| [ obj[:cmd].size, memo ].max } + 4

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: blitz
 version: !ruby/object:Gem::Version 
-  hash: 25
+  hash: 31
   prerelease: false
   segments: 
   - 0
   - 1
-  - 1
-  version: 0.1.1
+  - 2
+  version: 0.1.2
 platform: ruby
 authors: 
 - pcapr
@@ -15,14 +15,30 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-03-10 00:00:00 -08:00
+date: 2011-03-15 00:00:00 -07:00
 default_executable: blitz
 dependencies: 
 - !ruby/object:Gem::Dependency 
   type: :runtime
   prerelease: false
-  name: rest-client
+  name: couchrest
   version_requirements: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 21
+        segments: 
+        - 1
+        - 0
+        - 1
+        version: 1.0.1
+  requirement: *id001
+- !ruby/object:Gem::Dependency 
+  type: :runtime
+  prerelease: false
+  name: rest-client
+  version_requirements: &id002 !ruby/object:Gem::Requirement 
     none: false
     requirements: 
     - - ~>
@@ -33,12 +49,12 @@ dependencies:
         - 6
         - 1
         version: 1.6.1
-  requirement: *id001
+  requirement: *id002
 - !ruby/object:Gem::Dependency 
   type: :runtime
   prerelease: false
   name: json
-  version_requirements: &id002 !ruby/object:Gem::Requirement 
+  version_requirements: &id003 !ruby/object:Gem::Requirement 
     none: false
     requirements: 
     - - ~>
@@ -49,12 +65,12 @@ dependencies:
         - 4
         - 6
         version: 1.4.6
-  requirement: *id002
+  requirement: *id003
 - !ruby/object:Gem::Dependency 
   type: :runtime
   prerelease: false
   name: json_pure
-  version_requirements: &id003 !ruby/object:Gem::Requirement 
+  version_requirements: &id004 !ruby/object:Gem::Requirement 
     none: false
     requirements: 
     - - ~>
@@ -65,12 +81,12 @@ dependencies:
         - 4
         - 6
         version: 1.4.6
-  requirement: *id003
+  requirement: *id004
 - !ruby/object:Gem::Dependency 
   type: :runtime
   prerelease: false
   name: hexy
-  version_requirements: &id004 !ruby/object:Gem::Requirement 
+  version_requirements: &id005 !ruby/object:Gem::Requirement 
     none: false
     requirements: 
     - - ~>
@@ -81,12 +97,12 @@ dependencies:
         - 1
         - 1
         version: 0.1.1
-  requirement: *id004
+  requirement: *id005
 - !ruby/object:Gem::Dependency 
   type: :development
   prerelease: false
   name: bundler
-  version_requirements: &id005 !ruby/object:Gem::Requirement 
+  version_requirements: &id006 !ruby/object:Gem::Requirement 
     none: false
     requirements: 
     - - ~>
@@ -97,12 +113,12 @@ dependencies:
         - 0
         - 0
         version: 1.0.0
-  requirement: *id005
+  requirement: *id006
 - !ruby/object:Gem::Dependency 
   type: :development
   prerelease: false
   name: jeweler
-  version_requirements: &id006 !ruby/object:Gem::Requirement 
+  version_requirements: &id007 !ruby/object:Gem::Requirement 
     none: false
     requirements: 
     - - ~>
@@ -113,7 +129,7 @@ dependencies:
         - 5
         - 1
         version: 1.5.1
-  requirement: *id006
+  requirement: *id007
 description: Make load and performance testing a fun sport
 email: kowsik@gmail.com
 executables: 
@@ -122,13 +138,13 @@ extensions: []
 
 extra_rdoc_files: 
 - LICENSE.txt
-- README.rdoc
+- README.md
 files: 
 - .document
 - Gemfile
 - Gemfile.lock
 - LICENSE.txt
-- README.rdoc
+- README.md
 - Rakefile
 - bin/blitz
 - blitz.gemspec
@@ -136,6 +152,7 @@ files:
 - lib/blitz/client.rb
 - lib/blitz/command.rb
 - lib/blitz/command/api.rb
+- lib/blitz/command/couch.rb
 - lib/blitz/command/curl.rb
 - lib/blitz/command/help.rb
 - lib/blitz/curl/error.rb

data/README.rdoc

@@ -1,22 +0,0 @@
-## ![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.