couchbase 0.9.8 → 1.0.0

data/test/test_cas.rb

@@ -0,0 +1,59 @@
+# 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 TestCas < MiniTest::Unit::TestCase
+
+  def setup
+    @mock = start_mock
+  end
+
+  def teardown
+    stop_mock(@mock)
+  end
+
+  def test_compare_and_swap
+    connection = Couchbase.new(:port => @mock.port,
+                               :default_format => :document)
+    connection.set(test_id, {"bar" => 1})
+    connection.cas(test_id) do |val|
+      val["baz"] = 2
+      val
+    end
+    val = connection.get(test_id)
+    expected = {"bar" => 1, "baz" => 2}
+    assert_equal expected, val
+  end
+
+  def test_compare_and_swap_async
+    connection = Couchbase.new(:port => @mock.port,
+                               :default_format => :document)
+    connection.set(test_id, {"bar" => 1})
+    connection.run do |conn|
+      conn.cas(test_id) do |ret|
+        new_val = ret.value
+        new_val["baz"] = 2
+        new_val
+      end
+    end
+    val = connection.get(test_id)
+    expected = {"bar" => 1, "baz" => 2}
+    assert_equal expected, val
+  end
+
+end

data/test/test_couchbase.rb

@@ -1,9 +1,28 @@
+# 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 TestCouchbase < MiniTest::Unit::TestCase
+
   def test_that_it_create_instance_of_bucket
-    assert_instance_of Couchbase::Bucket, Couchbase.new('http://localhost:8091/pools/default')
+    with_mock do |mock|
+      assert_instance_of Couchbase::Bucket, Couchbase.new("http://localhost:#{mock.port}/pools/default")
+    end
   end
-end
-
 
+end

data/test/test_delete.rb

@@ -0,0 +1,63 @@
+# 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::Unit::TestCase
+
+  def setup
+    @mock = start_mock
+  end
+
+  def teardown
+    stop_mock(@mock)
+  end
+
+  def test_trivial_delete
+    connection = Couchbase.new(:port => @mock.port)
+    connection.set(test_id, "bar")
+    assert connection.delete(test_id)
+    refute connection.delete(test_id)
+  end
+
+  def test_delete_missing
+    connection = Couchbase.new(:port => @mock.port)
+    refute connection.delete(test_id(:missing))
+    connection.quiet = false
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.delete(test_id(:missing))
+    end
+    refute connection.delete(test_id(:missing), :quiet => true)
+  end
+
+  def test_delete_with_cas
+    connection = Couchbase.new(:port => @mock.port)
+    cas = connection.set(test_id, "bar")
+    missing_cas = cas - 1
+    assert_raises(Couchbase::Error::KeyExists) do
+      connection.delete(test_id, :cas => missing_cas)
+    end
+    assert connection.delete(test_id, :cas => cas)
+  end
+
+  def test_allow_fixnum_as_cas_parameter
+    connection = Couchbase.new(:port => @mock.port)
+    cas = connection.set(test_id, "bar")
+    assert connection.delete(test_id, cas)
+  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::Unit::TestCase
+
+  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(:port => @mock.port)
+    msg1 = {"author" => "foo", "message" => "hi all", "time" => "2012-01-12 11:29:09"}
+    key1 = test_id(genkey(msg1))
+    msg2 = {"author" => "foo", "message" => "hi all", "time" => "2012-01-12 11:29:30"}
+    key2 = test_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 = test_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 = test_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 = test_id(genkey(msg5))
+    assert_equal msg5, connection.get(key5)
+  end
+
+end

data/test/test_flush.rb

@@ -0,0 +1,49 @@
+# 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 TestFlush < MiniTest::Unit::TestCase
+
+  def setup
+    @mock = start_mock(:num_nodes => 7)
+  end
+
+  def teardown
+    stop_mock(@mock)
+  end
+
+  def test_trivial_flush
+    connection = Couchbase.new(:port => @mock.port)
+    assert connection.flush
+  end
+
+  def test_flush_with_block
+    connection = Couchbase.new(:port => @mock.port)
+    flushed = {}
+    on_node_flush = lambda{|res| flushed[res.node] = res.success?}
+    connection.run do |conn|
+      conn.flush(&on_node_flush)
+    end
+    assert_equal 7, flushed.size
+    flushed.each do |node, res|
+      assert node.is_a?(String)
+      assert res
+    end
+  end
+
+end

