scottwillson/racing_on_rails

# frozen_string_literal: true

require File.expand_path("../../../test_case", __dir__)
require File.expand_path("../../../../app/models/competitions/bars/discipline", __dir__)

module Competitions
  module Bars
    # :stopdoc:
    # Used to only award bonus points for races of five or less, but now all races get equal points
    class DisciplineTest < Ruby::TestCase
      def setup
        @bar = Object.new
        @bar.extend Bars::Discipline
      end

      def test_discipline_for_road_race
        race = stub("race", discipline: "Road")
        assert_equal %w[Road Circuit].sort, @bar.disciplines_for(race).sort, "disciplines"
      end

      def test_discipline_for_track
        race = stub("race", discipline: "Track")
        assert_equal_enumerables ["Track"], @bar.disciplines_for(race), "disciplines"
      end

      def test_discipline_for_mtb
        race = stub("race", discipline: "Mountain Bike")
        assert_equal_enumerables ["Mountain Bike", "Downhill", "Super D"], @bar.disciplines_for(race), "disciplines"
      end

      def test_discipline_for_dh
        race = stub("race", discipline: "Downhill")
        assert_equal_enumerables ["Downhill"], @bar.disciplines_for(race), "disciplines"
      end
    end
  end
end