stella 0.6.0 → 0.7.0.002

data/lib/stella/crypto.rb

@@ -1,88 +0,0 @@
-require 'base64'  
-
-if RUBY_PLATFORM !~ /java/
-  require 'openssl'
-else
-  module JRuby #:nodoc:
-     module OpenSSL #:nodoc:
-       GEM_ONLY = false unless defined?(GEM_ONLY)
-     end
-   end
-  
-   if JRuby::OpenSSL::GEM_ONLY
-     require 'jruby/openssl/gem'
-   else
-     module OpenSSL #:nodoc:all
-       class OpenSSLError < StandardError; end
-       # These require the gem
-       %w[
-       ASN1
-       BN
-       Cipher
-       Config
-       Netscape
-       PKCS7
-       PKey
-       Random
-       SSL
-       X509
-       ].each {|c| autoload c, "jruby/openssl/gem"}
-     end
-     require "jruby/openssl/builtin"
-   end
-end
-
-# A small collection of helper methods for dealing with RSA keys. 
-module Stella::Crypto
-  VERSION = 1.0
-  
-  def self.create_keys(bits = 2048)
-    pk = OpenSSL::PKey::RSA.new(bits)
-  end
-  
-  @@digest = OpenSSL::Digest::Digest.new("sha1")
-  def self.sign(secret, string)
-    sig = OpenSSL::HMAC.hexdigest(@@digest, secret, string).strip
-    #sig.gsub(/\+/, "%2b")
-  end
-  def self.aws_sign(secret, string)
-    Base64.encode64(self.sign(secret, string)).strip
-  end
-  
-  # A class which represents an RSA or DSA key. 
-  class Key
-    attr_reader :data, :key
-    
-    # Create an instance of Crypto::Key with the provided rsa or dsa 
-    # public or private key data. 
-    def initialize(data)
-      @data = data 
-      @public = (data =~ /^-----BEGIN (RSA|DSA) PRIVATE KEY-----$/).nil?
-      @key = OpenSSL::PKey::RSA.new(@data)
-    end  
-    
-    # Create an instance of Crypto::Key using a key file. 
-    #   key = Crypto::Key.from_file('path/2/id_rsa')
-    def self.from_file(filename)    
-      self.new File.read( filename )
-    end
-    
-    # Encrypt and base64 encode the given text.
-    def encrypt(text)
-      Base64.encode64(@key.send("#{type}_encrypt", text))
-    end
-    
-    # Decrypt the given base64 encoded text. 
-    def decrypt(text)
-      @key.send("#{type}_decrypt", Base64.decode64(text))
-    end
-  
-    def private?();  !@public; end
-    def public?();  @public;  end 
-    
-    def type
-      @public ? :public : :private
-    end
-  end
-end
-

data/lib/stella/data/domain.rb

@@ -1,82 +0,0 @@
-
-
-module Stella::Data
-  class DomainRequest < Storable
-    attr_accessor :dns_data
-    attr_reader :raw_data
-    
-    field :time => DateTime
-    field :client_ip => String
-    field :server_ip => String
-    field :domain_name => String
-    field :header => String
-    
-    def initialize(raw_data)
-      @raw_data = raw_data
-      @dns_data, @domain_name, @header = DomainUtil::parse_domain_request(@raw_data)
-    end
-    
-    def has_request?
-      false
-    end
-    def has_response?
-      false
-    end
-    def has_body?
-      false
-    end
-    
-    def to_s
-      "%s: %s -> %s (%s)" % [@time, @client_ip, @server_ip, @domain_name]
-    end
-    
-    def inspect
-      str = "#{$/};; REQUEST         #{@time.to_s}"
-      str << "#{$/};; %s %s> %s" % [@client_ip, '-'*30, @server_ip]
-      str << "#{$/};;#{$/}"
-      str << @dns_data.inspect
-      str
-    end
-    
-  end
-  
-  class DomainResponse < Storable
-    attr_accessor :dns_data
-    attr_reader :raw_data
-    
-    field :time => DateTime
-    field :client_ip => String
-    field :server_ip => String
-    field :domain_name => String
-    field :header => String
-    field :addresses => Array
-    field :cnames => Array
- 
- 
-    def initialize(raw_data)
-      @raw_data = raw_data
-      @dns_data, @domain_name, @header, @addresses, @cnames = DomainUtil::parse_domain_response(@raw_data)
-    end
-    
-    def has_request?
-      false
-    end
-    def has_response?
-      false
-    end
-    def has_body?
-      false
-    end
-    
-    def to_s
-      "%s: %s <- %s (%s) %s" % [@time, @client_ip, @server_ip, @domain_name, (@addresses || []).join(',')]
-    end
-    
-    def inspect
-      str =  "#{$/};; RESPONSE        #{@time.strftime(NICE_TIME_FORMAT)}"
-      str << "#{$/};; %s <%s %s" % [@client_ip, '-'*30, @server_ip]
-      str << "#{$/};;#{$/}"
-      str << @dns_data.inspect
-    end
-  end
-end

