test/test_get.rb in couchbase-jruby-client-0.1.2 vs test/test_get.rb in couchbase-jruby-client-0.1.3

@@ -17,159 +17,140 @@
 
 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)
+    cb.set(uniq_id, "bar")
+    val = cb.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)
+    orig_cas = cb.set(uniq_id, "bar")
+    val, flags, cas = cb.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)
+    orig_cas = cb.set(uniq_id, "bar", :flags => 0x1000)
+    val, flags, cas = cb.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)
+    cb.set(uniq_id(1), "foo1")
+    cb.set(uniq_id(2), "foo2")
 
-    connection.set(uniq_id(1), "foo1")
-    connection.set(uniq_id(2), "foo2")
-
-    val1, val2 = connection.get(uniq_id(1), uniq_id(2))
+    val1, val2 = cb.get(uniq_id(1), uniq_id(2))
     assert_equal "foo1", val1
     assert_equal "foo2", val2
   end
 
   def test_multi_get_extended
     skip 'Looking like Java client doesnt support 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")
+    cas1 = cb.set(uniq_id(1), "foo1")
+    cas2 = cb.set(uniq_id(2), "foo2")
 
-    results = connection.get(uniq_id(1), uniq_id(2), :extended => true)
+    results = cb.get(uniq_id(1), uniq_id(2), :extended => true)
     assert_equal ["foo1", nil, cas1], results[uniq_id(1)]
     assert_equal ["foo2", nil, 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")
+    cb.set(uniq_id(1), "foo1")
+    cb.set(uniq_id(2), "foo2")
 
-    results = connection.get(uniq_id(1) => 1, uniq_id(2) => 1)
+    results = cb.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))
+      cb.get(uniq_id(1), uniq_id(2))
     end
-    assert connection.get(uniq_id(1), uniq_id(2), :quiet => true).compact.empty?
+    assert cb.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 = cb.set(uniq_id(1), "foo1")
+    cas2 = cb.set(uniq_id(2), "foo2")
 
-    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)
+    results = cb.get({uniq_id(1) => 1, uniq_id(2) => 1}, :extended => true)
     assert_equal ["foo1", nil, cas1], results[uniq_id(1)]
     assert_equal ["foo2", nil, 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")
+    cb.set(uniq_id, "foo1")
 
-    results = connection.get(uniq_id => 1)
+    results = cb.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)
+      cb.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")
+    cb.quiet = true
+    cas1 = cb.set(uniq_id(1), "foo1")
+    cas2 = cb.set(uniq_id(2), "foo2")
 
-    val = connection.get(uniq_id(:missing))
+    val = cb.get(uniq_id(:missing))
     refute(val)
-    val = connection.get(uniq_id(:missing), :extended => true)
+    val = cb.get(uniq_id(:missing), :extended => true)
     refute(val)
 
-    val1, missing, val2  = connection.get(uniq_id(1), uniq_id(:missing), uniq_id(2))
+    val1, missing, val2  = cb.get(uniq_id(1), uniq_id(:missing), uniq_id(2))
     assert_equal "foo1", val1
     refute missing
     assert_equal "foo2", val2
 
     # TODO: multi get with cas
-    # results = connection.get(uniq_id(1), uniq_id(:missing), uniq_id(2), :extended => true)
+    # results = cb.get(uniq_id(1), uniq_id(:missing), uniq_id(2), :extended => true)
     # assert_equal ["foo1", nil, cas1], results[uniq_id(1)]
     # refute results[uniq_id(:missing)]
     # assert_equal ["foo2", nil, cas2], results[uniq_id(2)]
+  ensure
+    cb.quiet = false
   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))
+      cb.get(uniq_id(:missing))
     end
-    refute connection.get(uniq_id(:missing), :quiet => true)
+    refute cb.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")
+    cb.set(uniq_id(1), "foo1")
+    cb.set(uniq_id(2), "foo2")
 
     assert_raises(Couchbase::Error::NotFound) do
-      connection.get(uniq_id(:missing))
+      cb.get(uniq_id(:missing))
     end
 
     assert_raises(Couchbase::Error::NotFound) do
-      connection.get(uniq_id(:missing), :extended => true)
+      cb.get(uniq_id(:missing), :extended => true)
     end
 
     assert_raises(Couchbase::Error::NotFound) do
-      connection.get(uniq_id(1), uniq_id(:missing), uniq_id(2))
+      cb.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)
+      cb.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")
+    cas = cb.set(uniq_id, "foo")
     res = []
 
     suite = lambda do |conn|
       res.clear
       conn.get(uniq_id) # ignore result
@@ -186,25 +167,24 @@
         assert_equal "foo", r.value
         # assert_equal cas, r.cas TODO: GetFuture does not hold CAS
       end
     end
 
-    connection.run(&suite)
+    cb.run(&suite)
     checks.call
 
-    connection.run{ suite.call(connection) }
+    cb.run{ suite.call(cb) }
     checks.call
   end
 
   def test_asynchronous_multi_get
     skip
-    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
-    connection.set(uniq_id(1), "foo")
-    connection.set(uniq_id(2), "bar")
+    cb.set(uniq_id(1), "foo")
+    cb.set(uniq_id(2), "bar")
 
     res = {}
-    connection.run do |conn|
+    cb.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)]
@@ -212,12 +192,11 @@
     assert_equal "bar", res[uniq_id(2)]
   end
 
   def test_asynchronous_get_missing
     skip
-    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
-    connection.set(uniq_id, "foo")
+    cb.set(uniq_id, "foo")
     res = {}
     missing = []
 
     get_handler = lambda do |ret|
       assert_equal :get, ret.operation
@@ -237,180 +216,158 @@
       missing.clear
       conn.get(uniq_id(:missing1), &get_handler)
       conn.get(uniq_id, uniq_id(:missing2), &get_handler)
     end
 
