couchbase-jruby-client 0.1.1

data/test/test_store.rb

@@ -0,0 +1,213 @@
+# 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_set
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    cas = connection.set(uniq_id, "bar")
+    assert(cas > 0)
+  end
+
+  def test_set_with_cas
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    cas1 = connection.set(uniq_id, "bar1")
+    assert cas1 > 0
+
+    assert_raises(Couchbase::Error::KeyExists) do
+      connection.set(uniq_id, "bar2", :cas => cas1+1)
+    end
+
+    cas2 = connection.set(uniq_id, "bar2", :cas => cas1)
+    assert cas2 > 0
+    refute_equal cas2, cas1
+
+    cas3 = connection.set(uniq_id, "bar3")
+    assert cas3 > 0
+    refute_equal cas3, cas2
+    refute_equal cas3, cas1
+  end
+
+  def test_add
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    cas1 = connection.add(uniq_id, "bar")
+    assert cas1 > 0
+
+    assert_raises(Couchbase::Error::KeyExists) do
+      connection.add(uniq_id, "bar")
+    end
+
+    assert_raises(Couchbase::Error::KeyExists) do
+      connection.add(uniq_id, "bar", :cas => cas1)
+    end
+  end
+
+  def test_replace
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.replace(uniq_id, "bar")
+    end
+
+    cas1 = connection.set(uniq_id, "bar")
+    assert cas1 > 0
+
+    connection.replace(uniq_id, "bar")
+  end
+
+  def test_acceptable_keys
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    cas = connection.set(uniq_id.to_sym, "bar")
+    assert cas > 0
+
+    cas = connection.set(uniq_id.to_s, "bar")
+    assert cas > 0
+
+    assert_raises(TypeError) do
+      connection.set(nil, "bar")
+    end
+
+    obj = {:foo => "bar", :baz => 1}
+    assert_raises(TypeError) do
+      connection.set(obj, "bar")
+    end
+
+    class << obj
+      alias :to_str :to_s
+    end
+
+    connection.set(obj, "bar")
+    assert cas > 0
+  end
+
+  def test_asynchronous_set
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    ret = nil
+    connection.run do |conn|
+      conn.set(uniq_id("1"), "foo1") {|res| ret = res}
+      conn.set(uniq_id("2"), "foo2") # ignore result
+    end
+
+    assert ret.is_a?(Couchbase::Result)
+    assert ret.success?
+    assert_equal uniq_id("1"), ret.key
+    assert_equal :set, ret.operation
+    assert ret.cas.is_a?(Numeric)
+  end
+
+  def test_it_raises_error_when_appending_or_prepending_to_missing_key
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    assert_raises(Couchbase::Error::NotStored) do
+      connection.append(uniq_id(:missing), "foo")
+    end
+
+    assert_raises(Couchbase::Error::NotStored) do
+      connection.prepend(uniq_id(:missing), "foo")
+    end
+  end
+
+  def test_append
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :default_format => :plain)
+
+    cas1 = connection.set(uniq_id, "foo")
+    assert cas1 > 0
+    cas2 = connection.append(uniq_id, "bar")
+    assert cas2 > 0
+    refute_equal cas2, cas1
+
+    val = connection.get(uniq_id)
+    assert_equal "foobar", val
+  end
+
+  def test_prepend
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :default_format => :plain)
+
+    cas1 = connection.set(uniq_id, "foo")
+    assert cas1 > 0
+    cas2 = connection.prepend(uniq_id, "bar")
+    assert cas2 > 0
+    refute_equal cas2, cas1
+
+    val = connection.get(uniq_id)
+    assert_equal "barfoo", val
+  end
+
+  def test_set_with_prefix
+    skip
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :key_prefix => "prefix:")
+    connection.set(uniq_id(:foo), "bar")
+    assert_equal "bar", connection.get(uniq_id(:foo))
+    expected = {uniq_id(:foo) => "bar"}
+    assert_equal expected, connection.get(uniq_id(:foo), :assemble_hash => true)
+
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :key_prefix => nil)
+    expected = {"prefix:#{uniq_id(:foo)}" => "bar"}
+    assert_equal expected, connection.get("prefix:#{uniq_id(:foo)}", :assemble_hash => true)
+  end
+
+  ArbitraryData = Struct.new(:baz)
+
+  def test_set_using_brackets
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    connection[uniq_id(1)] = "foo"
+    val = connection.get(uniq_id(1))
+    assert_equal "foo", val
+
+    # if RUBY_VERSION =~ /^1\.9/
+    #   eval <<-EOC
+    #   connection[uniq_id(3), :format => :marshal] = ArbitraryData.new("thing")
+    #   val = connection.get(uniq_id(3))
+    #   assert val.is_a?(ArbitraryData)
+    #   assert_equal "thing", val.baz
+    #   EOC
+    # end
+  end
+
+  def test_multi_store
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :default_format => :plain)
+    connection.add(uniq_id(:a) => "bbb", uniq_id(:z) => "yyy")
+    assert_equal ["bbb", "yyy"], connection.get(uniq_id(:a), uniq_id(:z))
+
+    # connection.prepend(uniq_id(:a) => "aaa", uniq_id(:z) => "xxx")
+    # assert_equal ["aaabbb", "xxxyyy"], connection.get(uniq_id(:a), uniq_id(:z))
+
+    # connection.append(uniq_id(:a) => "ccc", uniq_id(:z) => "zzz")
+    # assert_equal ["aaabbbccc", "xxxyyyzzz"], connection.get(uniq_id(:a), uniq_id(:z))
+
+    # connection.replace(uniq_id(:a) => "foo", uniq_id(:z) => "bar")
+    # assert_equal ["foo", "bar"], connection.get(uniq_id(:a), uniq_id(:z))
+
+    res = connection.set(uniq_id(:a) => "bar", uniq_id(:z) => "foo")
+    assert_equal ["bar", "foo"], connection.get(uniq_id(:a), uniq_id(:z))
+    assert res.is_a?(Hash)
+  end
+end

