require "test_helper"
require "helpers/mocked_instrumentation_service"

class HTML::PipelineTest < Test::Unit::TestCase
  Pipeline = HTML::Pipeline
  class TestFilter
    def self.call(input, context, result)
      input.reverse
    end
  end

  def setup
    @context = {}
    @result_class = Hash
    @pipeline = Pipeline.new [TestFilter], @context, @result_class
  end

  def test_filter_instrumentation
    service = MockedInstrumentationService.new
    events = service.subscribe "call_filter.html_pipeline"
    @pipeline.instrumentation_service = service
    filter(body = "hello")
    event, payload, res = events.pop
    assert event, "event expected"
    assert_equal "call_filter.html_pipeline", event
    assert_equal TestFilter.name, payload[:filter]
    assert_equal @pipeline.class.name, payload[:pipeline]
    assert_equal body.reverse, payload[:result][:output]
  end

  def test_pipeline_instrumentation
    service = MockedInstrumentationService.new
    events = service.subscribe "call_pipeline.html_pipeline"
    @pipeline.instrumentation_service = service
    filter(body = "hello")
    event, payload, res = events.pop
    assert event, "event expected"
    assert_equal "call_pipeline.html_pipeline", event
    assert_equal @pipeline.filters.map(&:name), payload[:filters]
    assert_equal @pipeline.class.name, payload[:pipeline]
    assert_equal body.reverse, payload[:result][:output]
  end

  def test_default_instrumentation_service
    service = 'default'
    Pipeline.default_instrumentation_service = service
    pipeline = Pipeline.new [], @context, @result_class
    assert_equal service, pipeline.instrumentation_service
  ensure
    Pipeline.default_instrumentation_service = nil
  end

  def test_setup_instrumentation
    assert_nil @pipeline.instrumentation_service

    service = MockedInstrumentationService.new
    events = service.subscribe "call_pipeline.html_pipeline"
    @pipeline.setup_instrumentation name = 'foo', service

    assert_equal service, @pipeline.instrumentation_service
    assert_equal name, @pipeline.instrumentation_name

    filter(body = 'foo')

    event, payload, res = events.pop
    assert event, "expected event"
    assert_equal name, payload[:pipeline]
    assert_equal body.reverse, payload[:result][:output]
  end

  def filter(input)
    @pipeline.call(input)
  end
end