jmoses-couchbase 1.3.6

data/test/test_arithmetic.rb

@@ -0,0 +1,185 @@
+# 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 TestArithmetic < MiniTest::Test
+
+  def setup
+    @mock = start_mock
+  end
+
+  def teardown
+    stop_mock(@mock)
+  end
+
+  def test_trivial_incr_decr
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    connection.set(uniq_id, 1)
+    val = connection.incr(uniq_id)
+    assert_equal 2, val
+    val = connection.get(uniq_id)
+    assert_equal 2, val
+
+    connection.set(uniq_id, 7)
+    val = connection.decr(uniq_id)
+    assert_equal 6, val
+    val = connection.get(uniq_id)
+    assert_equal 6, val
+  end
+
+  def test_it_fails_to_incr_decr_missing_key
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.incr(uniq_id(:missing))
+    end
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.decr(uniq_id(:missing))
+    end
+  end
+
+  def test_it_allows_to_make_increments_less_verbose_by_forcing_create_by_default
+    connection = Couchbase.connect(:hostname => @mock.host, :port => @mock.port,
+                                   :default_arithmetic_init => true)
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id)
+    end
+    assert_equal 0, connection.incr(uniq_id), "return value"
+    assert_equal 0, connection.get(uniq_id), "via get command"
+  end
+
+  def test_it_allows_to_setup_initial_value_during_connection
+    connection = Couchbase.connect(:hostname => @mock.host, :port => @mock.port,
+                                   :default_arithmetic_init => 10)
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id)
+    end
+    assert_equal 10, connection.incr(uniq_id), "return value"
+    assert_equal 10, connection.get(uniq_id), "via get command"
+  end
+
+  def test_it_allows_to_change_default_initial_value_after_connection
+    connection = Couchbase.connect(:hostname => @mock.host, :port => @mock.port)
+
+    assert_equal 0, connection.default_arithmetic_init
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.incr(uniq_id)
+    end
+
+    connection.default_arithmetic_init = 10
+    assert_equal 10, connection.default_arithmetic_init
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id)
+    end
+    assert_equal 10, connection.incr(uniq_id), "return value"
+    assert_equal 10, connection.get(uniq_id), "via get command"
+  end
+
+  def test_it_creates_missing_key_when_initial_value_specified
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    val = connection.incr(uniq_id(:missing), :initial => 5)
+    assert_equal 5, val
+    val = connection.incr(uniq_id(:missing), :initial => 5)
+    assert_equal 6, val
+    val = connection.get(uniq_id(:missing))
+    assert_equal 6, val
+  end
+
+  def test_it_uses_zero_as_default_value_for_missing_keys
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    val = connection.incr(uniq_id(:missing), :create => true)
+    assert_equal 0, val
+    val = connection.incr(uniq_id(:missing), :create => true)
+    assert_equal 1, val
+    val = connection.get(uniq_id(:missing))
+    assert_equal 1, val
+  end
+
+  def test_it_allows_custom_ttl
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    val = connection.incr(uniq_id(:missing), :create => true, :ttl => 1)
+    assert_equal 0, val
+    val = connection.incr(uniq_id(:missing), :create => true)
+    assert_equal 1, val
+    sleep(2)
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id(:missing))
+    end
+  end
+
+  def test_decrement_with_absolute_ttl
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    # absolute TTL: one second from now
+    exp = Time.now.to_i + 1
+    val = connection.decr(uniq_id, 12, :initial => 0, :ttl => exp)
+    assert_equal 0, val
+    assert_equal 0, connection.get(uniq_id)
+    sleep(2)
+    assert_raises(Couchbase::Error::NotFound) do
+      refute connection.get(uniq_id)
+    end
+  end
+
+  def test_it_allows_custom_delta
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    connection.set(uniq_id, 12)
+    val = connection.incr(uniq_id, 10)
+    assert_equal 22, val
+  end
+
+  def test_it_allows_to_specify_delta_in_options
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    connection.set(uniq_id, 12)
+    options = {:delta => 10}
+    val = connection.incr(uniq_id, options)
+    assert_equal 22, val
+  end
+
+  def test_multi_incr
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    connection.set(uniq_id(:foo) => 1, uniq_id(:bar) => 1)
+
+    assert_equal [2, 2],   connection.incr(uniq_id(:foo), uniq_id(:bar)).values.sort
+    assert_equal [12, 12], connection.incr(uniq_id(:foo), uniq_id(:bar), :delta => 10).values.sort
+    assert_equal [14, 15], connection.incr(uniq_id(:foo) => 2, uniq_id(:bar) => 3).values.sort
+  end
+
+  def test_multi_decr
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    connection.set(uniq_id(:foo) => 14, uniq_id(:bar) => 15)
+
+    assert_equal [12, 12], connection.decr(uniq_id(:foo) => 2, uniq_id(:bar) => 3).values.sort
+    assert_equal [2, 2],   connection.decr(uniq_id(:foo), uniq_id(:bar), :delta => 10).values.sort
+    assert_equal [1, 1],   connection.decr(uniq_id(:foo), uniq_id(:bar)).values.sort
+  end
+
+  def test_it_returns_cas_value_in_extended_mode
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    orig_cas = connection.set(uniq_id(:foo), 1)
+    val, cas = connection.incr(uniq_id(:foo), :extended => true)
+    assert_equal 2, val
+    assert cas.is_a?(Numeric), "CAS should be numeric value: #{cas.inspect}"
+    refute_equal orig_cas, cas
+  end
+end