data/test/test_timer.rb

@@ -0,0 +1,43 @@
+# Author:: Couchbase <info@couchbase.com>
+# Copyright:: 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 TestTimer < MiniTest::Test
+
+  def setup
+    @mock = start_mock
+  end
+
+  def teardown
+    stop_mock(@mock)
+  end
+
+  def test_initialization
+    skip
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    num = 0
+    connection.run do
+      assert Couchbase::Timer.new(connection, 100)  { num += 1}
+      assert connection.create_timer(100)           { num += 1}
+
+      assert Couchbase::Timer.new(connection, 100, :periodic => true)  {|t| num += 1; t.cancel}
+      assert connection.create_periodic_timer(100)                     {|t| num += 1; t.cancel}
+    end
+    assert_equal 4, num
+  end
+end

data/test/test_touch.rb

@@ -0,0 +1,97 @@
+# 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 TestTouch < MiniTest::Test
+
+  def setup
+    @mock = start_mock
+  end
+
+  def teardown
+    stop_mock(@mock)
+  end
+
+  def test_trivial_touch
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    connection.set(uniq_id, "bar", :ttl => 1)
+    connection.touch(uniq_id, :ttl => 2)
+    sleep(1)
+    assert connection.get(uniq_id)
+    sleep(2)
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id)
+    end
+  end
+
+  def test_multi_touch
+    connection = Couchbase.new(:port => @mock.port)
+    connection.set(uniq_id(1), "bar")
+    connection.set(uniq_id(2), "baz")
+    ret = connection.touch(uniq_id(1) => 1, uniq_id(2) => 1)
+    assert ret[uniq_id(1)]
+    assert ret[uniq_id(2)]
+    sleep(2)
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id(1))
+    end
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id(2))
+    end
+  end
+
+  def test_it_uses_default_ttl_for_touch
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :default_ttl => 1)
+    connection.set(uniq_id, "bar", :ttl => 10)
+    connection.touch(uniq_id)
+    sleep(2)
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id)
+    end
+  end
+
+  def test_it_accepts_ttl_for_get_command
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    connection.set(uniq_id, "bar", :ttl => 10)
+    val = connection.get(uniq_id, :ttl => 1)
+    assert_equal "bar", val
+    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")
+
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.touch(uniq_id(:missing), :quiet => false)
+    end
+
+    val = connection.touch(uniq_id(:missing))
+    refute(val)
+
+    ret = connection.touch(uniq_id(1), uniq_id(:missing), uniq_id(2))
+    assert_equal true, ret[uniq_id(1)]
+    assert_equal false, ret[uniq_id(:missing)]
+    assert_equal true, ret[uniq_id(2)]
+  end
+
+end

