solutious-stella 0.6.0 → 0.7.0.001

data/lib/stella/utils.rb

@@ -0,0 +1,126 @@
+
+require 'socket'
+require 'open-uri'
+require 'date'
+
+require 'timeout'
+
+module Stella
+  
+  # A motley collection of methods that Stella loves to call!
+  module Utils
+    extend self
+    include Socket::Constants
+    
+    # Return the external IP address (the one seen by the internet)
+    def external_ip_address
+      ip = nil
+      begin
+        %w{solutious.heroku.com/ip}.each do |sponge|
+          ipstr = Net::HTTP.get(URI.parse("http://#{sponge}")) || ''
+          ip = /([0-9]{1,3}\.){3}[0-9]{1,3}/.match(ipstr).to_s
+          break if ip && !ip.empty?
+        end
+      rescue SocketError, Errno::ETIMEDOUT => ex
+        Stella.le "Connection Error. Check your internets!"
+      end
+      ip
+    end
+    
+    # Return the local IP address which receives external traffic
+    # from: http://coderrr.wordpress.com/2008/05/28/get-your-local-ip-address/
+    # NOTE: This <em>does not</em> open a connection to the IP address. 
+    def internal_ip_address
+      # turn off reverse DNS resolution temporarily 
+      orig, Socket.do_not_reverse_lookup = Socket.do_not_reverse_lookup, true   
+      ip = UDPSocket.open {|s| s.connect('75.101.137.7', 1); s.addr.last } # Solutious IP
+      ip
+    ensure  
+      Socket.do_not_reverse_lookup = orig
+    end
+    
+    
+    # <tt>require</tt> a glob of files. 
+    # * +path+ is a list of path elements which is sent to File.join 
+    # and then to Dir.glob. The list of files found are sent to require. 
+    # Nothing is returned but LoadError exceptions are caught. The message
+    # is printed to STDERR and the program exits with 7. 
+    def require_glob(*path)
+      begin
+        Dir.glob(File.join(*path.flatten)).each do |path|
+          require path
+        end
+      rescue LoadError => ex
+        puts "Error: #{ex.message}"
+        exit 7
+      end
+    end
+    
+    # Checks whether something is listening to a socket. 
+    # * +host+ A hostname
+    # * +port+ The port to check
+    # * +wait+ The number of seconds to wait for before timing out. 
+    #
+    # Returns true if +host+ allows a socket connection on +port+. 
+    # Returns false if one of the following exceptions is raised:
+    # Errno::EAFNOSUPPORT, Errno::ECONNREFUSED, SocketError, Timeout::Error
+    #
+    def service_available?(host, port, wait=3)
+      if Stella.sysinfo.vm == :java
+        begin
+          iadd = Java::InetSocketAddress.new host, port      
+          socket = Java::Socket.new
+          socket.connect iadd, wait * 1000  # milliseconds
+          success = !socket.isClosed && socket.isConnected
+        rescue NativeException => ex
+          puts ex.message, ex.backtrace if Stella.debug?
+          false
+        end
+      else 
+        begin
+          status = Timeout::timeout(wait) do
+            socket = Socket.new( AF_INET, SOCK_STREAM, 0 )
+            sockaddr = Socket.pack_sockaddr_in( port, host )
+            socket.connect( sockaddr )
+          end
+          true
+        rescue Errno::EAFNOSUPPORT, Errno::ECONNREFUSED, SocketError, Timeout::Error => ex
+          puts ex.class, ex.message, ex.backtrace if Stella.debug?
+          false
+        end
+      end
+    end
+    
+    # A basic file writer
+    def write_to_file(filename, content, mode, chmod=600)
+      mode = (mode == :append) ? 'a' : 'w'
+      f = File.open(filename,mode)
+      f.puts content
+      f.close
+      return unless Stella.sysinfo.os == :unix
+      raise "Provided chmod is not a Fixnum (#{chmod})" unless chmod.is_a?(Fixnum)
+      File.chmod(chmod, filename)
+    end
+
+    # 
+    # Generates a string of random alphanumeric characters.
+    # * +len+ is the length, an Integer. Default: 8
+    # * +safe+ in safe-mode, ambiguous characters are removed (default: true):
+    #       i l o 1 0
+    def strand( len=8, safe=true )
+       chars = ("a".."z").to_a + ("0".."9").to_a
+       chars.delete_if { |v| %w(i l o 1 0).member?(v) } if safe
+       str = ""
+       1.upto(len) { |i| str << chars[rand(chars.size-1)] }
+       str
+    end
+    
+    # Returns +str+ with the leading indentation removed. 
+    # Stolen from http://github.com/mynyml/unindent/ because it was better.
+    def noindent(str)
+      indent = str.split($/).each {|line| !line.strip.empty? }.map {|line| line.index(/[^\s]/) }.compact.min
+      str.gsub(/^[[:blank:]]{#{indent}}/, '')
+    end
+    
+  end
+end

data/lib/stella/version.rb

@@ -0,0 +1,15 @@
+
+
+module Stella
+  module Version
+    unless defined?(MAJOR)
+      MAJOR = 0.freeze
+      MINOR = 7.freeze
+      TINY  = 0.freeze
+      PATCH = '001'.freeze
+    end
+    def self.to_s; [MAJOR, MINOR, TINY].join('.'); end
+    def self.to_f; self.to_s.to_f; end
+    def self.patch; PATCH; end
+  end
+end

data/lib/threadify.rb

@@ -137,12 +137,6 @@ class Thread
   end
 end
 
-class Object
-  def threadify! *values
-    throw :threadify, *values
-  end
-end
-
 
 if __FILE__ == $0
   require 'open-uri'

data/stella.gemspec

@@ -1,69 +1,63 @@
 @spec = Gem::Specification.new do |s|
-	s.name = "stella"
-  s.rubyforge_project = "stella"
-	s.version = "0.6.0"
-	s.summary = "Your friend in performance testing"
-	s.description = s.summary
-	s.author = "Delano Mandelbaum"
-	s.email = "delano@solutious.com"
-	s.homepage = "http://github.com/solutious/stella"
+  s.name = "stella"
+  s.rubyforge_project = 'stella'
+  s.version = "0.7.0.001"
+  s.summary = "Stella: Your friend in performance testing."
+  s.description = s.summary
+  s.author = "Delano Mandelbaum"
+  s.email = "delano@solutious.com"
+  s.homepage = "http://solutious.com/projects/stella/"
   
-  # = DEPENDENCIES =
-  # Add all gem dependencies
-  s.add_dependency 'httpclient'
+  s.extra_rdoc_files = %w[README.rdoc LICENSE.txt CHANGES.txt]
+  s.has_rdoc = true
+  s.rdoc_options = ["--line-numbers", "--title", s.summary, "--main", "README.rdoc"]
+  s.require_paths = %w[lib]
   
+  s.executables = %w[stella]
+  
+  s.add_dependency 'drydock',    '>= 0.6.7'
+  s.add_dependency 'gibbler',    '>= 0.6.2'
+  s.add_dependency 'storable',   '>= 0.5.7'
+  s.add_dependency 'httpclient', '>= 2.1.5'
   
   # = MANIFEST =
-  # The complete list of files to be included in the release. When GitHub packages your gem, 
-  # it doesn't allow you to run any command that accesses the filesystem. You will get an
-  # error. You can ask your VCS for the list of versioned files:
   # git ls-files
-  # svn list -R
   s.files = %w(
   CHANGES.txt
   LICENSE.txt
   README.rdoc
   Rakefile
-  bin/example_test.rb
-  bin/example_webapp.rb
-  lib/logger.rb
+  bin/stella
+  examples/basic/listing_ids.csv
+  examples/basic/plan.rb
   lib/stella.rb
-  lib/stella/clients.rb
-  lib/stella/command/base.rb
-  lib/stella/command/form.rb
-  lib/stella/command/get.rb
-  lib/stella/common.rb
-  lib/stella/crypto.rb
-  lib/stella/data/domain.rb
+  lib/stella/cli.rb
+  lib/stella/client.rb
+  lib/stella/config.rb
+  lib/stella/data.rb
   lib/stella/data/http.rb
-  lib/stella/environment.rb
-  lib/stella/functest.rb
-  lib/stella/loadtest.rb
+  lib/stella/data/http/body.rb
+  lib/stella/data/http/request.rb
+  lib/stella/data/http/response.rb
+  lib/stella/dsl.rb
+  lib/stella/engine.rb
+  lib/stella/engine/functional.rb
+  lib/stella/engine/load.rb
+  lib/stella/exceptions.rb
+  lib/stella/guidelines.rb
+  lib/stella/mixins.rb
   lib/stella/stats.rb
   lib/stella/testplan.rb
-  lib/stella/testrunner.rb
-  lib/storable.rb
+  lib/stella/testplan/stats.rb
+  lib/stella/testplan/usecase.rb
+  lib/stella/utils.rb
+  lib/stella/utils/httputil.rb
+  lib/stella/version.rb
   lib/threadify.rb
-  lib/timeunits.rb
-  lib/util/httputil.rb
   stella.gemspec
-  tryouts/drb/drb_test.rb
-  tryouts/drb/open4.rb
-  tryouts/drb/slave.rb
-  tryouts/oo_tryout.rb
+  support/example_webapp.rb
+  support/useragents.txt
   )
-  
-  # = EXECUTABLES =
-  # The list of executables in your project (if any). Don't include the path, 
-  # just the base filename.
-  s.executables = %w[]
-  
-  s.extra_rdoc_files = %w[README.rdoc LICENSE.txt]
-  s.rdoc_options = ["--line-numbers", "--title", s.summary, "--main", "README.rdoc"]
-  
-  s.has_rdoc = true
-  s.require_paths = %w[lib]
-  s.rubygems_version = '1.1.1'
 
   
-end
+end

data/support/example_webapp.rb

@@ -0,0 +1,246 @@
+#!/usr/bin/ruby
+
+# Use Ruby 1.8
+
+require "rubygems"
+require "rack"
+require "sinatra"
+
+require 'yaml'
+
+set :run => true
+set :environment => :development
+set :dump_errors => true
+set :port => 3114
+set :reload => true
+set :max_listings => 1000
+
+#log = File.new("/dev/null", "a")
+#STDOUT.reopen(log)
+#STDERR.reopen(log)
+
+#use Rack::Auth::Basic do |username, password|
+#  username == 'stella' && password == 'stella'
+#end
+
+before do
+  @title = "Business Finder"
+end
+
+get '/' do
+  @title << " - Search"
+  erb :search_form
+end
+
+
+get '/search/?' do
+  redirect '/' if blank?(params[:what]) && blank?(params[:where])
+  params[:what] ||= ''
+  params[:where] ||= ''
+  @title << " - Search Results"
+  @listings = filter_name(params[:what], options.listings)
+  if !blank?(params[:where])
+    @listings = filter_city(params[:where], @listings)
+  end
+  if @listings.empty?
+    status 404
+    erb :search_error 
+  else
+    erb :search_results
+  end
+end
+
+get '/listing/add' do
+  erb :add_form
+end
+
+post '/listing/add' do
+  @title = "Add a Business"
+  if blank?(params[:name]) || blank?(params[:city])
+    status 500
+    @msg = blank?(params[:city]) ? "Must specify city" : "Must specify name"
+    erb :add_form
+  else
+    @listings = options.listings
+    if find_name(params[:name], @listings).empty?
+      @listings.shift if @listings.size >= options.max_listings
+      @listings << { :name => params[:name], :id => rand(100000), :city => params[:city] }
+      redirect '/listings'
+    else
+      status 500
+      @msg = "That business exists (#{params[:name]})"
+      erb :add_form
+    end
+  end
+end
+
+get '/listing/:id.yaml' do 
+  content_type "text/yaml"
+  listing = filter_id params[:id], options.listings
+  listing.to_yaml
+end
+
+get '/listing/:id' do 
+  @listings = filter_id(params[:id], options.listings)
+  redirect '/' if @listings.empty?
+  @title = "Business Listing - #{@listings.first[:name]}"
+  erb :listings
+end
+
+get '/listings' do
+  @listings = options.listings
+  @title = "Business Listings"
+  erb :listings
+end
+
+get '/listings.yaml' do
+  content_type "text/yaml"
+  @listings = options.listings
+  @title = "Business Listings"
+  @listings.to_yaml
+end
+
+set :listings => [
+  { :id => 1000, :name => 'John West Smoked Oysters', :city => 'Toronto'  },
+  { :id => 1001, :name => 'Fire Town Lightning Rods', :city => 'Toronto'  },
+  { :id => 1002, :name => 'Oversized Pen and Ink Co', :city => 'Toronto'  },
+  { :id => 1003, :name => 'The Rathzenburg Brothers', :city => 'Toronto'  },
+  { :id => 1004, :name => 'Forever and Always Beads', :city => 'Montreal' },
+  { :id => 1005, :name => "Big Al's Flavour Country", :city => 'Montreal' },
+  { :id => 1006, :name => 'Big Time Furniture World', :city => 'Montreal' },
+  { :id => 1007, :name => 'High-End Keyboard Makers', :city => 'Montreal' }
+]
+
+before do
+  @cookie = request.cookies["bff-history"]
+  @cookie = blank?(@cookie) ? {} : YAML.load(@cookie)
+  @cookie[:history] ||= []
+  if params[:clear] == 'true'
+    @cookie[:history] = []
+    @cookie[:location] = ''
+  end
+  @cookie[:history].delete params[:what]
+  @cookie[:history].unshift params[:what] unless blank?(params[:what])
+  @cookie[:history].pop if @cookie[:history].size > 5
+  @cookie[:location] = params[:where] unless blank?(params[:where])
+  response.set_cookie "bff-history", :path => '/', :value => @cookie.to_yaml
+end
+
+helpers do
+  
+  def blank?(v)
+    v.nil? || v.empty?
+  end
+  
+  def filter_id(id, listings)
+    listings.select { |l| l[:id] == id.to_i }
+  end
+  
+  def filter_name(name, listings)
+    listings.select { |l| l[:name].match(/#{name}/i) }
+  end
+  
+  def find_name(name, listings)
+    listings.select { |l| l[:name] == name }
+  end
+  
+  def filter_city(city, listings)
+    listings.select { |l| l[:city].match(/#{city}/i) }
+  end
+  
+  def format_listing(lid, name, city)
+    listing = %Q{<div class="listing" id="listing-#{lid}">}
+    listing << %Q{<a href="/listing/#{lid}.yaml">#{name}</a> }
+    listing << %Q{#{city}</div>}
+  end
+  
+  # Generates a string of random alphanumeric characters
+  # These are used as IDs throughout the system
+  def strand( len )
+     chars = ("a".."z").to_a + ("0".."9").to_a
+     newpass = ""
+     1.upto(len) { |i| newpass << chars[rand(chars.size-1)] }
+     return newpass
+  end
+  
+end
+
+__END__
+
+@@layout
+<html>
+<head>
+<title><%= @title %></title>
+<style>
+.hilite { background-color: #FEE00B; font-weight: bold; }
+.footer { color: #ccc; font-weight: lighter; font-size: 80%; margin-top: 30px; }
+.footer a { color: #69c;}
+</style>
+</head>
+<body>
+<h1>Business Finder</h1>
+<p style="margin-left: 50px; margin-top: 20px;"><em>
+<a href="/">New Search</a> -
+<a href="/listing/add?name=<%= params[:what] %>&amp;city=<%= params[:where] %>">Add Listing</a> - 
+<a href="/?clear=true">!!</a> -
+<a href="/listings">View All</a>
+</em></p>
+<%= yield %>
+<div class="footer">
+A <a href="http://solutious.com/projects/stella/">Stella</a> demo by <a href="http://solutious.com/">Solutious</a>. 
+</div>
+</body>
+</html>
+
+@@add_form
+<% city = blank?(params[:city]) ? 'Toronto' : params[:city] %>
+<% if !blank?(@msg) %>
+<p style="color: red"><em>Error: <%= @msg %></em></p>
+<% end %>
+<form method="post">
+Name: <input name="name" value="<%= params[:name] %>"/><br/>
+City: <input name="city" value="<%= city %>" /><br/>
+<input type="submit" />
+</form>
+
+@@listings
+<% for l in @listings %>
+  <%= format_listing(l[:id], l[:name], l[:city]) %>
+<% end %>
+
+@@search_form
+<form action="/search">
+What: <input name="what" value="" /><br/>
+Where: <input name="where" value="<%= @cookie[:location] %>" /><br/>
+<input type="submit" />
+</form>
+<ul id="history">
+  <% unless @cookie[:history].empty? %>
+  <h3>Previous Searches</h3>
+  <% end %>
+  <% for what in @cookie[:history] %>
+  <li><em><a href="/search/?what=<%= what %>&amp;where=<%= @cookie[:location] %>"><%= what %></a></em></li>
+  <% end %>
+</ul>
+
+@@search_results
+Looking 
+<% if !blank?(params[:what]) %>
+for "<b><%= params[:what] %></b>"
+<% end %>
+<% if !blank?(params[:where]) %>
+in "<b><%= params[:where] %></b>"
+<% end %>
+<br/><br/> 
+<% for l in @listings %>
+  <% name = l[:name].gsub(/(#{params[:what]})/i, "<span class='hilite'>\\1</span>") %>
+  <% city = l[:city].gsub(/(#{params[:where]})/i, "<span class='hilite'>\\1</span>") %>
+  <%= format_listing(l[:id], name, city) %>
+<% end %>
+
+@@search_error
+Looking for "<b><%= params[:what] %></b>" 
+<% if !blank?(params[:where]) %>
+in "<b><%= params[:where] %></b>"
+<% end %>
+<p><i>Nothing found</i></p>