"""Tests for enum utility."""

from abacus_common_logic.utils.enum import enum


def test_enum():
    """Test creating an enum."""
    FELLOWSHIP = enum(
        'FellowshipOfTheRing',
        ARAGORN='Aragorn II Elessar',
        BOROMIR='Boromir',
        GANDALF='Gandalf',
        GIMLI='Gimli',
        FRODO='Frodo Baggins',
        LEGOLAS='Legolas',
        MERRY='Meriadoc Brandybuck',
        PIPPIN='Peregrin Took',
        SAM='Samwise Gamgee',
    )

    assert FELLOWSHIP.ARAGORN == 'Aragorn II Elessar'
    assert FELLOWSHIP.BOROMIR == 'Boromir'
    assert FELLOWSHIP.GANDALF == 'Gandalf'
    assert FELLOWSHIP.GIMLI == 'Gimli'
    assert FELLOWSHIP.FRODO == 'Frodo Baggins'
    assert FELLOWSHIP.LEGOLAS == 'Legolas'
    assert FELLOWSHIP.MERRY == 'Meriadoc Brandybuck'
    assert FELLOWSHIP.PIPPIN == 'Peregrin Took'
    assert FELLOWSHIP.SAM == 'Samwise Gamgee'