import pytest

from app.dsp.enums import DSPClientStatus
from app.dsp.models import DSPClient

from tests.unit.helpers import create_model


class TestDSPClientQuery:
    @pytest.mark.db
    @pytest.mark.parametrize("status", [DSPClientStatus.active])
    def test_active_statuses_included(self, status: DSPClientStatus) -> None:
        create_model(DSPClient, status=status)

        assert DSPClient.query.active().count() == 1

    @pytest.mark.db
    @pytest.mark.parametrize(
        "status", [DSPClientStatus.paused, DSPClientStatus.inactive]
    )
    def test_inactive_statuses_excluded(self, status: DSPClientStatus) -> None:
        create_model(DSPClient, status=status)

        assert DSPClient.query.active().count() == 0