solutious-stella 0.6.0 → 0.7.0.001

data/lib/stella/clients.rb

@@ -1,161 +0,0 @@
-
-require "observer"
-require "tempfile"
-
-
-module Stella
-  class Client
-    include Observable
-    
-    @@IDS = {}
-    
-    attr_reader :request_stats
-    attr_accessor :client_id
-    def initialize(client_id=1)
-      @client_id = client_id
-      @request_stats = {
-      }
-    end
-    
-    def execute_testplan(request_stats, http_client, machine, namespace, plan, verbose=1)
-      changed
-      notify_observers(:start, machine, plan.name)
-      
-      if plan.auth
-        auth_domain = "#{plan.protocol}://#{machine.to_s}/"
-        http_client.set_auth(auth_domain, plan.auth.user, plan.auth.pass)
-        changed
-        notify_observers(:authorized, auth_domain, plan.auth.user, plan.auth.pass)
-      end
-      
-      tf = Tempfile.new('stella-cookie')
-      http_client.set_cookie_store(tf.to_s)
-    
-      request_methods = namespace.methods.select { |meth| meth =~ /\d+\s[A-Z]/ }
-      
-      retry_count = 1
-      previous_methname = nil
-      request_methods.each do |methname|
-        retry_count = 1 unless previous_methname == methname
-        previous_methname = methname
-        
-        # We need to define the request only the first time it's run. 
-        req = namespace.send(methname) unless retry_count > 1 
-        req.set_unique_id(self.object_id)
-        
-        request_stats[req.stella_id.to_sym] ||= {
-          :name => req.name,
-          :stats => Stats.new( req.name )
-        }
-        
-        uri = req.uri.is_a?(URI) ? req.uri : URI.parse(req.uri.to_s)
-        uri.scheme ||= plan.protocol
-        uri.host ||= machine.host
-        uri.port ||= machine.port
-        
-        query = {}.merge!(req.params)
-        
-        
-        
-        begin
-          
-          if req.http_method =~ /POST|PUT/
-            
-            if req.body.has_content?
-              body = req.body.content
-              param = req.body.form_param || 'file'  # How do we handle bodies with no form name?
-              query[param] = body # NOTE: HTTPClient prefers a file handle rather than reading in the file
-            end
-
-          end
-          
-          # Make the request. 
-          time_started = Time.now
-          res = http_client.send(req.http_method.downcase, uri.to_s, query)
-          request_stats[req.stella_id.to_sym][:stats].sample(Time.now - time_started)
-          
-          id1 = (query || {})['id']
-          id2 = YAML.load(res.body.content || '')[:id]
-          if id1          
-            #puts "#{id1} (#{req.unique_id}): #{@@IDS[id1].join(', ')}" if @@IDS.has_key?(id1) && @@IDS[id1].size > 1
-            @@IDS[id1] ||= []
-            @@IDS[id1] << @client_id
-            @@IDS[id1].uniq!
-          end
-          
-        rescue => ex
-          changed
-          notify_observers(:request_exception, req.http_method, uri, query, ex)
-          next
-        end
-        
-        
-        response_headers = res.header.all.stella_to_hash
-        
-
-        
-        unless req.response_handler.has_key?(res.status)
-          changed
-          notify_observers(:request_unexpected_response, req.http_method, uri, query, res.status, response_headers, res.body.content)
-        else
-          
-          changed
-          notify_observers(:request, req.http_method, uri, query, res.status, response_headers, res.body.content)
-          
-          response_handler_ret = req.response_handler[res.status].call(response_headers, res.body.content, @client_id, req.unique_id)
-        
-          if response_handler_ret.is_a?(Stella::TestPlan::ResponseHandler) && response_handler_ret.action == :repeat
-            retry_count ||= 1
-          
-            if retry_count > response_handler_ret.times
-              retry_count = 1
-              next
-            else  
-              changed
-              notify_observers(:retrying, uri, retry_count, response_handler_ret.times)
-              run_sleeper(response_handler_ret.wait)
-              retry_count += 1
-              redo
-            end
-          end
-        end
-        
-      end
-      
-      http_client.save_cookie_store
-      changed
-      notify_observers(:done)
-      
-    end
-  
-    def run_sleeper(duration, quiet=true)
-      remainder = duration % 1 
-      duration.to_i.times {
-        print '.' unless duration <= 1 || quiet
-        sleep 1
-      }
-      sleep remainder if remainder > 0
-    end  
-  end
-end
-
-#module Stella
-#  class Clients
-#
-#      # The default authentication, a Stella::Common::Auth object
-#    attr_accessor :auth
-#    
-#  end
-#end
-
-#module Stella
-#  module DSL
-#    module Clients
-#      
-#      def clients(name = :anonymous, &define)
-#        @stella_clients ||= {}
-#        @stella_clients[name] = []
-#      end
-#    end
-#  end
-#end

