rebar 0.2.1 → 0.3.0

data/Rakefile

@@ -1,2 +1,11 @@
 #!/usr/bin/env rake
+require 'rake/testtask'
 require "bundler/gem_tasks"
+
+task :default => :test
+Rake::TestTask.new do |t|
+  t.libs << 'test'
+  t.pattern = 'test/**/*_test.rb'
+  t.warning = true
+  t.verbose = true
+end

data/lib/rebar/builder.rb

@@ -5,9 +5,28 @@ module Rebar
     end
 
     attr_reader :connection
+    attr_reader :system_cpu_time
+    attr_reader :user_cpu_time
+    attr_reader :memory_delta
 
     def build(&block)
+      initial_cpu_sys  = connection.info['used_cpu_sys'].to_i
+      initial_cpu_user = connection.info['used_cpu_user'].to_i
+      initial_memory   = connection.info['used_memory'].to_i
+
       Rebar::DSL::Database.new(connection).instance_eval(&block)
+
+      @system_cpu_time = connection.info['used_cpu_sys'].to_i  - initial_cpu_sys
+      @user_cpu_time   = connection.info['used_cpu_user'].to_i - initial_cpu_user
+      @memory_delta    = connection.info['used_memory'].to_i   - initial_memory
+
+      self
+    end
+
+    def report
+      STDERR.puts "System CPU:   \t#{system_cpu_time}s"
+      STDERR.puts "User CPU:     \t#{user_cpu_time}s"
+      STDERR.puts "Memory delta: \t#{memory_delta} bytes"
     end
 
     def clean!

data/lib/rebar/dsl/database.rb

@@ -16,25 +16,25 @@ module Rebar
 
       def hashes(count, &block)
         count.times do
-          Rebar::DSL::Hash.new(connection, "rebar:hash:#{next_id(:hash)}").instance_eval(&block)
+          Rebar::DSL::Hash.new(connection, "rebar:hash:#{next_id(:hash)}", &block)
         end
       end
 
       def lists(count, &block)
         count.times do
-          Rebar::DSL::List.new(connection, "rebar:list:#{next_id(:list)}").instance_eval(&block)
+          Rebar::DSL::List.new(connection, "rebar:list:#{next_id(:list)}", &block)
         end
       end
 
       def sets(count, &block)
         count.times do
-          Rebar::DSL::Set.new(connection, "rebar:set:#{next_id(:set)}").instance_eval(&block)
+          Rebar::DSL::Set.new(connection, "rebar:set:#{next_id(:set)}", &block)
         end
       end
 
       def sorted_sets(count, &block)
         count.times do
-          Rebar::DSL::SortedSet.new(connection, "rebar:sorted_set:#{next_id(:sorted_set)}").instance_eval(&block)
+          Rebar::DSL::SortedSet.new(connection, "rebar:sorted_set:#{next_id(:sorted_set)}", &block)
         end
       end
 

data/lib/rebar/dsl/hash.rb

@@ -1,9 +1,10 @@
 module Rebar
   module DSL
     class Hash
-      def initialize(connection, key)
+      def initialize(connection, key, &block)
         @connection = connection
         @key        = key
+        instance_eval(&block) if block_given?
       end
 
       attr_reader :connection

data/lib/rebar/dsl/list.rb

@@ -1,9 +1,10 @@
 module Rebar
   module DSL
     class List
-      def initialize(connection, key)
+      def initialize(connection, key, &block)
         @connection = connection
         @key        = key
+        instance_eval(&block) if block_given?
       end
 
       attr_reader :connection

data/lib/rebar/dsl/set.rb

@@ -1,9 +1,10 @@
 module Rebar
   module DSL
     class Set
-      def initialize(connection, key)
+      def initialize(connection, key, &block)
         @connection = connection
         @key        = key
+        instance_eval(&block) if block_given?
       end
 
       attr_reader :connection

data/lib/rebar/dsl/sorted_set.rb

@@ -1,9 +1,10 @@
 module Rebar
   module DSL
     class SortedSet
-      def initialize(connection, key)
+      def initialize(connection, key, &block)
         @connection = connection
         @key        = key
+        instance_eval(&block) if block_given?
       end
 
       attr_reader :connection

data/lib/rebar/version.rb

@@ -1,3 +1,3 @@
 module Rebar
-  VERSION = "0.2.1"
+  VERSION = "0.3.0"
 end

data/test/builder_test.rb

@@ -0,0 +1,9 @@
+require 'test_helper'
+
+class BuilderTest < Test::Unit::TestCase
+  def test_clean_should_remove_all_keys
+    connection.set('rebar:foo', 'bar')
+    Rebar::Builder.new(connection).clean!
+    assert_equal 0, connection.keys('rebar:*').length
+  end
+end

