from . import MuseDashTestBase

# The worst case settings are DLC songs off, and enabling streamer mode.
# This ends up with only 25 valid songs that can be chosen.
# These tests ensure that this won't fail generation


class TestWorstCaseHighDifficulty(MuseDashTestBase):
    options = {
        "starting_song_count": 10,
        "dlc_packs": [],
        "streamer_mode_enabled": True,
        "song_difficulty_mode": 6,
        "song_difficulty_min": 11,
        "song_difficulty_max": 11,
    }


class TestWorstCaseMidDifficulty(MuseDashTestBase):
    options = {
        "starting_song_count": 10,
        "dlc_packs": [],
        "streamer_mode_enabled": True,
        "song_difficulty_mode": 6,
        "song_difficulty_min": 6,
        "song_difficulty_max": 6,
    }


class TestWorstCaseLowDifficulty(MuseDashTestBase):
    options = {
        "starting_song_count": 10,
        "dlc_packs": [],
        "streamer_mode_enabled": True,
        "song_difficulty_mode": 6,
        "song_difficulty_min": 1,
        "song_difficulty_max": 1,
    }
