jmoses-couchbase 1.3.6

data/test/test_delete.rb

@@ -0,0 +1,120 @@
+# Author:: Couchbase <info@couchbase.com>
+# Copyright:: 2011, 2012 Couchbase, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require File.join(File.dirname(__FILE__), 'setup')
+
+class TestStore < MiniTest::Test
+
+  def setup
+    @mock = start_mock
+  end
+
+  def teardown
+    stop_mock(@mock)
+  end
+
+  def test_trivial_delete
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    connection.set(uniq_id, "bar")
+    assert connection.delete(uniq_id)
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.delete(uniq_id)
+    end
+  end
+
+  def test_delete_missing
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.delete(uniq_id(:missing))
+    end
+    refute connection.delete(uniq_id(:missing), :quiet => true)
+    refute connection.quiet?
+    connection.quiet = true
+    refute connection.delete(uniq_id(:missing))
+  end
+
+  def test_delete_with_cas
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    cas = connection.set(uniq_id, "bar")
+    missing_cas = cas - 1
+    assert_raises(Couchbase::Error::KeyExists) do
+      connection.delete(uniq_id, :cas => missing_cas)
+    end
+    assert connection.delete(uniq_id, :cas => cas)
+  end
+
+  def test_allow_fixnum_as_cas_parameter
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    cas = connection.set(uniq_id, "bar")
+    assert connection.delete(uniq_id, cas)
+  end
+
+  def test_delete_with_prefix
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :key_prefix => "prefix:")
+    connection.set(uniq_id(:foo), "bar")
+    assert connection.delete(uniq_id(:foo))
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id(:foo))
+    end
+  end
+
+  def test_simple_multi_delete
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :quiet => true)
+    connection.set(uniq_id(1) => "bar", uniq_id(2) => "foo")
+    res = connection.delete(uniq_id(1), uniq_id(2))
+    assert res.is_a?(Hash)
+    assert res[uniq_id(1)]
+    assert res[uniq_id(2)]
+  end
+
+  def test_simple_multi_delete_missing
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :quiet => true)
+    connection.set(uniq_id(1) => "bar", uniq_id(2) => "foo")
+    res = connection.delete(uniq_id(1), uniq_id(:missing), :quiet => true)
+    assert res.is_a?(Hash)
+    assert res[uniq_id(1)]
+    refute res[uniq_id(:missing)]
+  end
+
+  def test_multi_delete_with_cas_check
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :quiet => true)
+    cas = connection.set(uniq_id(1) => "bar", uniq_id(2) => "foo")
+    res = connection.delete(uniq_id(1) => cas[uniq_id(1)], uniq_id(2) => cas[uniq_id(2)])
+    assert res.is_a?(Hash)
+    assert res[uniq_id(1)]
+    assert res[uniq_id(2)]
+  end
+
+  def test_multi_delete_missing_with_cas_check
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :quiet => true)
+    cas = connection.set(uniq_id(1) => "bar", uniq_id(2) => "foo")
+    res = connection.delete(uniq_id(1) => cas[uniq_id(1)], uniq_id(:missing) => cas[uniq_id(2)])
+    assert res.is_a?(Hash)
+    assert res[uniq_id(1)]
+    refute res[uniq_id(:missing)]
+  end
+
+  def test_multi_delete_with_cas_check_mismatch
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :quiet => true)
+    cas = connection.set(uniq_id(1) => "bar", uniq_id(2) => "foo")
+
+    assert_raises(Couchbase::Error::KeyExists) do
+      connection.delete(uniq_id(1) => cas[uniq_id(1)] + 1,
+                        uniq_id(2) => cas[uniq_id(2)])
+    end
+  end
+end

data/test/test_errors.rb