data/lib/stella/environment.rb

@@ -1,66 +0,0 @@
-module Stella
-  class Environment
-    
-    
-  end
-end
-
-module Stella
-  class Environment
-    
-    attr_accessor :name
-    
-      # An array of `Stella::Common::Machine objects to be use during the test.
-      #     @stella_environments.machines << "stellaaahhhh.com:80"
-    attr_accessor :machines
-      # The default proxy, a Stella::Common::Proxy object containing the proxy to be used for the test.
-    attr_accessor :proxy
-    
-    def initialize(name=:development)
-      @name = name
-      @machines = []
-    end
-
-    
-    def add_machines(*args)
-      return if args.empty?
-      args.each do |machine|
-        @machines << Stella::Common::Machine.new(machine)
-      end
-    end
-    
-    # Creates a Stella::TestPlan::Proxy object and stores it to +@proxy+
-    def proxy=(*args)
-      uri, user, pass = args.flatten
-      @proxy = Stella::Common::Proxy.new(uri, user, pass)
-    end
-    
-    
-  end
-end
-
-module Stella
-  module DSL
-    module Environment
-      attr_accessor :stella_current_environment
-
-      def environments
-        @stella_environments 
-      end
-    
-      def environment(name, &define)
-        @stella_environments ||= {}
-        @stella_current_environment = @stella_environments[name] = Stella::Environment.new(name)
-        define.call if define
-      end
-      
-      def machines(*args)
-        return unless @stella_current_environment.is_a? Stella::Environment
-        args.each do |machine|
-          @stella_current_environment.add_machines machine
-        end
-      end
-      
-    end
-  end
-end

data/lib/stella/functest.rb

