smokey 0.0.1 → 0.0.2

data/lib/scripts/base_script.rb

@@ -0,0 +1,7 @@
+class BaseScript
+
+  def initialize
+
+  end
+
+end

data/lib/scripts/homepage.rb

@@ -0,0 +1,15 @@
+class Homepage < BaseScript
+  def initialize(base_url)
+    @base_url = base_url
+    @logger = Logger.new
+  end
+
+  def run
+    timer = Timer.new.time do
+      request = Request.new("bla", "bla")
+      request.get("#{@base_url}/people")
+    end
+    @logger.log("info", "response time: #{timer.round(2)} seconds")
+  end
+
+end

data/lib/smokey.rb

@@ -1,5 +1,32 @@
-require "smokey/version"
-
 module Smokey
-  # Your code goes here...
+  Dir[File.dirname(__FILE__) + '/smokey/*.rb'].each  {|file| require file }
+  Dir[File.dirname(__FILE__) + '/scripts/*.rb'].each {|file| require file }
+
+  class Runner
+    attr_reader :testscript, :vusers, :base_url, :duration, :delay, :rampup, :think_time
+
+    def initialize(testscript, vusers, base_url, duration, delay, rampup, think_time)
+      @testscript = testscript
+      @vusers     = vusers
+      @base_url   = base_url
+      @duration   = duration
+      @delay      = delay
+      @rampup     = rampup
+      @think_time = think_time
+    end
+
+    def start_test
+      threads = []
+      @vusers.times do
+        thread = Thread.new do
+          VirtualUser.new(@testscript, @base_url).start
+        end
+        threads << thread
+      end
+      threads.each { |t| t.join }
+    end
+  end
 end
+
+@smokey = Smokey::Runner.new('Homepage', 5, 'http://localhost:3000', 10, 5, 1, 2)
+@smokey.start_test

data/lib/smokey/logger.rb

@@ -0,0 +1,21 @@
+require "redis"
+
+class Logger
+
+  def initialize
+    @redis = Redis.new
+  end
+
+  def log(type, item)
+    case type
+      when "error"
+        puts "ERROR - #{item}"
+      when "success"
+        puts "SUCCESS - #{item}"
+      when "info"
+        @redis.set("logje", "INFO - #{item}")
+        storage = @redis.get("logje")
+        puts storage
+    end
+  end
+end

data/lib/smokey/request.rb

@@ -0,0 +1,22 @@
+require 'httparty'
+
+class Request
+  def initialize(username, password)
+    @auth   = { username: username, password: password }
+    @logger = Logger.new
+  end
+
+  def get(url, options={})
+    begin
+      options.merge!({basic_auth: @auth})
+      response = HTTParty.get(url, options)
+      @logger.log("info", "#{response.code} - #{response.message}")
+    rescue => e
+      e.response
+    end
+  end
+
+  def post(url, parameters)
+   # RestClient.post(url, parameters)
+  end
+end

data/lib/smokey/timer.rb

@@ -0,0 +1,7 @@
+class Timer
+  def time
+    start = Time.now
+    yield
+    Time.now.to_f - start.to_f
+  end
+end

data/lib/smokey/version.rb

@@ -1,3 +1,3 @@
 module Smokey
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/lib/smokey/virtual_user.rb

@@ -0,0 +1,52 @@
+class VirtualUser
+
+  def initialize(testscript, base_url)
+    @testscript = testscript
+    @base_url   = base_url
+    @logger     = Logger.new
+  end
+
+  def start
+    @logger.log("info", "Script started")
+    script = Kernel.const_get(@testscript)
+    script = script.new(@base_url)
+    script.run
+    @logger.log("info", "Script finished")
+  end
+end
+
+=begin
+dir = File.expand_path(File.join(File.dirname(__FILE__), '..', 'lib'))
+require File.join(dir, 'httparty')
+
+class TripIt
+  include HTTParty
+  base_uri 'http://www.tripit.com'
+  debug_output
+
+  def initialize(email, password)
+    @email = email
+    response = self.class.get('/account/login')
+    response = self.class.post(
+      '/account/login',
+      :body => {
+        :login_email_address => email,
+        :login_password => password
+      },
+      :headers => {'Cookie' => response.headers['Set-Cookie']}
+    )
+    @cookie = response.request.options[:headers]['Cookie']
+  end
+
+  def account_settings
+    self.class.get('/account/edit', :headers => {'Cookie' => @cookie})
+  end
+
+  def logged_in?
+    account_settings.include? "You're logged in as #{@email}"
+  end
+end
+
+tripit = TripIt.new('email', 'password')
+puts "Logged in: #{tripit.logged_in?}"
+=end

data/spec/smokey_spec.rb

@@ -0,0 +1,15 @@
+require 'spec_helper'
+require 'smokey'
+
+describe Smokey do
+  before :each do
+    @smokey = Smokey::Runner.new(5, 'http://localhost:3000', 10, 5, 1, 2)
+  end
+
+  describe "new" do
+    it "returns a Smokey object" do
+      @smokey.should be_an_instance_of Smokey::Runner
+    end
+  end
+
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: smokey
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -39,9 +39,16 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- lib/scripts/base_script.rb
+- lib/scripts/homepage.rb
 - lib/smokey.rb
+- lib/smokey/logger.rb
+- lib/smokey/request.rb
+- lib/smokey/timer.rb
 - lib/smokey/version.rb
+- lib/smokey/virtual_user.rb
 - smokey.gemspec
+- spec/smokey_spec.rb
 - spec/spec_helper.rb
 homepage: ''
 licenses: []
@@ -69,4 +76,5 @@ specification_version: 3
 summary: With Smokey it is easy to execute a simple performance test for a webapp
   to see if it can handle the load.
 test_files:
+- spec/smokey_spec.rb
 - spec/spec_helper.rb