computron 0.1.0

data/.bundle/config

@@ -0,0 +1,2 @@
+--- 
+BUNDLE_WITHOUT: ""

data/.document

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

data/.gitignore

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

data/Gemfile

@@ -0,0 +1,3 @@
+gem 'eventmachine'
+gem 'em-http-request'
+gem 'colored'

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Brad Gessler
+
+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,13 @@
+= computron
+
+Computron is like ab on steroids, implemented in Ruby using em-http-request. It was designed out of a need to simulate real users on a website. It also does a few smart things like decode JSON into a Ruby hash so you can use those responses in your next requests.
+
+= The DSL
+
+Computron is a simple DLS around HTTP em that makes more sense to write user tests. 
+
+= Report Summary
+
+== Copyright
+
+Copyright (c) 2010 Brad Gessler. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,48 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "computron"
+    gem.summary = %Q{An evented HTTP computron}
+    gem.description = %Q{Computron was created to help Poll Everywhere peform load tests that simulate how real users work.}
+    gem.email = "brad@bradgessler.com"
+    gem.homepage = "http://github.com/bradgessler/computron"
+    gem.authors = ["Brad Gessler"]
+    gem.add_dependency "eventmachine", ">= 0.12.10"
+    gem.add_dependency "em-http-request", ">= 0.2.7"
+    gem.add_dependency "colored", ">= 1.1"
+    gem.add_development_dependency "rspec", ">= 1.2.9"
+    gem.add_development_dependency "yard", ">= 0"
+    # 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
+
+begin
+  require 'yard'
+  YARD::Rake::YardocTask.new
+rescue LoadError
+  task :yardoc do
+    abort "YARD is not available. In order to run yardoc, you must: sudo gem install yard"
+  end
+end

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/bin/computron