@@ -1,105 +0,0 @@
-
-
-
-module Stella
-  class FunctionalTest
-    include TestRunner
-    
-    def type; "functional"; end
-    
-    
-    
-    def update_start(machine, name)
-      puts '-'*60
-      puts "%10s: %s" % ["MACHINE", machine]  
-    end
-    
-    def update_authorized(domain, user, pass)
-      note = user
-      note += ":****" if pass
-      puts "%10s: %s" % ["user", note]
-      puts
-    end
-    
-    def update_request(method, uri, query, response_status, response_headers, response_body)
-      puts "#{method} #{uri}"
-      puts "%18s: %s" % ["status", response_status]
-      puts "%18s: %s" % ["query", query] if @verbose > 0 && !query.empty?
-      puts "%18s: %s" % ["response_headers", response_headers]
-      puts "%18s: #{$/}%s" % ["response_body", response_body[0..100]] if @verbose > 0
-      puts
-    end
-    
-    def update_request_exception(method, uri, query, ex)
-      puts "#{method} #{uri}"
-      puts "EXCEPTION: #{ex.message}"
-      puts ex.backtrace
-    end
-    
-    def update_request_unexpected_response(method, uri, query, response_status, response_headers, response_body)
-      puts "#{method} #{uri}"
-      puts "%18s: %s" % ["status", response_status]
-      puts "%18s: %s" % ["note", "unexpected response status"]
-      puts "", response_body[0..500]
-      puts '...' if response_body.length >= 500
-    end
-    
-    def update_retrying(uri, retry_count, total)
-      puts "retrying: #{uri} (#{retry_count} of #{total})"
-    end
-
-
-    # +environment+ is a Stella::Common::Environment object. 
-    # +namespace+ is a reference to the namespace which contains the instance
-    # variables. This will be the section of code that makes use of the DSL.
-    def run(environment, namespace)
-      raise "No testplan defined" unless @testplan
-      raise "No machines defined for #{environment.name}" if environment.machines.empty?
-      
-      
-      begin
-        if environment.proxy
-          http_client = HTTPClient.new(environment.proxy.uri)
-          http_client.set_proxy_auth(environment.proxy.user, environment.proxy.pass) if environment.proxy.user
-        else
-          http_client = HTTPClient.new
-        end
-      rescue => ex
-        puts ex.class
-      end
-      
-      request_stats = {}
-      environment.machines.each do |machine|
-        client = Stella::Client.new
-        client.add_observer(self)
-        client.execute_testplan(request_stats, http_client, machine, namespace, @testplan, @verbose)
-      end
-      
-
-      request_stats.each do |rstat|
-        puts rstat[1][:stats].to_s
-      end
-    end
-    
-
-  end
-end
-
-
-
-
-module Stella
-    module DSL 
-      module FunctionalTest
-        include Stella::DSL::TestRunner
-      
-      def functest(name=:default, &define)
-        @tests ||= {}
-        @current_test = @tests[name] = Stella::FunctionalTest.new(name)
-        define.call if define
-      end
-      
-      
-    end
-  end
-end

data/lib/stella/loadtest.rb

@@ -1,186 +0,0 @@
-# See, re Threadify on JRuby: http://www.ruby-forum.com/topic/158180
-
-#
-#
-module Stella
-  class LoadTest
-    include TestRunner
-
-    attr_accessor :clients
-    attr_accessor :repetitions
-    attr_accessor :duration
-    
-    attr_reader :testplans_started
-    attr_reader :testplans_completed
-    
-    attr_reader :requests_successful
-    attr_reader :requests_failed
-    
-    def init
-      @repetitions = 1
-      @clients = 1
-      @duration = 0
-      reset
-    end
-    
-    def reset
-      @testplans_started = 0
-      @testplans_completed = 0
-      @requests_successful = 0
-      @requests_failed = 0
-    end
-    
-    def type; "load"; end
-    
-    def requests_total
-      @requests_successful + @requests_failed
-    end
-    
-    def update_start(machine, name)
-      @testplans_started += 1
-    end
-    
-    def update_done(*args)
-      @testplans_completed += 1
-    end
-    
-    def update_authorized(domain, user, pass)
-    end
-    
-    def update_request(method, uri, query, response_status, response_headers, response_body)
-      @requests_successful += 1
-    end
-    
-    def update_request_exception(method, uri, query, ex)
-      @requests_failed += 1
-      puts [method, uri, query, ex.message].join("|")
-    end
-    
-    def update_request_unexpected_response(method, uri, query, response_status, response_headers, response_body)
-      @requests_failed += 1
-      #puts [method, uri, query, response_status, response_headers, response_body].join("|")
-    end
-    
-    def update_retrying(uri, retry_count, total)
-      #print retry_count
-    end
-    
-    
-    # +environment+ is a Stella::Common::Environment object. 
-    # +namespace+ is a reference to the namespace which contains the instance
-    # variables. 
-    def run(environment, namespace)
-      raise "No testplan defined" unless @testplan
-      raise "No machines defined for #{environment.name}" if environment.machines.empty?
-
-      [:duration, :clients, :repetitions].each do |p|
-        val = instance_variable_get("@#{p}")
-        puts " %11s: %s" % [p, val] if val
-      end
-      
-      stats = Stats.new("LoadTest")
-      request_stats = {}
-
-      time_started = Time.now
-      seconds_elapsed = 0
-      (1..@clients).to_a.threadify do |i|
-        
-        (0..@repetitions).to_a.each do |rep|
-
-          if environment.proxy
-            http_client = HTTPClient.new(environment.proxy.uri)
-            http_client.set_proxy_auth(environment.proxy.user, environment.proxy.pass) if environment.proxy.user
-          else
-            http_client = HTTPClient.new
-          end
-
-
-          environment.machines.each do |machine|
-            client = Stella::Client.new(i)
-            client.add_observer(self)
-            client.execute_testplan(request_stats, http_client, machine, namespace, @testplan, @verbose)
-          end
-          
-          seconds_elapsed = Time.now - time_started
-
-          #request_stats.each do |rstat|
-          #  puts rstat[1][:stats].to_s
-          #end
-          
-          # If a duration was given, we make sure to run for that
-          # amount of time.
-          if @duration > 0
-            redo if seconds_elapsed <= @duration 
-            break if seconds_elapsed >= @duration
-          end
-        end
-      end
-
-      stats.tick
-      puts "DONE! (#{seconds_elapsed.minutes} minutes)"
-      instance_variables.each do |name|
-        #next unless name =~ /request/
-        puts "%20s: %s" % [name, instance_variable_get(name)]
-      end
-      
-      puts "Final Status"
-      puts stats.to_s
-      puts
-
-      request_stats.each do |rstat|
-        puts rstat[1][:stats].to_s
-      end
-    end
-    
-    
-    def clients=(*args)
-      count = args.flatten.first
-      @clients = count
-    end
-    
-    def repetitions=(*args)
-      @repetitions = args.flatten.first
-    end
-    
-    def duration=(*args)
-      @duration = args.flatten.first
-    end
-  end
-end
-
-
-
-
-
-
-
-module Stella
-  module DSL 
-    module LoadTest
-      include Stella::DSL::TestRunner
-      
-      def loadtest(name=:default, &define)
-        @tests ||= {}
-        @current_test = @tests[name] = Stella::LoadTest.new(name)
-        define.call if define
-      end
-      
-      def rampup(*args)
-      end 
-      
-      def warmup(*args)
-      end
-          
-      [:repetitions, :duration, :clients].each do |method_name|
-        eval <<-RUBY, binding, '(Stella::DSL::LoadTest)', 1
-        def #{method_name}(*val)
-          return unless @current_test.is_a? Stella::LoadTest
-          @current_test.#{method_name}=(val)
-        end
-        private :#{method_name}
-        RUBY
-      end
-      
-    end
-  end
-end

