flaskbb/tests/unit/utils/test_populate.py

import pytest

from flaskbb.utils.populate import create_default_groups, GROUPS
from flaskbb.user.models import Group


@pytest.fixture()
def default_groups():
    """Overwrite for the default fixture, in this case we don't want the default groups created."""
    return []


def test_create_default_groups(database):
    """Test that the default groups are created correctly."""

    assert Group.query.count() == 0

    create_default_groups()

    assert Group.query.count() == len(GROUPS)

    for key, attributes in GROUPS.items():
        group = Group.query.filter_by(name=key).first()

        for attribute, value in attributes.items():
            assert getattr(group, attribute) == value