@@ -0,0 +1,82 @@
+# Author:: Couchbase <info@couchbase.com>
+# Copyright:: 2011, 2012 Couchbase, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require File.join(File.dirname(__FILE__), 'setup')
+require 'digest/md5'
+
+class TestErrors < MiniTest::Test
+
+  def setup
+    @mock = start_mock
+  end
+
+  def teardown
+    stop_mock(@mock)
+  end
+
+  def genkey(item)
+    tuple = [item["author"], item["message"]]
+    Digest::MD5.hexdigest(tuple.join('-'))
+  end
+
+  def test_graceful_add_with_collision
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    msg1 = {"author" => "foo", "message" => "hi all", "time" => "2012-01-12 11:29:09"}
+    key1 = uniq_id(genkey(msg1))
+    msg2 = {"author" => "foo", "message" => "hi all", "time" => "2012-01-12 11:29:30"}
+    key2 = uniq_id(genkey(msg2))
+
+    connection.add(key1, msg1)
+    begin
+      connection.add(key2, msg2)
+    rescue Couchbase::Error::KeyExists => ex
+      # using info from exception
+      # it could be done with cas operation, but we can save one request
+      # here (in real world cas operation will be more consistent because it
+      # fetch fresh version from the cluster)
+      #
+      # connection.cas(key2) do |msg|
+      #   msg.merge("time" => [msg["time"], msg2["time"]])
+      # end
+      msg2 = msg1.merge("time" => [msg1["time"], msg2["time"]])
+      connection.set(key2, msg2, :cas => ex.cas)
+    end
+
+    msg3 = {"author" => "foo", "message" => "hi all",
+                "time" => ["2012-01-12 11:29:09", "2012-01-12 11:29:30"]}
+    key3 = uniq_id(genkey(msg3))
+    assert_equal msg3, connection.get(key3)
+
+    connection.run do |conn|
+      msg4 = {"author" => "foo", "message" => "hi all", "time" => "2012-01-12 11:45:34"}
+      key4 = uniq_id(genkey(msg4))
+
+      connection.add(key4, msg4) do |ret|
+        assert_equal :add, ret.operation
+        assert_equal key4, ret.key
+        msg4 = msg3.merge("time" => msg3["time"] + [msg4["time"]])
+        connection.set(ret.key, msg4, :cas => ret.cas)
+      end
+    end
+
+    msg5 = {"author" => "foo", "message" => "hi all",
+                "time" => ["2012-01-12 11:29:09", "2012-01-12 11:29:30", "2012-01-12 11:45:34"]}
+    key5 = uniq_id(genkey(msg5))
+    assert_equal msg5, connection.get(key5)
+  end
+
+end

data/test/test_eventmachine.rb

@@ -0,0 +1,70 @@
+# Author:: Couchbase <info@couchbase.com>
+# Copyright:: 2011, 2012 Couchbase, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require File.join(File.dirname(__FILE__), 'setup')
+require 'eventmachine'
+
+class TestEventmachine < MiniTest::Test
+
+  def setup
+    @mock = start_mock
+  end
+
+  def teardown
+    stop_mock(@mock)
+  end
+
+  if RUBY_VERSION.to_f >= 1.9
+
+    def test_trivial_set
+      EM.run do
+        conn = Couchbase.new(:hostname => @mock.host, :port => @mock.port,
+                             :engine => :eventmachine, :async => true)
+        conn.on_connect do |res|
+          assert res.success?, "connection must be successful"
+          conn.set(uniq_id, "bar") do |res|
+            assert res.success?, "the operation must be successful"
+            assert res.cas > 0, "the CAS value must be non-zero"
+            EM.stop
+          end
+        end
+      end
+    end
+
+    def test_trivial_get
+      EM.run do
+        conn = Couchbase.new(:hostname => @mock.host, :port => @mock.port,
+                             :engine => :eventmachine, :async => true)
+        conn.on_connect do |res|
+          assert res.success?, "connection must be successful"
+          conn.set(uniq_id, "bar") do |res|
+            assert res.success?, "the set operation must be successful"
+            cas = res.cas
+            conn.get(uniq_id) do |res|
+              assert res.success?, "the get operation must be successful"
+              assert_equal "bar", res.value
+              assert_equal cas, res.cas
+              EM.stop
+            end
+          end
+        end
+      end
+    end
+
+  end
+
+end

data/test/test_format.rb

