require 'test/unit'
require File.dirname(__FILE__) + '/test_helper'

class MemcacheTest < Test::Unit::TestCase
  PORTS = [11212, 11213, 11214, 11215, 11216, 11217]
  def setup
    init_memcache(*PORTS) do
      Memcache.new(:servers => PORTS.collect {|p| "localhost:#{p}"})
    end
  end

  def test_get_and_set
    100.times do |i|
      m.set(i.to_s, i)
      assert_equal i, m.get(i.to_s)
    end

    keys = (0..200).collect {|key| key.to_s}
    results = m.get(keys)
    assert_equal 100, results.size
    results.each do |key, value|
      assert_equal key.to_i, value
    end

    100.times do |i|
      m.set(i.to_s, i.to_s, :raw => true)
      assert_equal i.to_s, m.get(i.to_s, :raw => true)
    end

    results = m.get(keys ,:raw => true)
    assert_equal 100, results.size
    results.each do |key, value|
      assert_equal key, value
    end
  end

  def test_alternate_accessors
    m['baz'] = 24
    assert_equal 24, m['baz']
  end

  def test_add_and_replace
    100.times do |i|
      m.replace(i.to_s, [:foo, i])
      assert_equal nil, m.get(i.to_s)

      m.add(i.to_s, [:bar, i])
      assert_equal [:bar, i], m.get(i.to_s)

      m.replace(i.to_s, [:foo, i])
      assert_equal [:foo, i], m.get(i.to_s)

      m.add(i.to_s, [:baz, i])
      assert_equal [:foo, i], m.get(i.to_s)

      m.replace(i.to_s, 'blah', :raw => true)
      assert_equal 'blah', m.get(i.to_s, :raw => true)

      m.delete(i.to_s)
      assert_equal nil, m.get(i.to_s, :raw => true)

      m.add(i.to_s, 'homerun', :raw => true)
      assert_equal 'homerun', m.get(i.to_s, :raw => true)
    end
  end

  def test_append_and_prepend
    100.times do |i|
      m.append(i.to_s, 'doh!')
      assert_equal nil, m.read(i.to_s)

      m.write(i.to_s, 'bar')
      m.prepend(i.to_s, 'foo')
      assert_equal 'foobar', m.read(i.to_s)

      m.append(i.to_s, i.to_s)
      assert_equal "foobar#{i}", m.read(i.to_s)
    end
  end

  def test_get_or_set
    100.times do |i|
      m.get_or_set("foo#{i}", [i, :foo])
      assert_equal [i, :foo], m["foo#{i}"]

      m.get_or_set("foo#{i}") {raise}
      assert_equal [i, :foo], m["foo#{i}"]

      # Overwrite if changed.
      m.get_or_set("bar#{i}") do
        m.set("bar#{i}", [i, :foo])
        [i, :bar]
      end
      assert_equal [i, :bar], m["bar#{i}"]
    end
  end

  def test_get_or_add
    100.times do |i|
      m.get_or_add("foo#{i}", [:foo, i])
      assert_equal [:foo, i], m["foo#{i}"]

      m.get_or_add("foo#{i}") {raise}
      assert_equal [:foo, i], m["foo#{i}"]

      # Don't overwrite if changed.
      m.get_or_add("bar#{i}") do
        m.set("bar#{i}", [:foo, i])
        :bar
      end
      assert_equal [:foo, i], m["bar#{i}"]
    end
  end

  def test_add_or_get
    100.times do |i|
      assert_equal [i, :foo], m.add_or_get("FOO#{i}", [i, :foo])
      assert_equal nil,       m.add("FOO#{i}", [i, :bar])
      assert_equal [i, :foo], m.add_or_get("FOO#{i}", [i, :baz])
    end
  end

  def test_update
    100.times do |i|
      m.set("foo#{i}", [:foo, i])
      assert_equal [:foo, i], m["foo#{i}"]

      m.update("foo#{i}") do |list|
        list << i.to_s
        list << :bar
        list
      end
      assert_equal [:foo, i, i.to_s, :bar], m["foo#{i}"]
    end
  end

  def test_get_some
    100.times do |i|
      i = i * 2
      m.set(i.to_s, i)
      assert_equal i, m.get(i.to_s)
    end

    keys = (0...200).collect {|key| key.to_s}
    results = m.get_some(keys) do |missing_keys|
      assert_equal 100, missing_keys.size
      r = {}
      missing_keys.each do |key|
        r[key] = key.to_i
      end
      r
    end

    assert_equal 200, results.size
    results.each do |key, value|
      assert_equal key.to_i, value
    end
  end

  def test_get_some_with_validation
    100.times do |i|
      m.set(i.to_s, i.odd? ? -i : i)
    end
    keys = (0...100).collect {|key| key.to_s}

    results = m.get_some(keys, :validation => lambda {|k,v| k.to_i == v}) do |missing_keys|
      assert_equal 50, missing_keys.size
      r = {}
      missing_keys.each do |key|
        r[key] = key.to_i
      end
      r
    end
    assert_equal 100, results.size

    results.each do |key, value|
      assert_equal key.to_i, value
    end

    results = m.get_some(keys, :validation => lambda {|k,v| k.to_i == v}) do |missing_keys|
      flunk "no values should be invalid"
    end
  end

  def test_get_with_reset_expiry
    m.add('foo', 'quick brown fox', :expiry => 1)
    assert_equal 'quick brown fox', m.get('foo', :expiry => 2)
    sleep(1)
    assert_equal 'quick brown fox', m.get('foo')
  end

  def test_expiry
    100.times do |i|
      m.set("int#{i}", i, :expiry => 1)
      assert_equal i, m.get("int#{i}")

      m.set("time#{i}", i, :expiry => Time.now + 1)
      assert_equal i, m.get("time#{i}")
    end

    sleep 2

    100.times do |i|
      assert_equal nil, m.get("int#{i}")
      assert_equal nil, m.get("time#{i}")
    end
  end

  def test_in_namespace
    threads = []
    10.times do |i|
      m.in_namespace("_#{i}_") do
        10.times do |j|
          m.in_namespace("_#{j}_") do
            assert_equal nil, m.get('foo')
            m.set('foo', 'bar')
            assert_equal 'bar', m.get('foo')
          end
        end
      end
    end
  end

  def test_incr_and_decr
    m.write('foo', 0)

    m.incr('foo', 100)
    assert_equal 100, m.count('foo')

    m.decr('foo', 100)
    assert_equal 0, m.count('foo')

    m.incr('foo', 500)
    assert_equal 500, m.count('foo')

    m.decr('foo', 300)
    assert_equal 200, m.count('foo')

    m.decr('foo', 300)
    assert_equal 0, m.count('foo')
  end

  def test_flags
    m.set('foo', :foo, :flags => 43)
    assert_equal 43, m.get('foo').memcache_flags

    m.set('foo', 'foo', :raw => true, :flags => 43)
    assert_equal 43, m.get('foo', :raw => true).memcache_flags
  end

  def test_clone
    m.set('foo', 1)
    c = m.clone

    assert_not_equal m.servers.collect {|s| s.send(:socket)},
                     c.servers.collect {|s| s.send(:socket)}
  end

  def test_native_hashing
    n = Memcache.new(:servers => PORTS.collect {|p| "localhost:#{p}"}, :native => true)
    n.namespace = m.namespace

    1000.times do |i|
      i = i.to_s
      n.set(i, i, :raw => true)
      assert_equal i, m.get(i, :raw => true)
    end
  end
end