data/test/hashes_test.rb

@@ -0,0 +1,18 @@
+require 'test_helper'
+
+class HashesTest < Test::Unit::TestCase
+  def test_creates_hashes
+    Rebar.build(connection) do
+      hashes(5) do
+        keys(10)
+      end
+    end
+
+    keys = connection.keys('rebar:hash:*')
+    assert_equal 5, keys.length, "Created incorrect number of hashes"
+
+    keys.each do |key|
+      assert_equal 10, connection.hkeys(key).length, "Created incorrect number of keys in hash #{key}"
+    end
+  end
+end

data/test/lists_test.rb

@@ -0,0 +1,18 @@
+require 'test_helper'
+
+class ListsTest < Test::Unit::TestCase
+  def test_creates_lists
+    Rebar.build(connection) do
+      lists(5) do
+        elements(10)
+      end
+    end
+
+    keys = connection.keys('rebar:list:*')
+    assert_equal 5, keys.length, "Created incorrect number of lists"
+
+    keys.each do |key|
+      assert_equal 10, connection.llen(key), "Created incorrect number of elements in set #{key}"
+    end
+  end
+end

data/test/sets_test.rb

@@ -0,0 +1,18 @@
+require 'test_helper'
+
+class SetsTest < Test::Unit::TestCase
+  def test_creates_sets
+    Rebar.build(connection) do
+      sets(5) do
+        members(10)
+      end
+    end
+
+    keys = connection.keys('rebar:set:*')
+    assert_equal 5, keys.length, "Created incorrect number of sets"
+
+    keys.each do |key|
+      assert_equal 10, connection.smembers(key).length, "Created incorrect number of members in set #{key}"
+    end
+  end
+end

data/test/sorted_sets_test.rb

@@ -0,0 +1,18 @@
+require 'test_helper'
+
+class SortedSetsTest < Test::Unit::TestCase
+  def test_creates_sorted_sets
+    Rebar.build(connection) do
+      sorted_sets(5) do
+        members(10)
+      end
+    end
+
+    keys = connection.keys('rebar:sorted_set:*')
+    assert_equal 5, keys.length, "Created incorrect number of sorted sets"
+
+    keys.each do |key|
+      assert_equal 10, connection.zrange(key, 0, -1).length, "Created incorrect number of members in sorted set #{key}"
+    end
+  end
+end

data/test/strings_test.rb

@@ -0,0 +1,12 @@
+require 'test_helper'
+
+class StringsTest < Test::Unit::TestCase
+  def test_creates_strings
+    Rebar.build(connection) do
+      strings(5)
+    end
+
+    keys = connection.keys('rebar:string:*')
+    assert_equal 5, keys.length, "Created incorrect number of strings"
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,23 @@
+$:.unshift File.expand_path('../lib', File.dirname(__FILE__))
+
+require 'test/unit'
+require 'redis'
+require 'rebar'
+
+
+begin
+  require 'ruby-debug'
+rescue LoadError
+end
+
+class Test::Unit::TestCase
+  def connection
+    @connection ||= Redis.new(:db => 15).tap do |connection|
+                      connection.flushdb
+                    end
+  end
+
+  def teardown
+    connection.flushdb
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rebar
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.3.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-02-14 00:00:00.000000000 Z
+date: 2012-02-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: redis
-  requirement: &70328361538880 !ruby/object:Gem::Requirement
+  requirement: &70260397074220 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,7 +21,7 @@ dependencies:
         version: '2.2'
   type: :runtime
   prerelease: false
-  version_requirements: *70328361538880
+  version_requirements: *70260397074220
 description: ''
 email: paul.rosania@gmail.com
 executables: []
@@ -44,6 +44,13 @@ files:
 - lib/rebar/random.rb
 - lib/rebar/version.rb
 - rebar.gemspec
+- test/builder_test.rb
+- test/hashes_test.rb
+- test/lists_test.rb
+- test/sets_test.rb
+- test/sorted_sets_test.rb
+- test/strings_test.rb
+- test/test_helper.rb
 homepage: https://github.com/paulrosania/rebar
 licenses:
 - MIT
@@ -69,4 +76,11 @@ rubygems_version: 1.8.11
 signing_key: 
 specification_version: 3
 summary: A simple stress testing tool for Redis
-test_files: []
+test_files:
+- test/builder_test.rb
+- test/hashes_test.rb
+- test/lists_test.rb
+- test/sets_test.rb
+- test/sorted_sets_test.rb
+- test/strings_test.rb
+- test/test_helper.rb