from ormar.models.helpers.models import group_related_list


def test_group_related_list():
    given = [
        "friends__least_favourite_game",
        "least_favourite_game",
        "friends",
        "favourite_game",
        "friends__favourite_game",
    ]
    expected = {
        "least_favourite_game": [],
        "favourite_game": [],
        "friends": ["favourite_game", "least_favourite_game"],
    }
    assert group_related_list(given) == expected