test_arithmetic.rb in couchbase-jruby-client-0.1.3

- old
+ new
@@ -17,161 +17,138 @@
 
 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)
+    cb.set(uniq_id, 1)
+    val = cb.incr(uniq_id)
     assert_equal 2, val
-    val = connection.get(uniq_id)
+    val = cb.get(uniq_id)
     assert_equal 2, val
 
-    connection.set(uniq_id, 7)
-    val = connection.decr(uniq_id)
+    cb.set(uniq_id, 7)
+    val = cb.decr(uniq_id)
     assert_equal 6, val
-    val = connection.get(uniq_id)
+    val = cb.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))
+      cb.incr(uniq_id(:missing))
     end
     assert_raises(Couchbase::Error::NotFound) do
-      connection.decr(uniq_id(:missing))
+      cb.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)
+    cb.default_arithmetic_init = true
     assert_raises(Couchbase::Error::NotFound) do
-      connection.get(uniq_id)
+      cb.get(uniq_id)
     end
-    assert_equal 0, connection.incr(uniq_id), "return value"
-    assert_equal 0, connection.get(uniq_id), "via get command"
+    assert_equal 0, cb.incr(uniq_id), "return value"
+    assert_equal 0, cb.get(uniq_id), "via get command"
+  ensure
+    cb.default_arithmetic_init = 0
   end
 
   def test_it_allows_to_setup_initial_value_during_connection
-    connection = Couchbase.connect(:hostname => @mock.host, :port => @mock.port,
-                                   :default_arithmetic_init => 10)
+    cb.default_arithmetic_init = 10
     assert_raises(Couchbase::Error::NotFound) do
-      connection.get(uniq_id)
+      cb.get(uniq_id)
     end
 
-    assert_equal 10, connection.incr(uniq_id), "return value"
-    assert_equal 10, connection.get(uniq_id), "via get command"
+    assert_equal 10, cb.incr(uniq_id), "return value"
+    assert_equal 10, cb.get(uniq_id), "via get command"
+  ensure
+    cb.default_arithmetic_init = 0
   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_equal 0, cb.default_arithmetic_init
     assert_raises(Couchbase::Error::NotFound) do
-      connection.incr(uniq_id)
+      cb.incr(uniq_id)
     end
 
-    connection.default_arithmetic_init = 10
-    assert_equal 10, connection.default_arithmetic_init
+    cb.default_arithmetic_init = 10
+    assert_equal 10, cb.default_arithmetic_init
     assert_raises(Couchbase::Error::NotFound) do
-      connection.get(uniq_id)
+      cb.get(uniq_id)
     end
-    assert_equal 10, connection.incr(uniq_id), "return value"
-    assert_equal 10, connection.get(uniq_id), "via get command"
+    assert_equal 10, cb.incr(uniq_id), "return value"
+    assert_equal 10, cb.get(uniq_id), "via get command"
+  ensure
+    cb.default_arithmetic_init = 0
   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)
+    val = cb.incr(uniq_id(:missing), :initial => 5)
     assert_equal 5, val
-    val = connection.incr(uniq_id(:missing), :initial => 5)
+    val = cb.incr(uniq_id(:missing), :initial => 5)
     assert_equal 6, val
-    val = connection.get(uniq_id(:missing))
+    val = cb.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)
+    val = cb.incr(uniq_id(:missing), :create => true)
     assert_equal 0, val
-    val = connection.incr(uniq_id(:missing), :create => true)
+    val = cb.incr(uniq_id(:missing), :create => true)
     assert_equal 1, val
-    val = connection.get(uniq_id(:missing))
+    val = cb.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)
+    val = cb.incr(uniq_id(:missing), :create => true, :ttl => 1)
     assert_equal 0, val
-    val = connection.incr(uniq_id(:missing), :create => true)
+    val = cb.incr(uniq_id(:missing), :create => true)
     assert_equal 1, val
     sleep(2)
     assert_raises(Couchbase::Error::NotFound) do
-      connection.get(uniq_id(:missing))
+      cb.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)
+    val = cb.decr(uniq_id, 12, :initial => 0, :ttl => exp)
     assert_equal 0, val
-    assert_equal 0, connection.get(uniq_id)
+    assert_equal 0, cb.get(uniq_id)
     sleep(3)
     assert_raises(Couchbase::Error::NotFound) do
-      connection.get(uniq_id)
+      cb.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)
+    cb.set(uniq_id, 12)
+    val = cb.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)
+    cb.set(uniq_id, 12)
     options = {:delta => 10}
-    val = connection.incr(uniq_id, options)
+    val = cb.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)
+    cb.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
+    assert_equal [2, 2],   cb.incr(uniq_id(:foo), uniq_id(:bar)).values.sort
+    assert_equal [12, 12], cb.incr(uniq_id(:foo), uniq_id(:bar), :delta => 10).values.sort
+    assert_equal [14, 15], cb.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)
+    cb.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
+    assert_equal [12, 12], cb.decr(uniq_id(:foo) => 2, uniq_id(:bar) => 3).values.sort
+    assert_equal [2, 2],   cb.decr(uniq_id(:foo), uniq_id(:bar), :delta => 10).values.sort
+    assert_equal [1, 1],   cb.decr(uniq_id(:foo), uniq_id(:bar)).values.sort
   end
 end