couchbase-jruby-client 0.1.0-java

data/test/test_delete.rb

@@ -0,0 +1,125 @@
+# 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
+    skip
+    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
+    skip
+    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
+    skip
+    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
+    skip
+    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
+    skip
+    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_format.rb

@@ -0,0 +1,161 @@
+# 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
+    skip
+    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
+    skip
+    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)
+    assert_equal "x\x01\xABVJ\xCB\xCFW\xB2RJJ,R\xAA\x05\0\x1Dz\x044", doc
+  end
+
+end