import pytest

from styles import STYLES, get_style
from styles.blink import BlinkStyle
from styles.simple import SimpleStyle


def test_registry_has_blink_and_simple():
    assert STYLES["blink"] is BlinkStyle
    assert STYLES["simple"] is SimpleStyle


def test_get_style_returns_class():
    assert get_style("blink") is BlinkStyle
    assert get_style("simple") is SimpleStyle


def test_get_style_unknown_raises():
    with pytest.raises(ValueError, match="Unknown alarm style"):
        get_style("nope")