@@ -0,0 +1,164 @@
+# Author:: Couchbase <info@couchbase.com>
+# Copyright:: 2011, 2012 Couchbase, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require File.join(File.dirname(__FILE__), 'setup')
+
+class TestFormat < MiniTest::Test
+
+  ArbitraryClass = Struct.new(:name, :role)
+  class SkinyClass < Struct.new(:name, :role)
+    undef to_s rescue nil
+    undef to_json rescue nil
+  end
+
+  def setup
+    @mock = start_mock
+  end
+
+  def teardown
+    stop_mock(@mock)
+  end
+
+  def test_default_document_format
+    orig_doc = {'name' => 'Twoflower', 'role' => 'The tourist'}
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    assert_equal :document, connection.default_format
+    connection.set(uniq_id, orig_doc)
+    doc, flags, cas = connection.get(uniq_id, :extended => true)
+    assert_equal 0x00, flags & 0x11
+    assert doc.is_a?(Hash)
+    assert_equal 'Twoflower', doc['name']
+    assert_equal 'The tourist', doc['role']
+  end
+
+  def test_it_raises_error_for_document_format_when_neither_to_json_nor_to_s_defined
+    if (MultiJson.respond_to?(:engine) ? MultiJson.engine : MultiJson.adapter).name =~ /Yajl$/
+      orig_doc = SkinyClass.new("Twoflower", "The tourist")
+      refute orig_doc.respond_to?(:to_s)
+      refute orig_doc.respond_to?(:to_json)
+
+      connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :default_format => :document)
+      assert_raises(Couchbase::Error::ValueFormat) do
+        connection.set(uniq_id, orig_doc)
+      end
+
+      class << orig_doc
+        def to_json
+          MultiJson.dump(:name => name, :role => role)
+        end
+      end
+      connection.set(uniq_id, orig_doc) # OK
+
+      class << orig_doc
+        undef to_json
+        def to_s
+          MultiJson.dump(:name => name, :role => role)
+        end
+      end
+      connection.set(uniq_id, orig_doc) # OK
+    end
+  end
+
+  def test_it_could_dump_arbitrary_class_using_marshal_format
+    orig_doc = ArbitraryClass.new("Twoflower", "The tourist")
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    connection.set(uniq_id, orig_doc, :format => :marshal)
+    doc, flags, cas = connection.get(uniq_id, :extended => true)
+    assert_equal Couchbase::Bucket::FMT_MARSHAL, flags & Couchbase::Bucket::FMT_MASK
+    assert doc.is_a?(ArbitraryClass)
+    assert_equal 'Twoflower', doc.name
+    assert_equal 'The tourist', doc.role
+  end
+
+  def test_it_accepts_only_string_in_plain_mode
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :default_format => :plain)
+    connection.set(uniq_id, "1")
+
+    assert_raises(Couchbase::Error::ValueFormat) do
+      connection.set(uniq_id, 1)
+    end
+
+    assert_raises(Couchbase::Error::ValueFormat) do
+      connection.set(uniq_id, {:foo => "bar"})
+    end
+  end
+
+  def test_bignum_conversion
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :default_format => :plain)
+    cas = 0xffff_ffff_ffff_ffff
+    assert cas.is_a?(Bignum)
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.delete(uniq_id => cas)
+    end
+  end
+
+  def test_it_allows_to_turn_off_transcoder
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :transcoder => nil)
+    connection.set(uniq_id, "value", :flags => 0xffff_ffff)
+    doc, flags, _ = connection.get(uniq_id, :extended => true)
+    assert_equal "value", doc
+    assert_equal 0xffff_ffff, flags
+  end
+
+  require 'zlib'
+  # This class wraps any other transcoder and performs compression
+  # using zlib
+  class ZlibTranscoder
+    FMT_ZLIB = 0x04
+
+    def initialize(base)
+      @base = base
+    end
+
+    def dump(obj, flags, options = {})
+      obj, flags = @base.dump(obj, flags, options)
+      z = Zlib::Deflate.new(Zlib::BEST_SPEED)
+      buffer = z.deflate(obj, Zlib::FINISH)
+      z.close
+      [buffer, flags|FMT_ZLIB]
+    end
+
+    def load(blob, flags, options = {})
+      # decompress value only if Zlib flag set
+      if (flags & FMT_ZLIB) == FMT_ZLIB
+        z = Zlib::Inflate.new
+        blob = z.inflate(blob)
+        z.finish
+        z.close
+      end
+      @base.load(blob, flags, options)
+    end
+  end
+
+  def test_it_can_use_custom_transcoder
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    connection.transcoder = ZlibTranscoder.new(Couchbase::Transcoder::Document)
+    connection.set(uniq_id, {"foo" => "bar"})
+    doc, flags, _ = connection.get(uniq_id, :extended => true)
+    assert_equal({"foo" => "bar"}, doc)
+    assert_equal(ZlibTranscoder::FMT_ZLIB|Couchbase::Bucket::FMT_DOCUMENT, flags)
+    connection.transcoder = nil
+    doc = connection.get(uniq_id)
+    case RUBY_VERSION
+    when /^1\.8/
+      assert_equal "x\x01\xABVJ\xCB\xCFW\xB2RJJ,R\xAA\x05\0\x1Dz\x044", doc
+    else
+      assert_equal "x\u0001\xABVJ\xCB\xCFW\xB2RJJ,R\xAA\u0005\u0000\u001Dz\u00044", doc
+    end
+  end
+
+end

