"""Unit Tests for DistributionType model."""

from abacus_legacy_sync.models.distribution_type import DistributionType


def test_create_distribution_type():
    """Test creating a distribution_type in a database."""
    assert DistributionType.count() == 0
    distribution_type_id = 1
    distribution_type_name = 'Full Track'
    DistributionType.create(
        distribution_type_id=distribution_type_id,
        distribution_type_name=distribution_type_name,
    )
    result = DistributionType.query.all()

    assert len(result) == 1
    assert result[0].distribution_type_id == distribution_type_id
    assert result[0].distribution_type_name == distribution_type_name