data/lib/stella/command/base.rb

@@ -1,20 +0,0 @@
-
-      
-module Stella::Command
-  module Base
-
-    
-    def run
-      raise "Override 'run'"
-    end
-    
-    def run_sleeper(duration)
-      remainder = duration % 1 
-      duration.to_i.times {
-        Stella::LOGGER.info_print('.') unless @quiet
-        sleep 1
-      }
-      sleep remainder if remainder > 0
-    end
-  end
-end

data/lib/stella/command/form.rb

@@ -1,36 +0,0 @@
-
-require 'httpclient'
-
-require 'util/httputil'
-
-require 'stella/command/base'
-require 'stella/data/http'
-
-
-module Stella::Command #:nodoc: all
-  class Form < Drydock::Command #:nodoc: all
-    include Stella::Command::Base
-    
-    
-    
-  end
-end
-
-
-
-
-__END__
-
-headers = { 'Content-Type' => 'text/xml' }
-                       
-@client = HTTPClient.new('http://localhost:3114')
-@url = URI.parse "http://solutious.com/"
-#@url = URI.parse 'https://delaagsterekening.nl/api/suggestions/status.json?token=253'
-#@client.set_auth("https://delaagsterekening.nl/", "stella", "stella")
-@client.set_cookie_store("/tmp/cookie.dat")
-body, resp = @client.get @url, headers
-body.header.all.each do |h|
-  puts "#{h[0]}: #{h[1]}"
-end
-#puts @client
-@client.save_cookie_store

data/lib/stella/command/get.rb

@@ -1,44 +0,0 @@
-
-require 'httpclient'
-
-require 'util/httputil'
-
-require 'stella/command/base'
-require 'stella/data/http'
-
-# NOTE: Not working
-
-#
-#
-module Stella::Command #:nodoc: all
-  class Get < Drydock::Command #:nodoc: all
-    include Stella::Command::Base
-    
-    attr_accessor :raw
-    attr_accessor :uri
-    attr_accessor :proxy
-    
-    def run
-      @req ||= req
-      raise "No request defined" unless @req
-      
-      c = (@proxy) ? HTTPClient.new(@proxy) : HTTPClient.new
-      c.get @req.uri, @req.headers
-    end
-    
-    def read_raw
-      Stella.info("Enter the raw GET request:")
-      @raw = gets
-      while raw !~ /^#{$/}$/ 
-        val = gets
-        @raw << val if val
-      end
-      @raw
-    end
-    
-    def req
-
-    end
-    
-  end
-end

data/lib/stella/common.rb

@@ -1,53 +0,0 @@
-
-
-module Stella
-  module Common
-    class Auth
-      attr_accessor :type
-      attr_accessor :user
-      attr_accessor :pass
-      attr_accessor :port
-      def initialize(type, user, pass=nil, port=nil)
-        @uri = type
-        @user = user
-        @pass = pass if pass
-        @port = port if port
-      end
-    end
-    class Proxy 
-      attr_accessor :uri
-      attr_accessor :user
-      attr_accessor :pass
-      def initialize(uri, user=nil, pass=nil)
-        @uri = uri
-        @user = user if user
-        @pass = pass if pass
-      end
-    end
-    class Machine
-      attr_accessor :host
-      attr_accessor :port
-      attr_accessor :role
-      attr_accessor :ssh
-      
-      def initialize(*args)
-        raise "You must at least a hostname or IP address" if args.empty?
-        if args.first.is_a? String
-          @host, @port = args.first.split(":") 
-        else  
-          @host, @port, @role = args.flatten
-        end
-        @role ||= "app"
-        @port = @port.to_i if @port
-      end
-      def to_s
-        str = "#{@host}"
-        str << ":#{@port}" if @port
-        str
-      end
-    end
-  end
-end
-
-
-

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