data/lib/stella/testrunner.rb

@@ -1,64 +0,0 @@
-# ---
-# See: http://codeforpeople.com/lib/ruby/flow/flow-2.0.0/sample/a.rb
-# +++
-
-#
-#
-#
-module Stella
-  module TestRunner
-    attr_accessor :name
-      # Name or instance of the testplan to execute
-    attr_accessor :testplan
-      # Determines the amount of output. Default: 0
-    attr_accessor :verbose
-    
-    def initialize(name=:default)
-      @name = name
-      @verbose = 0
-      init if respond_to? :init
-    end
-    
-    def update(*args)
-      what, *args = args
-      self.send("update_#{what}", *args) if respond_to? "update_#{what}"
-    end
-    
-  end
-  module DSL
-    module TestRunner
-      attr_accessor :current_test
-      
-      def plan(testplan)
-        raise "Unknown testplan, '#{testplan}'" unless @plans.has_key?(testplan)
-        return unless @current_test
-        @current_test.testplan = @plans[testplan]
-      end
-      
-      def run(env_name=nil, test_name=nil)
-        to_run = test_name.nil? ? @tests : [@tests[test_name]]
-        env = env_name.nil? ? @stella_environments.first : @stella_environments[env_name]
-        to_run.each do |t|
-          puts '='*60
-          puts "RUNNING TEST: #{test_name}"
-          puts " %11s: %s" % ['type', t.type]
-          puts " %11s: %s" % ['testplan', t.testplan.name]
-          puts " %11s: %s" % ['desc', t.testplan.description]
-          puts " %11s: %s" % ['env', env_name]
-           
-          
-          t.run(env, self)
-        end
-      end
-      
-      def verbose(*args)
-        @current_test.verbose += args.first || 1
-      end
-      
-    private
-      
-    end
-  end
-end
-  
-