@@ -0,0 +1,15 @@
+#!/usr/bin/env ruby
+
+computron_dir = File.expand_path(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(computron_dir) unless $LOAD_PATH.include?(computron_dir)
+
+require 'rubygems'
+require 'computron'
+
+simulation = Computron::DSL.new(File.read(ARGV.last))
+
+Signal.trap('INT'){ 
+  EM.stop 
+}
+
+simulation.run!

data/computron.gemspec

@@ -0,0 +1,76 @@
+# 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{computron}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Brad Gessler"]
+  s.date = %q{2010-05-24}
+  s.default_executable = %q{computron}
+  s.description = %q{Computron was created to help Poll Everywhere peform load tests that simulate how real users work.}
+  s.email = %q{brad@bradgessler.com}
+  s.executables = ["computron"]
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.rdoc"
+  ]
+  s.files = [
+    ".bundle/config",
+     ".document",
+     ".gitignore",
+     "Gemfile",
+     "LICENSE",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "bin/computron",
+     "computron.gemspec",
+     "lib/computron.rb",
+     "lib/computron/client.rb",
+     "lib/computron/dsl.rb",
+     "lib/computron/report.rb",
+     "spec/computron_spec.rb",
+     "spec/spec.opts",
+     "spec/spec_helper.rb",
+     "test.rb"
+  ]
+  s.homepage = %q{http://github.com/bradgessler/computron}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.6}
+  s.summary = %q{An evented HTTP computron}
+  s.test_files = [
+    "spec/computron_spec.rb",
+     "spec/spec_helper.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<eventmachine>, [">= 0.12.10"])
+      s.add_runtime_dependency(%q<em-http-request>, [">= 0.2.7"])
+      s.add_runtime_dependency(%q<colored>, [">= 1.1"])
+      s.add_development_dependency(%q<rspec>, [">= 1.2.9"])
+      s.add_development_dependency(%q<yard>, [">= 0"])
+    else
+      s.add_dependency(%q<eventmachine>, [">= 0.12.10"])
+      s.add_dependency(%q<em-http-request>, [">= 0.2.7"])
+      s.add_dependency(%q<colored>, [">= 1.1"])
+      s.add_dependency(%q<rspec>, [">= 1.2.9"])
+      s.add_dependency(%q<yard>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<eventmachine>, [">= 0.12.10"])
+    s.add_dependency(%q<em-http-request>, [">= 0.2.7"])
+    s.add_dependency(%q<colored>, [">= 1.1"])
+    s.add_dependency(%q<rspec>, [">= 1.2.9"])
+    s.add_dependency(%q<yard>, [">= 0"])
+  end
+end
+

data/lib/computron/client.rb

@@ -0,0 +1,186 @@
+require 'logger'
+require 'eventmachine'
+require 'em-http'
+require 'colored'
+
+module Computron
+  class Client
+    class Response
+
+      attr_reader :status
+
+      def initialize(opts={})
+        @status = {}
+        yield self if block_given?
+      end
+
+      def call(http)
+        if callback = status[http.response_header.status]
+          callback.call(decode_response(http))
+        end
+        @finished.call  if @finished
+      end
+
+      # A shortcut for specifying a callback on an HTTP code if you don't know the status name.
+      def code(code, &block)
+        status[code] = block
+      end
+
+      # Runs after every HTTP request
+      def finished(&block)
+        @finished = block
+      end
+
+      # Returns a hash containing the response code to the status like {200 => ok}.
+      def self.http_status_codes
+        @http_status_codes ||= %(100 continue
+          101 switching_protocols
+          200 ok
+          201 created
+          202 accepted
+          203 non_authoritive_information
+          204 no_content
+          205 reset_content
+          206 partial_content
+          300 multiple_choices
+          301 moved_permanently
+          302 found
+          303 see_other
+          304 not_modified
+          305 use_proxy
+          307 temporary_redirect
+          400 bad_request
+          401 unauthorized
+          402 payment_required
+          403 forbidden
+          404 not_found
+          405 method_not_allowed
+          406 not_acceptable
+          407 proxy_authentication_required
+          408 request_timeout
+          409 conflict
+          410 gone
+          411 length_required
+          412 precondition_failed
+          413 request_entity_too_large
+          414 request_uri_too_long
+          415 unsupported_media_type
+          416 request_range_not_satisfiable
+          417 expectation_failed
+          500 internal_server_error
+          501 not_implemented
+          502 bad_gateway
+          503 service_unavailable
+          504 gateway_timeout
+          505 http_version_not_supported).inject({}) do |hash, line|
+            code, status = line.split(' ')
+            hash[code.to_i] = status
+            hash
+          end
+      end
+
+      http_status_codes.each do |code, status|
+        eval %(
+          def #{status}(&block)
+            status[#{code}] = block
+          end)
+      end
+
+    private
+      def decode_response(http)
+        content_type = http.response_header['CONTENT_TYPE']
+
+        case content_type
+        when %r{^application/json}
+          response = ActiveSupport::JSON.decode(http.response)
+        else
+          response = http.response # Do nothing, just return text
+        end
+
+      rescue
+        raise StandardError.new("Problem decoding '#{content_type}': \n#{http.response}")
+      end
+    end
+
+    attr_accessor :cookie, :name, :logger
+
+    def initialize
+      @logger = Logger.new($stdout)
+      yield self if block_given?
+    end
+
+    def long_poll(url, opts={}, &block)
+      response  = Response.new
+      request   = EM::HttpRequest.new(url).get(opts)
+
+      request.callback {|http|
+        persist_cookie(http)
+        repoll = Struct.new(:url).new(url)
+        block.call(response, repoll) if block_given?
+        response.call(http)
+        log_request(http, "Repolling")
+        long_poll(repoll.url, &block)
+      }
+      request.errback {|http|
+        log_request(http, "Long poll timed-out. Repolling.", 'yellow')
+      }
+      
+      
+      # get url do |response|
+      #   repoll = Struct.new(:url).new(url)
+      #   block.call(response, repoll)
+      #   response.finished {
+      #     logger.info "Repolling".yellow
+      #     long_poll(repoll.url, &block)
+      #   }
+      # end
+    end
+
+    # HTTP requests
+    %w[get put post delete head].each do |http_meth|
+      eval %{
+        def #{http_meth}(*args, &block)
+          request(:#{http_meth}, *args, &block)
+        end}
+    end
+
+  private
+    def log_request(http, extra=nil, color=nil)
+      status = http.response_header.status.to_s
+      if color
+        status = status.send(color)
+      else
+        case status
+        when 500
+          status = status.red
+        when 0, 400..499
+          status = status.yellow
+        else
+          status = status.green
+        end
+      end
+      logger.info "#{status.ljust(3)} #{http.method.ljust(4)} #{http.uri}#{" (#{extra})".magenta if extra}\n  #{http.response}\n"
+    end
+  
+    # Simplifies em-http-client into something more suitable for tests. Interpets JSON, etc.
+    def request(http_meth, url, opts={}, &block)
+      response  = Response.new
+      request   = EM::HttpRequest.new(url).send(http_meth, opts)
+
+      request.callback {|http|
+        persist_cookie(http)
+        block.call(response) if block_given?
+        response.call(http)
+        log_request(http)
+      }
+      request.errback {|http|
+        log_request(http, "Network error or timeout.")
+      }
+    end
+
+    # Maintains session for mock users
+    def persist_cookie(http)
+      self.cookie = http.response_header['SET_COOKIE'] if http.response_header.include?('SET_COOKIE')
+    end
+  end
+end

data/lib/computron/dsl.rb

@@ -0,0 +1,47 @@
+module Computron
+  class DSL
+
+    attr_reader :client
+
+    def initialize(code=nil, &block)
+      @client     = Client.new
+      @simulation = Proc.new { code ? eval(code, self.send(:binding)) : instance_eval(&block) }
+    end
+
+    def run!
+      EM::run { @simulation.call }
+    end
+
+    def halt!
+      EM::stop
+    end
+
+    def default_host(host=nil)
+      host ? @default_host = host : @default_host
+    end
+
+    def every(seconds, opts={}, &block)
+      timer = EventMachine::PeriodicTimer.new(0) do
+        timer.interval = seconds
+        block.call(timer)
+      end
+    end
+
+    # HTTP utility methods. These are not meant to behave like a client
+    %w[get put post delete head].each do |http_meth|
+      eval %{
+        def #{http_meth}(*args, &block)
+          client.send(:#{http_meth}, *args, &block)
+        end}
+    end
+
+    # Deal with paths and URLs
+    def url(path)
+      url = String.new
+      url.concat "http://#{default_host}" unless url =~ %r{^(\w+)://} if default_host
+      url.concat path =~ /^\// ? path : "/#{path}" # Deal with leading /\
+      url
+    end
+
+  end
+end

data/lib/computron/report.rb

@@ -0,0 +1,48 @@
+# TODO integrate this puppy into the runner
+module Computron
+  class Report
+
+    class Request
+      
+      class Response
+        attr_accessor :headers, :status, :at
+      end
+
+      attr_accessor :uri, :response, :headers, :at, :method
+
+      def response
+        yield @response if block_given?
+        @response
+      end
+
+      def duration
+        @duration ||= response.at.to_f - at.to_f if at and response.at
+      end
+
+      def initialize
+        @response = Response.new
+        yield self if block_given?
+      end
+      
+    end
+
+    attr_reader :requests
+
+    def initialize
+      @requests = []
+    end
+
+    def by_status
+      requests.inject({}) do |memo, request|
+        memo[request.response.status] ||= []
+        memo[request.response.status] << request
+        memo
+      end
+    end
+
+    def sample(&block)
+      requests << Request.new(&block)
+    end
+
+  end  
+end

data/lib/computron.rb

@@ -0,0 +1,5 @@
+$stdout.sync = true
+
+require 'computron/client'
+require 'computron/dsl'
+require 'computron/report'

data/spec/computron_spec.rb

@@ -0,0 +1,7 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "Computron" do
+  it "fails" do
+    fail "hey buddy, you should probably rename this file and start specing for real"
+  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 'computron'
+require 'spec'
+require 'spec/autorun'
+
+Spec::Runner.configure do |config|
+  
+end

data/test.rb

@@ -0,0 +1,34 @@
+require '/Users/bgessler/Projects/imprezo/web/config/environment.rb'
+require 'lib/computron'
+
+clients_count = 2
+post_interval = 10
+
+clients = clients_count.times.map{ Computron::Client.new }
+
+default_host 'localhost:3000'
+
+get url('/events/1.json') do |response|
+  response.ok {|json|
+    event_id = json['event']['id']
+    ticks_url = url("/events/#{event_id}/ticks/stream.json")
+
+    clients.each do |client|
+      client.long_poll ticks_url do |response, repoll|
+        response.ok {|ticks|
+          if last_tick = ticks.last
+            repoll.url = "#{ticks_url}?since_id=#{last_tick['routing']['id']}"
+          end
+        }
+      end
+
+      every post_interval do
+        client.post url("/events/#{event_id}/comments"), :body => Comment.new(:response => 'Hey!').to_json, :head => {'Content-Type' => 'application/json'} do |response|
+          # response.ok             { p "Super fun" }
+          # response.found          { p "Redirected" }
+          # response.code(405)      { p "Method not supported jagoff" }
+        end
+      end
+    end
+  }
+end

metadata

@@ -0,0 +1,147 @@
+--- !ruby/object:Gem::Specification 
+name: computron
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Brad Gessler
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-05-24 00:00:00 -07:00
+default_executable: computron
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: eventmachine
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        - 12
+        - 10
+        version: 0.12.10
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: em-http-request
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        - 2
+        - 7
+        version: 0.2.7
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: colored
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 1
+        - 1
+        version: "1.1"
+  type: :runtime
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 1
+        - 2
+        - 9
+        version: 1.2.9
+  type: :development
+  version_requirements: *id004
+- !ruby/object:Gem::Dependency 
+  name: yard
+  prerelease: false
+  requirement: &id005 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id005
+description: Computron was created to help Poll Everywhere peform load tests that simulate how real users work.
+email: brad@bradgessler.com
+executables: 
+- computron
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .bundle/config
+- .document
+- .gitignore
+- Gemfile
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- bin/computron
+- computron.gemspec
+- lib/computron.rb
+- lib/computron/client.rb
+- lib/computron/dsl.rb
+- lib/computron/report.rb
+- spec/computron_spec.rb
+- spec/spec.opts
+- spec/spec_helper.rb
+- test.rb
+has_rdoc: true
+homepage: http://github.com/bradgessler/computron
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: An evented HTTP computron
+test_files: 
+- spec/computron_spec.rb
+- spec/spec_helper.rb