-    connection.run(&suite)
+    cb.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)
+    cb.quiet = true
+    cb.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 = cb.set(uniq_id, "foo", :flags => 0x1100)
 
-    orig_cas = connection.set(uniq_id, "foo", :flags => 0x1100)
-
-    val = connection[uniq_id]
+    val = cb[uniq_id]
     assert_equal "foo", val
 
     if RUBY_VERSION =~ /^1\.9/
       eval <<-EOC
-      val, flags, cas = connection[uniq_id, :extended => true]
+      val, flags, cas = cb[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
     skip "TODO: figure out nil storage"
-    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
 
-    orig_cas = connection.set(uniq_id, nil)
+    orig_cas = cb.set(uniq_id, nil)
     assert orig_cas.is_a?(Numeric)
 
-    refute connection.get(uniq_id)
+    refute cb.get(uniq_id)
     # doesn't raise NotFound exception
-    refute connection.get(uniq_id, :quiet => false)
+    refute cb.get(uniq_id, :quiet => false)
     # returns CAS
-    value, flags, cas = connection.get(uniq_id, :extended => true)
+    value, flags, cas = cb.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
     skip("zero length string")
-    connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
 
-    connection.set(uniq_id, "", :format => :document)
-    assert_equal "", connection.get(uniq_id)
+    cb.set(uniq_id, "", :format => :document)
+    assert_equal "", cb.get(uniq_id)
 
-    connection.set(uniq_id, "", :format => :plain)
-    assert_equal "", connection.get(uniq_id)
+    cb.set(uniq_id, "", :format => :plain)
+    assert_equal "", cb.get(uniq_id)
 
-    connection.set(uniq_id, "", :format => :marshal)
-    assert_equal "", connection.get(uniq_id)
+    cb.set(uniq_id, "", :format => :marshal)
+    assert_equal "", cb.get(uniq_id)
 
-    connection.set(uniq_id, nil, :format => :document)
-    assert_equal nil, connection.get(uniq_id, :quiet => false)
+    cb.set(uniq_id, nil, :format => :document)
+    assert_equal nil, cb.get(uniq_id, :quiet => false)
 
     assert_raises Couchbase::Error::ValueFormat do
-      connection.set(uniq_id, nil, :format => :plain)
+      cb.set(uniq_id, nil, :format => :plain)
     end
 
-    connection.set(uniq_id, nil, :format => :marshal)
-    assert_equal nil, connection.get(uniq_id, :quiet => false)
+    cb.set(uniq_id, nil, :format => :marshal)
+    assert_equal nil, cb.get(uniq_id, :quiet => false)
   end
 
   def test_format_forcing
     skip("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)
+    cb.set(uniq_id, '{"foo":"bar"}', :format => :plain)
+    value, flags, _ = cb.get(uniq_id, :extended => true)
     assert_equal '{"foo":"bar"}', value
     assert_equal 0x02, flags
 
-    value, flags, _ = connection.get(uniq_id, :extended => true, :format => :document)
+    value, flags, _ = cb.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")
+    cb.prepend(uniq_id, "NOT-A-JSON")
     assert_raises Couchbase::Error::ValueFormat do
-      connection.get(uniq_id, :format => :document)
+      cb.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")
+    cas = cb.set(uniq_id("foo"), "foo")
+    cb.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"))
+    assert_equal "foo", cb.get(uniq_id("foo"))
+    assert_equal ["foo"], cb.get([uniq_id("foo")])
+    assert_equal ["foo", "bar"], cb.get([uniq_id("foo"), uniq_id("bar")])
+    assert_equal ["foo", "bar"], cb.get(uniq_id("foo"), uniq_id("bar"))
     # expected = {uniq_id("foo") => ["foo", nil, cas]}
-    # assert_equal expected, connection.get([uniq_id("foo")], :extended => true)
+    # assert_equal expected, cb.get([uniq_id("foo")], :extended => true)
     assert_raises TypeError do
-      connection.get([uniq_id("foo"), uniq_id("bar")], [uniq_id("foo")])
+      cb.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")
+    cb.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")
+    assert_equal "foo", cb.get(uniq_id, :lock => 1)
+    assert_raises Couchbase::Error::KeyExists do
+      cb.set(uniq_id, "bar")
     end
+    sleep(2)
+    cb.set(uniq_id, "bar")
   end
 
   def test_multi_get_with_lock
     skip("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")
+    cb.set(uniq_id(1), "foo1")
+    cb.set(uniq_id(2), "foo2")
+    assert_equal ["foo1", "foo2"], cb.get([uniq_id(1), uniq_id(2)], :lock => 1)
+    assert_raises Couchbase::Error::KeyExists do
+      cb.set(uniq_id(1), "bar")
     end
+    assert_raises Couchbase::Error::KeyExists do
+      cb.set(uniq_id(2), "bar")
+    end
   end
 
   def test_multi_get_with_custom_locks
     skip("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")
+    cb.set(uniq_id(1), "foo1")
+    cb.set(uniq_id(2), "foo2")
+    expected = {uniq_id(1) => "foo1", uniq_id(2) => "foo2"}
+    assert_equal expected, cb.get({uniq_id(1) => 1, uniq_id(2) => 2}, :lock => true)
+    assert_raises Couchbase::Error::KeyExists do
+      cb.set(uniq_id(1), "foo")
     end
+    assert_raises Couchbase::Error::KeyExists do
+      cb.set(uniq_id(2), "foo")
+    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")
+    cb.set(uniq_id(1), "foo")
+    cb.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)
+    assert_equal expected, cb.get(uniq_id(1), uniq_id(2), :assemble_hash => true)
   end
 end