class Cucumber::Formatter::Query::TestCaseStartedByTestCase

def attempt_by_test_case(test_case)
  raise TestCaseUnknownError, "No test case found for #{test_case.id} }. Known: #{@attempts_by_test_case_id.keys}" unless @attempts_by_test_case_id.key?(test_case.id)
  @attempts_by_test_case_id[test_case.id]
end

def initialize(config)
  @config = config
  config.on_event :test_case_created, &method(:on_test_case_created)
  config.on_event :test_case_started, &method(:on_test_case_started)
  @attempts_by_test_case_id = {}
  @test_case_started_id_by_test_case_id = {}
end

def on_test_case_created(event)
  @attempts_by_test_case_id[event.test_case.id] = 0
  @test_case_started_id_by_test_case_id[event.test_case.id] = nil
end

def on_test_case_started(event)
  @attempts_by_test_case_id[event.test_case.id] += 1
  @test_case_started_id_by_test_case_id[event.test_case.id] = @config.id_generator.new_id
end

def test_case_started_id_by_test_case(test_case)
  raise TestCaseUnknownError, "No test case found for #{test_case.id} }. Known: #{@test_case_started_id_by_test_case_id.keys}" unless @test_case_started_id_by_test_case_id.key?(test_case.id)
  @test_case_started_id_by_test_case_id[test_case.id]
end