require 'test_helper'

class TestAssertionsTest < BeanCounter::TestCase

  setup do
    BeanCounter.reset!
    @tube_name = SecureRandom.uuid
    client.transmit("use #{@tube_name}")
    @message = SecureRandom.uuid
  end


  # Fuller testing of assertions handled by matcher and strategy tests.
  # Just make sure assertions work as expected at high level
  context 'simple job assertions' do

    should 'match any matching job when not given a count' do
      refute_enqueued(:body => @message)
      assert_enqueues(:body => @message) do
        2.times do
          client.transmit("put 0 0 120 #{@message.bytesize}\r\n#{@message}")
        end
      end
      assert_enqueued(:body => @message)
    end


    should 'match count exactly when given integer count' do
      refute_enqueued(:body => @message)
      assert_enqueues(:body => @message, :count => 1) do
        client.transmit("put 0 0 120 #{@message.bytesize}\r\n#{@message}")
      end
      assert_enqueued(:body => @message, :count => 1)
    end


    should 'match count to range when given range' do
      refute_enqueued(:body => @message)
      assert_enqueues(:body => @message, :count => 1..3) do
        2.times do
          client.transmit("put 0 0 120 #{@message.bytesize}\r\n#{@message}")
        end
      end
      assert_enqueued(:body => @message, :count => 1..3)
    end


    context 'failed assertion expectations' do

      setup do
        # Setting expectation for call count behaves strangely probably because
        # of expectation on assert. So count assertions manually
        @assert_calls = 0
        self.expects(:assert).at_least_once.with do |truth, message|
          @assert_calls += 1
          !truth
        end
      end


      should 'fail assertion when no matching jobs enqueued' do
        assert_enqueues(:body => /.*/) { }
        assert_enqueued(:body => /.*/)

        reset_expectations
        assert_equal 2, @assert_calls
      end


      should 'fail assertion when no matching jobs enqueued during block' do
        client.transmit("put 0 0 120 #{@message.bytesize}\r\n#{@message}")
        assert_enqueues(:body => @message) {}
        refute_enqueued(:body => %r[(?!#{@message})])

        reset_expectations
        # Refute calls assert, so assert_calls is incremented
        assert_equal 2, @assert_calls
      end


      should 'fail assertion when too few matching jobs enqueued' do
        assert_enqueues(:body => @message, :count => 2) do
          client.transmit("put 0 0 120 #{@message.bytesize}\r\n#{@message}")
        end
        assert_enqueued(:body => @message, :count => 2)

        reset_expectations
        assert_equal 2, @assert_calls
      end


      should 'fail assertion when too many matching jobs enqueued' do
        assert_enqueues(:body => @message, :count => 1) do
          2.times do
            client.transmit("put 0 0 120 #{@message.bytesize}\r\n#{@message}")
          end
        end
        assert_enqueued(:body => @message, :count => 1)

        reset_expectations
        assert_equal 2, @assert_calls
      end


      should 'fail assertion when number of matching jobs outside given range' do
        assert_enqueues(:body => @message, :count => 1..2) do
          3.times do
            client.transmit("put 0 0 120 #{@message.bytesize}\r\n#{@message}")
          end
        end
        assert_enqueued(:body => @message, :count => 1..2)

        reset_expectations
        assert_equal 2, @assert_calls
      end

    end

  end


  context 'tube assertions' do

    should 'pass assertion when matching tube found' do
      client.transmit("put 0 0 120 2\r\nxx")
      client.transmit("put 0 120 120 2\r\nxx")
      client.transmit("pause-tube #{@tube_name} 0")

      assert_tube({
        'name' => @tube_name,
        'cmd-pause' => 1..3,
        'current-jobs-ready' => 1,
        'current-jobs-delayed' => 1,
        'current-using' => 1,
      })
    end


    context 'failed assertion expectations' do

      setup do
        # Setting expectation for call count behaves strangely probably because
        # of expectation on assert. So count assertions manually
        @assert_calls = 0
        self.expects(:assert).at_least_once.with do |truth, message|
          @assert_calls += 1
          !truth
        end
      end


      should 'fail assertion when no matching tubes found' do
        assert_tube(:name => /#{SecureRandom.uuid}/)

        reset_expectations
        assert_equal 1, @assert_calls
      end

    end

  end


  context 'job refutations' do

    should 'pass refutation when no matching jobs are found' do
      message = SecureRandom.uuid
      refute_enqueues(:body => message) do
        client.transmit("put 0 0 120 #{@message.bytesize}\r\n#{@message}")
      end
      assert_enqueued(:body => @message)
      refute_enqueued(:body => message)
    end


    should 'pass refutation when if no matching jobs are enqueued during block' do
      client.transmit("put 0 0 120 #{@message.bytesize}\r\n#{@message}")
      refute_enqueues(:body => @message) {}
      assert_enqueued(:body => @message)
    end


    context 'failed refutation expectations' do

      should 'fail refutation when any matching jobs are enqueued' do
        # Setting expectation for call count behaves strangely probably because
        # of expectation on refute. So count refutaions manually
        @refute_calls = 0
        self.expects(:assert).at_least_once.with do |truth, message|
          @refute_calls += 1
          !truth
        end
        uuid = SecureRandom.uuid
        client.transmit("use #{uuid}")
        refute_enqueues(:body => /.*/) do
          client.transmit("put 0 0 120 #{uuid.bytesize}\r\n#{uuid}")
        end
        refute_enqueued(:body => /.*/)

        reset_expectations
        assert_equal 2, @refute_calls
      end

    end

  end


  context 'tube refutations' do

    should 'pass refutation when no matching tubes are found' do
      refute_tube(:name => SecureRandom.uuid)

      default_stats = client.transmit('stats-tube default')[:body]
      urgent = default_stats['current-jobs-urgent']
      refute_tube({
        'name' => 'default',
        'current-jobs-urgent' => (urgent + 5)..(urgent + 10),
      })
      watching = default_stats['current-watching']
      refute_tube({
        'name' => 'default',
        'current-watching' => watching + 10,
      })
    end


    context 'failed refutation expectations' do

      setup do
        # Setting expectation for call count behaves strangely probably because
        # of expectation on assert. So count assertions manually
        @assert_calls = 0
        self.expects(:assert).at_least_once.with do |truth, message|
          @assert_calls += 1
          !truth
        end
      end


      should 'fail assertion when matching tube found' do
        client.transmit("put 0 0 120 2\r\nxx")
        client.transmit("put 0 120 120 2\r\nxx")
        client.transmit("pause-tube #{@tube_name} 0")

        refute_tube({
          'name' => @tube_name,
          'cmd-pause' => 1..3,
          'current-jobs-ready' => 1,
          'current-jobs-delayed' => 1,
          'current-using' => 1,
        })

        reset_expectations
        assert_equal 1, @assert_calls
      end

    end

  end


  def reset_expectations
    # Must manually call mocha_teardown inside test body or expectation
    # when setting expectation on assert or refute, otherwise expectation
    # survives to the next test
    mocha_teardown
  end

end