"""Test file for release_grid model."""

from datetime import datetime

from product_digital.connectors import mysql
from product_digital.models import release_grid
from tests.testutils import db


@db.test_schema
def test_delete(product_id):
    """Test delete release grid for given product."""
    new_release_grid = release_grid.ReleaseGrid(
        release_id=product_id, product_no='abc', create_time=datetime.today()
    )
    db.seed_models(new_release_grid)
    with mysql.db_session() as session:
        release_grid.delete(product_id, session)
        result = session.query(release_grid.ReleaseGrid).all()
    assert result == []