data/test/test_get.rb

@@ -0,0 +1,407 @@
+# Author:: Couchbase <info@couchbase.com>
+# Copyright:: 2011, 2012 Couchbase, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require File.join(File.dirname(__FILE__), 'setup')
+
+class TestGet < MiniTest::Test
+
+  def setup
+    @mock = start_mock
+  end
+
+  def teardown
+    stop_mock(@mock)
+  end
+
+  def test_trivial_get
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    connection.set(uniq_id, "bar")
+    val = connection.get(uniq_id)
+    assert_equal "bar", val
+  end
+
+  def test_extended_get
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    orig_cas = connection.set(uniq_id, "bar")
+    val, flags, cas = connection.get(uniq_id, :extended => true)
+    assert_equal "bar", val
+    assert_equal 0x0, flags
+    assert_equal orig_cas, cas
+
+    orig_cas = connection.set(uniq_id, "bar", :flags => 0x1000)
+    val, flags, cas = connection.get(uniq_id, :extended => true)
+    assert_equal "bar", val
+    assert_equal 0x1000, flags
+    assert_equal orig_cas, cas
+  end
+
+  def test_multi_get
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    connection.set(uniq_id(1), "foo1")
+    connection.set(uniq_id(2), "foo2")
+
+    val1, val2 = connection.get(uniq_id(1), uniq_id(2))
+    assert_equal "foo1", val1
+    assert_equal "foo2", val2
+  end
+
+  def test_multi_get_extended
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    cas1 = connection.set(uniq_id(1), "foo1")
+    cas2 = connection.set(uniq_id(2), "foo2")
+
+    results = connection.get(uniq_id(1), uniq_id(2), :extended => true)
+    assert_equal ["foo1", 0x0, cas1], results[uniq_id(1)]
+    assert_equal ["foo2", 0x0, cas2], results[uniq_id(2)]
+  end
+
+  def test_multi_get_and_touch
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    connection.set(uniq_id(1), "foo1")
+    connection.set(uniq_id(2), "foo2")
+
+    results = connection.get(uniq_id(1) => 1, uniq_id(2) => 1)
+    assert results.is_a?(Hash)
+    assert_equal "foo1", results[uniq_id(1)]
+    assert_equal "foo2", results[uniq_id(2)]
+    sleep(2)
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id(1), uniq_id(2))
+    end
+    assert connection.get(uniq_id(1), uniq_id(2), :quiet => true).compact.empty?
+  end
+
+  def test_multi_get_and_touch_extended
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    cas1 = connection.set(uniq_id(1), "foo1")
+    cas2 = connection.set(uniq_id(2), "foo2")
+
+    results = connection.get({uniq_id(1) => 1, uniq_id(2) => 1}, :extended => true)
+    assert_equal ["foo1", 0x0, cas1], results[uniq_id(1)]
+    assert_equal ["foo2", 0x0, cas2], results[uniq_id(2)]
+  end
+
+  def test_multi_get_and_touch_with_single_key
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    connection.set(uniq_id, "foo1")
+
+    results = connection.get(uniq_id => 1)
+    assert results.is_a?(Hash)
+    assert_equal "foo1", results[uniq_id]
+    sleep(2)
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id)
+    end
+  end
+
+  def test_missing_in_quiet_mode
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :quiet => true)
+    cas1 = connection.set(uniq_id(1), "foo1")
+    cas2 = connection.set(uniq_id(2), "foo2")
+
+    val = connection.get(uniq_id(:missing))
+    refute(val)
+    val = connection.get(uniq_id(:missing), :extended => true)
+    refute(val)
+
+    val1, missing, val2  = connection.get(uniq_id(1), uniq_id(:missing), uniq_id(2))
+    assert_equal "foo1", val1
+    refute missing
+    assert_equal "foo2", val2
+
+    results = connection.get(uniq_id(1), uniq_id(:missing), uniq_id(2), :extended => true)
+    assert_equal ["foo1", 0x0, cas1], results[uniq_id(1)]
+    refute results[uniq_id(:missing)]
+    assert_equal ["foo2", 0x0, cas2], results[uniq_id(2)]
+  end
+
+  def test_it_allows_temporary_quiet_flag
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :quiet => false)
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id(:missing))
+    end
+    refute connection.get(uniq_id(:missing), :quiet => true)
+  end
+
+  def test_missing_in_verbose_mode
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :quiet => false)
+    connection.set(uniq_id(1), "foo1")
+    connection.set(uniq_id(2), "foo2")
+
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id(:missing))
+    end
+
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id(:missing), :extended => true)
+    end
+
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id(1), uniq_id(:missing), uniq_id(2))
+    end
+
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id(1), uniq_id(:missing), uniq_id(2), :extended => true)
+    end
+  end
+
+  def test_asynchronous_get
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    cas = connection.set(uniq_id, "foo", :flags => 0x6660)
+    res = []
+
+    suite = lambda do |conn|
+      res.clear
+      conn.get(uniq_id) # ignore result
+      conn.get(uniq_id) {|ret| res << ret}
+      handler = lambda {|ret| res << ret}
+      conn.get(uniq_id, &handler)
+    end
+
+    checks = lambda do
+      res.each do |r|
+        assert r.is_a?(Couchbase::Result)
+        assert r.success?
+        assert_equal uniq_id, r.key
+        assert_equal "foo", r.value
+        assert_equal 0x6660, r.flags
+        assert_equal cas, r.cas
+      end
+    end
+
+    connection.run(&suite)
+    checks.call
+
+    connection.run{ suite.call(connection) }
+    checks.call
+  end
+
+  def test_asynchronous_multi_get
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    connection.set(uniq_id(1), "foo")
+    connection.set(uniq_id(2), "bar")
+
+    res = {}
+    connection.run do |conn|
+      conn.get(uniq_id(1), uniq_id(2)) {|ret| res[ret.key] = ret.value}
+    end
+
+    assert res[uniq_id(1)]
+    assert_equal "foo", res[uniq_id(1)]
+    assert res[uniq_id(2)]
+    assert_equal "bar", res[uniq_id(2)]
+  end
+
+  def test_asynchronous_get_missing
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    connection.set(uniq_id, "foo")
+    res = {}
+    missing = []
+
+    get_handler = lambda do |ret|
+      assert_equal :get, ret.operation
+      if ret.success?
+        res[ret.key] = ret.value
+      else
+        if ret.error.is_a?(Couchbase::Error::NotFound)
+          missing << ret.key
+        else
+          raise ret.error
+        end
+      end
+    end
+
+    suite = lambda do |conn|
+      res.clear
+      missing.clear
+      conn.get(uniq_id(:missing1), &get_handler)
+      conn.get(uniq_id, uniq_id(:missing2), &get_handler)
+    end
+
+    connection.run(&suite)
+    refute res.has_key?(uniq_id(:missing1))
+    refute res.has_key?(uniq_id(:missing2))
+    assert_equal [uniq_id(:missing1), uniq_id(:missing2)], missing.sort
+    assert_equal "foo", res[uniq_id]
+
+    connection.quiet = true
+    connection.run(&suite)
+    assert_equal "foo", res[uniq_id]
+    assert res.has_key?(uniq_id(:missing1)) # handler was called with nil
+    refute res[uniq_id(:missing1)]
+    assert res.has_key?(uniq_id(:missing2))
+    refute res[uniq_id(:missing2)]
+    assert_empty missing
+  end
+
+  def test_get_using_brackets
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    orig_cas = connection.set(uniq_id, "foo", :flags => 0x1100)
+
+    val = connection[uniq_id]
+    assert_equal "foo", val
+
+    if RUBY_VERSION =~ /^1\.9/
+      eval <<-EOC
+      val, flags, cas = connection[uniq_id, :extended => true]
+      assert_equal "foo", val
+      assert_equal 0x1100, flags
+      assert_equal orig_cas, cas
+      EOC
+    end
+  end
+
+  def test_it_allows_to_store_nil
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    orig_cas = connection.set(uniq_id, nil)
+    assert orig_cas.is_a?(Numeric)
+
+    refute connection.get(uniq_id)
+    # doesn't raise NotFound exception
+    refute connection.get(uniq_id, :quiet => false)
+    # returns CAS
+    value, flags, cas = connection.get(uniq_id, :extended => true)
+    refute value
+    assert_equal 0x00, flags
+    assert_equal orig_cas, cas
+  end
+
+  def test_zero_length_string_is_not_nil
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    connection.set(uniq_id, "", :format => :document)
+    assert_equal "", connection.get(uniq_id)
+
+    connection.set(uniq_id, "", :format => :plain)
+    assert_equal "", connection.get(uniq_id)
+
+    connection.set(uniq_id, "", :format => :marshal)
+    assert_equal "", connection.get(uniq_id)
+
+    connection.set(uniq_id, nil, :format => :document)
+    assert_equal nil, connection.get(uniq_id, :quiet => false)
+
+    assert_raises Couchbase::Error::ValueFormat do
+      connection.set(uniq_id, nil, :format => :plain)
+    end
+
+    connection.set(uniq_id, nil, :format => :marshal)
+    assert_equal nil, connection.get(uniq_id, :quiet => false)
+  end
+
+  def test_format_forcing
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    connection.set(uniq_id, '{"foo":"bar"}', :format => :plain)
+    value, flags, _ = connection.get(uniq_id, :extended => true)
+    assert_equal '{"foo":"bar"}', value
+    assert_equal 0x02, flags
+
+    value, flags, _ = connection.get(uniq_id, :extended => true, :format => :document)
+    expected = {"foo" => "bar"}
+    assert_equal expected, value
+    assert_equal 0x02, flags
+
+    connection.prepend(uniq_id, "NOT-A-JSON")
+    assert_raises Couchbase::Error::ValueFormat do
+      connection.get(uniq_id, :format => :document)
+    end
+  end
+
+  # http://www.couchbase.com/issues/browse/RCBC-31
+  def test_consistent_behaviour_for_arrays
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    cas = connection.set(uniq_id("foo"), "foo")
+    connection.set(uniq_id("bar"), "bar")
+
+    assert_equal "foo", connection.get(uniq_id("foo"))
+    assert_equal ["foo"], connection.get([uniq_id("foo")])
+    assert_equal ["foo", "bar"], connection.get([uniq_id("foo"), uniq_id("bar")])
+    assert_equal ["foo", "bar"], connection.get(uniq_id("foo"), uniq_id("bar"))
+    expected = {uniq_id("foo") => ["foo", 0x00, cas]}
+    assert_equal expected, connection.get([uniq_id("foo")], :extended => true)
+    assert_raises TypeError do
+      connection.get([uniq_id("foo"), uniq_id("bar")], [uniq_id("foo")])
+    end
+  end
+
+  def test_get_with_lock_trivial
+    if @mock.real?
+      connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+      connection.set(uniq_id, "foo")
+      assert_equal "foo", connection.get(uniq_id, :lock => 1)
+      assert_raises Couchbase::Error::KeyExists do
+        connection.set(uniq_id, "bar")
+      end
+      sleep(2)
+      connection.set(uniq_id, "bar")
+    else
+      skip("implement GETL in CouchbaseMock.jar")
+    end
+  end
+
+  def test_multi_get_with_lock
+    if @mock.real?
+      connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+      connection.set(uniq_id(1), "foo1")
+      connection.set(uniq_id(2), "foo2")
+      assert_equal ["foo1", "foo2"], connection.get([uniq_id(1), uniq_id(2)], :lock => 1)
+      assert_raises Couchbase::Error::KeyExists do
+        connection.set(uniq_id(1), "bar")
+      end
+      assert_raises Couchbase::Error::KeyExists do
+        connection.set(uniq_id(2), "bar")
+      end
+    else
+      skip("implement GETL in CouchbaseMock.jar")
+    end
+  end
+
+  def test_multi_get_with_custom_locks
+    if @mock.real?
+      connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+      connection.set(uniq_id(1), "foo1")
+      connection.set(uniq_id(2), "foo2")
+      expected = {uniq_id(1) => "foo1", uniq_id(2) => "foo2"}
+      assert_equal expected, connection.get({uniq_id(1) => 1, uniq_id(2) => 2}, :lock => true)
+      assert_raises Couchbase::Error::KeyExists do
+        connection.set(uniq_id(1), "foo")
+      end
+      assert_raises Couchbase::Error::KeyExists do
+        connection.set(uniq_id(2), "foo")
+      end
+    else
+      skip("implement GETL in CouchbaseMock.jar")
+    end
+  end
+
+  def test_multi_get_result_hash_assembling
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    connection.set(uniq_id(1), "foo")
+    connection.set(uniq_id(2), "bar")
+
+    expected = {uniq_id(1) => "foo", uniq_id(2) => "bar"}
+    assert_equal expected, connection.get(uniq_id(1), uniq_id(2), :assemble_hash => true)
+  end
+end