"""Tests for the TrackMasterRights model."""

from product.connectors.mysql import db_session
from product.models import track_master_rights as tmr_model
from tests.factories.track_master_rights import TrackMasterRightsFactory as tmr_factory
from tests.testutils import db


@db.test_schema
def test_delete():
    """Test delete with valid track_id."""
    db.seed_models(tmr_factory.build(master_rights_id=1, tuid=1))
    db.seed_models(tmr_factory.build(master_rights_id=2, tuid=2))
    db.seed_models(tmr_factory.build(master_rights_id=3, tuid=3))

    with db_session() as session:
        assert tmr_model.delete(track_ids=[1, 2], session=session)
        assert tmr_model.delete(track_ids=[3], session=session)


@db.test_schema
def test_invalid_delete():
    """Test delete with invalid tuid."""
    db.seed_models(tmr_factory.build(master_rights_id=1, tuid=1))
    db.seed_models(tmr_factory.build(master_rights_id=2, tuid=2))
    db.seed_models(tmr_factory.build(master_rights_id=3, tuid=3))

    missing_tuid = 4

    with db_session() as session:
        assert not tmr_model.delete(track_ids=[missing_tuid], session=session)
        assert not tmr_model.delete(track_ids=[missing_tuid, 1], session=session)