riak-shim 0.0.10 → 0.0.12

data/lib/riak-shim/persistable.rb

@@ -45,7 +45,7 @@ module Riak
       module ClassMethods
         def delete_all
           bucket.keys.each do |key|
-            bucket.delete key
+            bucket.delete(key)
           end
         end
 
@@ -91,6 +91,12 @@ module Riak
         def gen_key
           UUIDTools::UUID.random_create.to_s
         end
+
+        def count
+          counter = 0
+          bucket.keys {|keys| counter += keys.count }
+          return counter
+        end
       end
 
       def self.included(base)

data/lib/riak-shim/version.rb

@@ -1,5 +1,5 @@
 module Riak
   module Shim
-    VERSION = "0.0.10"
+    VERSION = "0.0.12"
   end
 end

data/riak-shim.gemspec

@@ -24,6 +24,7 @@ Gem::Specification.new do |gem|
   gem.add_development_dependency "guard"
   gem.add_development_dependency "guard-rspec"
   gem.add_development_dependency "pry"
+  gem.add_development_dependency "awesome_print"
 
   if RUBY_PLATFORM.include? 'darwin'
     gem.add_development_dependency 'growl'

data/spec/persistable_spec.rb

@@ -20,10 +20,10 @@ class PersistableExample
 
   def ==(other)
     other.instance_of?(self.class) and
-        self.key == other.key and
-        self.foo == other.foo and
-        self.bar == other.bar and
-        self.baz == other.baz
+    self.key == other.key and
+    self.foo == other.foo and
+    self.bar == other.bar and
+    self.baz == other.baz
   end
 end
 
@@ -33,6 +33,14 @@ class PersistableExampleWithIndex < PersistableExample
   end
 end
 
+def create_test_data(copies)
+  copies.times do |i|
+    p = PersistableExample.new
+    p.foo, p.bar, p.baz = i, i, i
+    p.save
+  end
+end
+
 describe 'persistable' do
   let(:persistable) do
     p = PersistableExample.new ; p.foo = 'boo' ; p.bar = 'who' ; p
@@ -96,7 +104,7 @@ describe 'persistable' do
       @indexes[:oldkey] = ['oldval']
     end
 
-    context 'with a model which does not need indes' do
+    context 'with a model which does not need indexes' do
       it 'does not require #fields_to_index to be defined' do
         expect { persistable.set_indexes(@indexes) }.to_not raise_error
       end
@@ -116,6 +124,10 @@ describe 'persistable' do
         indexable.set_indexes(@indexes)
       end
 
+      after do
+        PersistableExampleWithIndex.delete_all
+      end
+
       it 'sets indexes for specified fields and no others' do
         @indexes.keys.should =~ ['bar_bin', 'baz_bin']
       end
@@ -124,22 +136,18 @@ describe 'persistable' do
 
   describe '#delete_all' do
     before do
-      10.times do |i|
-        p = PersistableExample.new
-        p.foo, p.bar, p.baz = i, i, i
-        p.save
-      end
-      @keys = bucket.keys
+      create_test_data(10)
       PersistableExample.delete_all
+      sleep(5)
     end
 
     it 'has no instances in the DB' do
-      @keys.map { |k| bucket.exists?(k).should be_false }
+      PersistableExample.bucket.keys.count.should eq 0
     end
   end
 
   describe '#de_camel' do
-    it 'handles a bacic camel-case name' do
+    it 'handles a basic camel-case name' do
       DeCamel.each do |camelcase, lowered|
         PersistableExample.de_camel(camelcase).should eq(lowered)
       end
@@ -158,8 +166,7 @@ describe 'persistable' do
     end
 
     it 'does not produce collisions' do
-      PersistableExample.de_camel('RiakShim').should_not
-          eq PersistableExample.de_camel('Riak::Shim')
+      PersistableExample.de_camel('RiakShim').should_not eq PersistableExample.de_camel('Riak::Shim')
     end
   end
 
@@ -209,5 +216,18 @@ describe 'persistable' do
       @keys.uniq!.should be_nil
     end
   end
-end
 
+  describe '#count' do
+    COUNT = 50
+
+    before do
+      PersistableExample.delete_all
+      sleep(5)
+      create_test_data(COUNT)
+    end
+
+    it 'returns an accurate count' do
+      PersistableExample.count.should eq COUNT
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -1,6 +1,7 @@
 $: << './lib'
 require 'riak-shim'
 require 'pry'
+require 'awesome_print'
 
 ENV['RACK_ENV'] = 'test' unless ENV['RACK_ENV']
 DB_CONFIG = { 'development' => { 'bucket_prefix' => 'dev_', 'host' => "localhost", 'http_port' => 8098},

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: riak-shim
 version: !ruby/object:Gem::Version
-  version: 0.0.10
+  version: 0.0.12
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-08-07 00:00:00.000000000 Z
+date: 2012-08-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: riak-client
@@ -139,6 +139,22 @@ dependencies:
     - - ! '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: awesome_print
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: growl
   requirement: !ruby/object:Gem::Requirement