"""Tests for the TrackWriter model."""

from product.connectors.mysql import db_session
from product.models import track_writer as tw_model
from tests.factories.track_writer import TrackWriterFactory as tw_factory
from tests.testutils import db


@db.test_schema
def test_delete():
    """Test delete with valid upc."""
    db.seed_models(tw_factory.build(upc="195497264421"))

    with db_session() as session:
        assert tw_model.delete("195497264421", session)


@db.test_schema
def test_invalid_delete():
    """Test delete with invalid product id."""
    with db_session() as session:
        assert not tw_model.delete("195497264421", session)