Code.require_file "test_helper.exs", __DIR__

defmodule Exq.RedisQueueTest do
  use ExUnit.Case
  use Timex

  setup_all do
    TestRedis.setup
    on_exit fn ->
      TestRedis.teardown
    end
  end


  test "enqueue/dequeue single queue" do
    Exq.RedisQueue.enqueue(:testredis, "test", "default", "MyWorker", [])
    {deq, _} = Exq.RedisQueue.dequeue(:testredis, "test", "default")
    assert deq != :none
    {deq, _} = Exq.RedisQueue.dequeue(:testredis, "test", "default")
    assert deq == :none
  end

  test "enqueue/dequeue multi queue" do
    Exq.RedisQueue.enqueue(:testredis, "test", "default", "MyWorker", [])
    Exq.RedisQueue.enqueue(:testredis, "test", "myqueue", "MyWorker", [])
    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", ["default", "myqueue"]), 0) != :none
    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", ["default", "myqueue"]), 0) != :none
    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", ["default", "myqueue"]), 0) == :none
  end

  test "scheduler_dequeue single queue" do
    Exq.RedisQueue.enqueue_in(:testredis, "test", "default", 0, "MyWorker", [])
    Exq.RedisQueue.enqueue_in(:testredis, "test", "default", 0, "MyWorker", [])
    assert Exq.RedisQueue.scheduler_dequeue(:testredis, "test", ["default"]) == 2
    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", "default"), 0) != :none
    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", "default"), 0) != :none
    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", "default"), 0) == :none
  end

  test "scheduler_dequeue multi queue" do
    Exq.RedisQueue.enqueue_in(:testredis, "test", "default", -1, "MyWorker", [])
    Exq.RedisQueue.enqueue_in(:testredis, "test", "myqueue", -1, "MyWorker", [])
    assert Exq.RedisQueue.scheduler_dequeue(:testredis, "test", ["default", "myqueue"]) == 2
    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", ["default", "myqueue"]), 0) != :none
    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", ["default", "myqueue"]), 0) != :none
    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", ["default", "myqueue"]), 0) == :none
  end

  test "scheduler_dequeue enqueue_at" do
    Exq.RedisQueue.enqueue_at(:testredis, "test", "default", Time.now, "MyWorker", [])
    assert Exq.RedisQueue.scheduler_dequeue(:testredis, "test", ["default"]) == 1
    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", "default"), 0) != :none
    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", "default"), 0) == :none
  end

  test "scheduler_dequeue max_score" do
    Exq.RedisQueue.enqueue_in(:testredis, "test", "default", 300, "MyWorker", [])
    now = Time.now
    time1 = Time.add(now, Time.from(140, :secs))
    Exq.RedisQueue.enqueue_at(:testredis, "test", "default", time1, "MyWorker", [])
    time2 = Time.add(now, Time.from(150, :secs))
    Exq.RedisQueue.enqueue_at(:testredis, "test", "default", time2, "MyWorker", [])
    time2a = Time.add(now, Time.from(151, :secs))
    time2b = Time.add(now, Time.from(159, :secs))
    time3 = Time.add(now, Time.from(160, :secs))
    Exq.RedisQueue.enqueue_at(:testredis, "test", "default", time3, "MyWorker", [])
    time4 = Time.add(now, Time.from(160000001, :usecs))
    Exq.RedisQueue.enqueue_at(:testredis, "test", "default", time4, "MyWorker", [])
    time5 = Time.add(now, Time.from(300, :secs))

    assert Exq.Enqueuer.queue_size(:testredis, "test", "default") == "0"
    assert Exq.Enqueuer.queue_size(:testredis, "test", :scheduled) == "5"

    assert Exq.RedisQueue.scheduler_dequeue(:testredis, "test", ["default"], Exq.RedisQueue.time_to_score(time2a)) == 2
    assert Exq.RedisQueue.scheduler_dequeue(:testredis, "test", ["default"], Exq.RedisQueue.time_to_score(time2b)) == 0
    assert Exq.RedisQueue.scheduler_dequeue(:testredis, "test", ["default"], Exq.RedisQueue.time_to_score(time3)) == 1
    assert Exq.RedisQueue.scheduler_dequeue(:testredis, "test", ["default"], Exq.RedisQueue.time_to_score(time3)) == 0
    assert Exq.RedisQueue.scheduler_dequeue(:testredis, "test", ["default"], Exq.RedisQueue.time_to_score(time4)) == 1
    assert Exq.RedisQueue.scheduler_dequeue(:testredis, "test", ["default"], Exq.RedisQueue.time_to_score(time5)) == 1

    assert Exq.Enqueuer.queue_size(:testredis, "test", "default") == "5"
    assert Exq.Enqueuer.queue_size(:testredis, "test", :scheduled) == "0"

    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", "default"), 0) != :none
    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", "default"), 0) != :none
    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", "default"), 0) != :none
    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", "default"), 0) != :none
    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", "default"), 0) != :none
    assert elem(Exq.RedisQueue.dequeue(:testredis, "test", "default"), 0) == :none
  end

  test "full_key" do
    assert Exq.RedisQueue.full_key("exq","k1") == "exq:k1"
    assert Exq.RedisQueue.full_key("","k1") == "k1"
    assert Exq.RedisQueue.full_key(nil,"k1") == "k1"
  end

  test "creates and returns a jid" do
    jid = Exq.RedisQueue.enqueue(:testredis, "test", "default", "MyWorker", [])
    assert jid != nil

    {job_str, _} = Exq.RedisQueue.dequeue(:testredis, "test", "default")
    job = Poison.decode!(job_str, as: Exq.Job)
    assert job.jid == jid
  end

end