# frozen_string_literal: true

module DiscourseEvent::TestHelper
  def trigger(event_name, *params)
    super(event_name, *params)

    if @events_trigger
      @events_trigger << { event_name: event_name, params: params }
    end
  end

  def track_events(event_name = nil, args: nil)
    @events_trigger = events_trigger = []
    yield
    @events_trigger = nil

    if event_name
      events_trigger = events_trigger.filter do |event|
        next if event[:event_name] != event_name
        next if args && event[:params] != args
        true
      end
    end

    events_trigger
  end

  def track(event_name, args: nil)
    events = track_events(event_name, args: args) { yield }
    events.first
  end

end

DiscourseEvent.singleton_class.prepend DiscourseEvent::TestHelper