Allow the error handler to be configured

fredjean · Jan 21, 2010 · d765121 · d765121
1 parent 75834d9
commit d765121
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/lib/memcache.rb b/lib/memcache.rb
@@ -87,6 +87,7 @@ def initialize(*args)
 
     @client = MemCachedClient.new(@pool_name)
 
+    @client.error_handler = opts[:error_handler] if opts[:error_handler]
     @client.primitiveAsString = true
     @client.sanitizeKeys = false
 

diff --git a/spec/jruby_memcache_spec.rb b/spec/jruby_memcache_spec.rb
@@ -36,6 +36,15 @@
       @client = MemCache.new([@server], :pool_name => 'new_pool')
       @client.pool_name.should == 'new_pool'
     end
+
+    it "should work with an error handler" do
+      include_class 'com.danga.MemCached.MemCachedClient'
+      java_memcache_client = mock.as_null_object
+      MemCachedClient.stub!(:new => java_memcache_client)
+      error_handler = Object.new
+      java_memcache_client.should_receive(:error_handler=).with(error_handler)
+      @client = MemCache.new([@server], :error_handler => error_handler)
+    end
   end
 
   describe "namespacing" do