data/test/test_async.rb

@@ -0,0 +1,316 @@
+# 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 TestAsync < MiniTest::Test
+
+  def setup
+    @mock = start_mock
+  end
+
+  def teardown
+    stop_mock(@mock)
+  end
+
+  def test_result_object_provides_enough_info
+    obj = Couchbase::Result.new
+    assert obj.respond_to?(:success?)
+    assert obj.respond_to?(:error)
+    assert obj.respond_to?(:key)
+    assert obj.respond_to?(:value)
+    assert obj.respond_to?(:node)
+    assert obj.respond_to?(:cas)
+    assert obj.respond_to?(:flags)
+  end
+
+  def test_it_requires_block_for_running_loop
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    refute connection.async?
+    assert_raises(LocalJumpError) do
+      connection.run
+    end
+    connection.run do |conn|
+      assert conn.async?
+    end
+  end
+
+  def test_it_resets_async_flag_when_raising_exception_from_callback
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    assert_raises(RuntimeError) do
+      connection.run do |conn|
+        conn.set(uniq_id, "foo") { raise }
+      end
+    end
+    refute connection.async?
+  end
+
+  def test_nested_async_get_set
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    connection.set(uniq_id, {"bar" => 1})
+    connection.set(uniq_id(:hit), 0)
+
+    connection.run do |conn|
+      conn.get(uniq_id) do
+        conn.get(uniq_id(:hit)) do |res|
+          conn.set(uniq_id(:hit), res.value + 1)
+        end
+      end
+    end
+
+    val = connection.get(uniq_id(:hit))
+    assert_equal 1, val
+  end
+
+  def test_nested_async_set_get
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    val = nil
+
+    connection.run do |conn|
+      conn.set(uniq_id, "foo") do
+        conn.get(uniq_id) do |res|
+          val = res.value
+        end
+      end
+    end
+
+    assert_equal "foo", val
+  end
+
+  def test_nested_async_touch_get
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    connection.set(uniq_id, "foo")
+    success = false
+    val = nil
+
+    connection.run do |conn|
+      conn.touch(uniq_id, :ttl => 1) do |res1|
+        success = res1.success?
+        conn.get(uniq_id) do |res2|
+          val = res2.value
+        end
+      end
+    end
+
+    assert success
+    assert_equal "foo", val
+    sleep(2)
+    assert_raises(Couchbase::Error::NotFound) do
+      connection.get(uniq_id)
+    end
+  end
+
+  def test_nested_async_delete_get
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    cas = connection.set(uniq_id, "foo")
+    success = false
+    val = :unknown
+
+    connection.run do |conn|
+      conn.delete(uniq_id, :cas => cas) do |res1|
+        success = res1.success?
+        conn.get(uniq_id, :quiet => true) do |res2|
+          val = res2.value
+        end
+      end
+    end
+
+    assert success
+    refute val
+  end
+
+  def test_nested_async_stats_set
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    stats = {}
+
+    connection.run do |conn|
+      conn.stats do |res1|
+        id = uniq_id(res1.node, res1.key)
+        stats[id] = false
+        conn.set(id, res1.value) do |res2|
+          stats[id] = res2.cas
+        end
+      end
+    end
+
+    stats.keys.each do |key|
+      assert stats[key].is_a?(Numeric)
+    end
+  end
+
+  def test_nested_async_flush_set
+    if @mock.real?
+      connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+      cas = connection.set(uniq_id, "foo")
+      res = {}
+
+      connection.run do |conn|
+        conn.flush do |res1|
+          assert res1.success?, "Expected: successful status code.\nActual: #{res1.error.inspect}"
+          id = uniq_id(res1.node)
+          res[id] = false
+          conn.set(id, true) do |res2|
+            res[id] = res2.cas
+          end
+        end
+      end
+
+      assert_raises(Couchbase::Error::NotFound) do
+        connection.get(uniq_id)
+      end
+      res.keys.each do |key|
+        assert res[key].is_a?(Numeric)
+        assert connection.get(key)
+      end
+    else
+      skip("REST FLUSH isn't implemented in CouchbaseMock.jar yet")
+    end
+  end
+
+  def test_nested_async_incr_get
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    cas = connection.set(uniq_id, 1)
+    val = nil
+
+    connection.run do |conn|
+      conn.incr(uniq_id) do
+        conn.get(uniq_id) do |res|
+          val = res.value
+        end
+      end
+    end
+
+    assert_equal 2, val
+  end
+
+  def test_it_doesnt_accept_callbacks_in_synchronous_mode
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    refute connection.async?
+
+    assert_raises(ArgumentError) { connection.add(uniq_id, "foo") {} }
+    assert_raises(ArgumentError) { connection.set(uniq_id, "foo") {} }
+    assert_raises(ArgumentError) { connection.replace(uniq_id, "foo") {} }
+    assert_raises(ArgumentError) { connection.get(uniq_id) {} }
+    assert_raises(ArgumentError) { connection.touch(uniq_id) {} }
+    assert_raises(ArgumentError) { connection.incr(uniq_id) {} }
+    assert_raises(ArgumentError) { connection.decr(uniq_id) {} }
+    assert_raises(ArgumentError) { connection.delete(uniq_id) {} }
+    assert_raises(ArgumentError) { connection.append(uniq_id, "bar") {} }
+    assert_raises(ArgumentError) { connection.prepend(uniq_id, "bar") {} }
+    assert_raises(ArgumentError) { connection.flush {} }
+    assert_raises(ArgumentError) { connection.stats {} }
+  end
+
+  def test_it_disallow_nested_run
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    assert_raises(Couchbase::Error::Invalid) do
+      connection.run do
+        connection.run do
+        end
+      end
+    end
+  end
+
+  def test_it_extends_timeout_in_async_mode_if_needed
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    connection.set(uniq_id, "foo")
+
+    connection.timeout = 100_000  # 100_000 us
+    connection.run do
+      connection.get(uniq_id) do |ret|
+        assert ret.success?
+        assert_equal "foo", ret.value
+      end
+      sleep(1.5)  # 1_500_000 us
+    end
+  end
+
+  def test_send_threshold
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+
+    sent = false
+    connection.run(:send_threshold => 100) do # 100 bytes
+      connection.set(uniq_id, "foo" * 100) {|r| sent = true}
+      assert sent
+    end
+  end
+
+  def test_asynchronous_connection
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :async => true)
+    refute connection.connected?, "new asynchronous connection must be disconnected"
+    connection.on_connect do |res|
+      assert res.success?, "on_connect called with error #{res.error.inspect}"
+      assert_same connection, res.bucket
+    end
+    connection.run {}
+    assert connection.connected?, "it should be connected after first run"
+  end
+
+  def test_it_calls_callback_immediately_if_connected_sync
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
+    assert connection.connected?, "connection wasn't established in sync mode"
+    called = false
+    connection.on_connect do |res|
+      assert res.success?, "on_connect called with error #{res.error.inspect}"
+      called = true
+    end
+    assert called, "the callback hasn't been called on set"
+    called = false
+    connection.on_connect do |res|
+      assert res.success?, "on_connect called with error #{res.error.inspect}"
+      called = true
+    end
+    refute called, "the callback must not be called on subsequent sets"
+  end
+
+  def test_it_calls_callback_immediately_if_connected_async
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :async => true)
+    refute connection.connected?, "new asynchronous connection must be disconnected"
+    called = false
+    connection.run {}
+    assert connection.connected?, "the connection must be established"
+    connection.run do
+      connection.on_connect do |res|
+        assert res.success?, "on_connect called with error #{res.error.inspect}"
+        called = true
+      end
+    end
+    assert called, "the callback hasn't been called on set"
+    called = false
+    connection.run do
+      connection.on_connect do |res|
+        assert res.success?, "on_connect called with error #{res.error.inspect}"
+        called = true
+      end
+    end
+    refute called, "the callback must not be called on subsequent sets"
+  end
+
+  def test_it_returns_error_if_user_start_work_on_disconnected_instance_outside_on_connect_callback
+    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port, :async => true)
+    refute connection.connected?, "new asynchronous connection must be disconnected"
+    error = nil
+    connection.on_error do |ex|
+      error = ex
+    end
+    connection.run do |c|
+      c.set("foo", "bar")
+    end
+    assert_instance_of(Couchbase::Error::Connect, error)
+  end
+end