# 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::Unit::TestCase

  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
end