data/test/test_unlock.rb

@@ -0,0 +1,121 @@
+# 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 TestUnlock < MiniTest::Test
+
+  def setup
+    @mock = start_mock
+  end
+
+  def teardown
+    stop_mock(@mock)
+  end
+
+  def test_trivial_unlock
+    if @mock.real?
+      connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+      connection.set(uniq_id, "foo")
+      _, _, cas = connection.get(uniq_id, :lock => true, :extended => true)
+      assert_raises Couchbase::Error::KeyExists do
+        connection.set(uniq_id, "bar")
+      end
+      assert connection.unlock(uniq_id, :cas => cas)
+      connection.set(uniq_id, "bar")
+    else
+      skip("GETL and UNL aren't implemented in CouchbaseMock.jar yet")
+    end
+  end
+
+  def test_alternative_syntax_for_single_key
+    if @mock.real?
+      connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+      connection.set(uniq_id, "foo")
+      _, _, cas = connection.get(uniq_id, :lock => true, :extended => true)
+      assert_raises Couchbase::Error::KeyExists do
+        connection.set(uniq_id, "bar")
+      end
+      assert connection.unlock(uniq_id, cas)
+      connection.set(uniq_id, "bar")
+    else
+      skip("GETL and UNL aren't implemented in CouchbaseMock.jar yet")
+    end
+  end
+
+  def test_multiple_unlock
+    skip
+    if @mock.real?
+      connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+      connection.set(uniq_id(1), "foo")
+      connection.set(uniq_id(2), "foo")
+      info = connection.get(uniq_id(1), uniq_id(2), :lock => true, :extended => true)
+      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
+      ret = connection.unlock(uniq_id(1) => info[uniq_id(1)][2],
+                              uniq_id(2) => info[uniq_id(2)][2])
+      assert ret[uniq_id(1)]
+      assert ret[uniq_id(2)]
+      connection.set(uniq_id(1), "bar")
+      connection.set(uniq_id(2), "bar")
+    else
+      skip("GETL and UNL aren't implemented in CouchbaseMock.jar yet")
+    end
+  end
+
+  def test_quiet_mode
+    skip
+    if @mock.real?
+      connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+      connection.set(uniq_id, "foo")
+      _, _, cas = connection.get(uniq_id, :lock => true, :extended => true)
+      assert_raises Couchbase::Error::NotFound do
+        connection.unlock(uniq_id(:missing), :cas => 0xdeadbeef)
+      end
+      keys = {
+        uniq_id => cas,
+        uniq_id(:missing) => 0xdeadbeef
+      }
+      ret = connection.unlock(keys, :quiet => true)
+      assert ret[uniq_id]
+      refute ret[uniq_id(:missing)]
+    else
+      skip("GETL and UNL aren't implemented in CouchbaseMock.jar yet")
+    end
+  end
+
+  def test_tmp_failure
+    if @mock.real?
+      connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+      cas1 = connection.set(uniq_id(1), "foo")
+      cas2 = connection.set(uniq_id(2), "foo")
+      connection.get(uniq_id(1), :lock => true) # get with lock will update CAS
+      assert_raises Couchbase::Error::TemporaryFail do
+        connection.unlock(uniq_id(1), cas1)
+      end
+      assert_raises Couchbase::Error::TemporaryFail do
+        connection.unlock(uniq_id(2), cas2)
+      end
+    else
+      skip("GETL and UNL aren't implemented in CouchbaseMock.jar yet")
+    end
+  end
+end

data/test/test_utils.rb