data/test/test_format.rb

@@ -0,0 +1,98 @@
+# 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::Unit::TestCase
+
+  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(:port => @mock.port)
+    assert_equal :document, connection.default_format
+    connection.set(test_id, orig_doc)
+    doc, flags, cas = connection.get(test_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
+    orig_doc = SkinyClass.new("Twoflower", "The tourist")
+    refute orig_doc.respond_to?(:to_s)
+    refute orig_doc.respond_to?(:to_json)
+
+    connection = Couchbase.new(:port => @mock.port, :default_format => :document)
+    assert_raises(Couchbase::Error::ValueFormat) do
+      connection.set(test_id, orig_doc)
+    end
+
+    class << orig_doc
+      def to_json
+        JSON.dump(:name => name, :role => role)
+      end
+    end
+    connection.set(test_id, orig_doc) # OK
+
+    class << orig_doc
+      undef to_json
+      def to_s
+        JSON.dump(:name => name, :role => role)
+      end
+    end
+    connection.set(test_id, orig_doc) # OK
+  end
+
+  def test_it_could_dump_arbitrary_class_using_marshal_format
+    orig_doc = ArbitraryClass.new("Twoflower", "The tourist")
+    connection = Couchbase.new(:port => @mock.port)
+    connection.set(test_id, orig_doc, :format => :marshal)
+    doc, flags, cas = connection.get(test_id, :extended => true)
+    assert_equal 0x01, flags & 0x11
+    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(:port => @mock.port, :default_format => :plain)
+    connection.set(test_id, "1")
+
+    assert_raises(Couchbase::Error::ValueFormat) do
+      connection.set(test_id, 1)
+    end
+
+    assert_raises(Couchbase::Error::ValueFormat) do
+      connection.set(test_id, {:foo => "bar"})
+    end
+  end
+
+end

data/test/test_get.rb

@@ -0,0 +1,236 @@
+# 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::Unit::TestCase
+
+  def setup
+    @mock = start_mock
+  end
+
+  def teardown
+    stop_mock(@mock)
+  end
+
+  def test_trivial_get
+    connection = Couchbase.new(:port => @mock.port)
+    connection.set(test_id, "bar")
+    val = connection.get(test_id)
+    assert_equal "bar", val
+  end
+
+  def test_extended_get
+    connection = Couchbase.new(:port => @mock.port)
+
+    orig_cas = connection.set(test_id, "bar")
+    val, flags, cas = connection.get(test_id, :extended => true)
+    assert_equal "bar", val
+    assert_equal 0x0, flags
+    assert_equal orig_cas, cas
+
+    orig_cas = connection.set(test_id, "bar", :flags => 0x1000)
+    val, flags, cas = connection.get(test_id, :extended => true)
+    assert_equal "bar", val
+    assert_equal 0x1000, flags
+    assert_equal orig_cas, cas
+  end
+
+  def test_multi_get
+    connection = Couchbase.new(:port => @mock.port)
+
+    connection.set(test_id(1), "foo1")
+    connection.set(test_id(2), "foo2")
+
+    val1, val2 = connection.get(test_id(1), test_id(2))
+    assert_equal "foo1", val1
+    assert_equal "foo2", val2
+  end
+
+  def test_multi_get_extended
+    connection = Couchbase.new(:port => @mock.port)
+
+    cas1 = connection.set(test_id(1), "foo1")
+    cas2 = connection.set(test_id(2), "foo2")
+
+    results = connection.get(test_id(1), test_id(2), :extended => true)
+    assert_equal ["foo1", 0x0, cas1], results[test_id(1)]
+    assert_equal ["foo2", 0x0, cas2], results[test_id(2)]
+  end
+
+  def test_missing_in_quiet_mode
+    connection = Couchbase.new(:port => @mock.port)
+    cas1 = connection.set(test_id(1), "foo1")
+    cas2 = connection.set(test_id(2), "foo2")
+
+    val = connection.get(test_id(:missing))
+    refute(val)
+    val = connection.get(test_id(:missing), :extended => true)
+    refute(val)
+
+    val1, missing, val2  = connection.get(test_id(1), test_id(:missing), test_id(2))
+    assert_equal "foo1", val1
+    refute missing
+    assert_equal "foo2", val2
+
+    results = connection.get(test_id(1), test_id(:missing), test_id(2), :extended => true)
+    assert_equal ["foo1", 0x0, cas1], results[test_id(1)]
+    refute results[test_id(:missing)]
+    assert_equal ["foo2", 0x0, cas2], results[test_id(2)]
+  end
+
+  def test_it_allows_temporary_quiet_flag
+    connection = Couchbase.new(:port => @mock.port, :quiet => false)
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(test_id(:missing))
+    end
+    refute connection.get(test_id(:missing), :quiet => true)
+  end
+
+  def test_missing_in_verbose_mode
+    connection = Couchbase.new(:port => @mock.port, :quiet => false)
+    connection.set(test_id(1), "foo1")
+    connection.set(test_id(2), "foo2")
+
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(test_id(:missing))
+    end
+
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(test_id(:missing), :extended => true)
+    end
+
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(test_id(1), test_id(:missing), test_id(2))
+    end
+
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(test_id(1), test_id(:missing), test_id(2), :extended => true)
+    end
+  end
+
+  def test_asynchronous_get
+    connection = Couchbase.new(:port => @mock.port)
+    cas = connection.set(test_id, "foo", :flags => 0x6660)
+    res = []
+
+    suite = lambda do |conn|
+      res.clear
+      conn.get(test_id) # ignore result
+      conn.get(test_id) {|ret| res << ret}
+      handler = lambda {|ret| res << ret}
+      conn.get(test_id, &handler)
+      assert_equal 3, conn.seqno
+    end
+
+    checks = lambda do
+      res.each do |r|
+        assert r.is_a?(Couchbase::Result)
+        assert r.success?
+        assert_equal test_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(:port => @mock.port)
+    connection.set(test_id(1), "foo")
+    connection.set(test_id(2), "bar")
+
+    res = {}
+    connection.run do |conn|
+      conn.get(test_id(1), test_id(2)) {|ret| res[ret.key] = ret.value}
+      assert_equal 2, conn.seqno
+    end
+
+    assert res[test_id(1)]
+    assert_equal "foo", res[test_id(1)]
+    assert res[test_id(2)]
+    assert_equal "bar", res[test_id(2)]
+  end
+
+  def test_asynchronous_get_missing
+    connection = Couchbase.new(:port => @mock.port)
+    connection.set(test_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(test_id(:missing1), &get_handler)
+      conn.get(test_id, test_id(:missing2), &get_handler)
+      assert 3, conn.seqno
+    end
+
+    connection.run(&suite)
+    assert_equal "foo", res[test_id]
+    assert res.has_key?(test_id(:missing1)) # handler was called with nil
+    refute res[test_id(:missing1)]
+    assert res.has_key?(test_id(:missing2))
+    refute res[test_id(:missing2)]
+    assert_empty missing
+
+    connection.quiet = false
+
+    connection.run(&suite)
+    refute res.has_key?(test_id(:missing1))
+    refute res.has_key?(test_id(:missing2))
+    assert_equal [test_id(:missing1), test_id(:missing2)], missing.sort
+    assert_equal "foo", res[test_id]
+  end
+
+  def test_get_using_brackets
+    connection = Couchbase.new(:port => @mock.port)
+
+    orig_cas = connection.set(test_id, "foo", :flags => 0x1100)
+
+    val = connection[test_id]
+    assert_equal "foo", val
+
+    if RUBY_VERSION =~ /^1\.9/
+      eval <<-EOC
+      val, flags, cas = connection[test_id, :extended => true]
+      assert_equal "foo", val
+      assert_equal 0x1100, flags
+      assert_equal orig_cas, cas
+      EOC
+    end
+  end
+end