@@ -0,0 +1,58 @@
+# 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 TestUtils < MiniTest::Test
+
+  def setup
+    @mock = start_mock
+  end
+
+  def teardown
+    stop_mock(@mock)
+  end
+
+  def test_complex_startkey
+    assert_equal "all_docs?startkey=%5B%22Deadmau5%22%2C%22%22%5D", Couchbase::Utils.build_query("all_docs", :startkey =>  ["Deadmau5", ""])
+  end
+
+  def test_it_provides_enough_info_with_value_error
+    class << MultiJson
+      alias dump_good dump
+      def dump(obj)
+        raise ArgumentError, "cannot accept your object"
+      end
+    end
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    assert_raises(Couchbase::Error::ValueFormat) do
+      connection.set(uniq_id, "foo")
+    end
+    begin
+      connection.set(uniq_id, "foo")
+    rescue Couchbase::Error::ValueFormat => ex
+      assert_match /cannot accept your object/, ex.to_s
+      assert_instance_of ArgumentError, ex.inner_exception
+    end
+  ensure
+    class << MultiJson
+      undef dump
+      alias dump dump_good
+    end
+  end
+
+end

data/test/test_version.rb

@@ -0,0 +1,53 @@
+# 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 TestVersion < MiniTest::Test
+
+  def setup
+    @mock = start_mock
+  end
+
+  def teardown
+    stop_mock(@mock)
+  end
+
+  def test_sync_version
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    ver = connection.version
+    assert ver.is_a?(Hash)
+    assert_equal @mock.num_nodes, ver.size
+  end
+
+  def test_async_version
+    skip
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    ver = {}
+    connection.run do |conn|
+      conn.version do |ret|
+        assert ret.success?
+        ver[ret.node] = ret.value
+      end
+    end
+    assert_equal @mock.num_nodes, ver.size
+    ver.each do |node, v|
+      assert v.is_a?(String)
+    end
+  end
+
+end

data/test/test_view.rb

@@ -0,0 +1,94 @@
+# Author:: Mike Evans <mike@urlgonomics.com>
+# Copyright:: 2013 Urlgonomics LLC.
+# 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 TestView < MiniTest::Test
+
+  def setup
+    @mock = start_mock
+    @path = '_design/users/_view/by_age'
+    @cb = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    @cb.save_design_doc(design_doc)
+    { bob: 32, frank: 25, sam: 42, fred: 21 }.each_pair do |name, age|
+      @cb.set(name, { type: 'user', name: name, age: age })
+    end
+    @view = Couchbase::View.new(@cb, @path)
+  end
+
+  def teardown
+    stop_mock(@mock)
+    @cb.disconnect
+  end
+
+  def test_initialize
+    assert_equal 'users',  @view.design_doc
+    assert_equal 'by_age', @view.name
+  end
+
+  def test_simple_fetch
+    assert results = @view.fetch
+    assert results.is_a?(Couchbase::View::ArrayWithTotalRows)
+  end
+
+  def test_fetch_without_stale
+    assert results = @view.fetch(stale: false)
+    assert results.first.is_a?(Couchbase::ViewRow)
+    assert results.first.doc.nil?
+    assert_equal 4, results.total_rows
+    results.each do |result|
+      %w(bob frank sam fred).include?(result.key)
+    end
+  end
+
+  def test_fetch_with_docs
+    assert results = @view.fetch(stale: false, include_docs: true)
+    assert results.is_a?(Array)
+    assert results.first.doc.is_a?(Hash)
+  end
+
+  def test_fetch_with_block
+    refute @view.fetch(stale: false, include_docs: true) { |row|
+      assert row.is_a?(Couchbase::ViewRow)
+      assert row.doc['name'].is_a?(String)
+      assert row.doc['age'].is_a?(Fixnum)
+    }
+  end
+
+  def test_design_doc_access
+    assert results = @cb.design_docs['users'].by_age.to_a
+    assert results.first.is_a?(Couchbase::ViewRow)
+  end
+
+  def design_doc
+    {
+      '_id'      => '_design/users',
+      'language' => 'javascript',
+      'views' => {
+        'by_age' => {
+          'map' => <<-JS
+            function (doc, meta) {
+              if (doc.type && doc.type == 'user')
+                emit(meta.id, doc.age);
+            }
+          JS
+        }
+      }
+    }
+  end
+
+end