"""Test file for rejection_notes model."""

from datetime import datetime

import pytest

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


@pytest.fixture
def release_approval_data():
    """Fake release_approval_id for testing."""
    return [123456]


@db.test_schema
def test_delete(release_approval_data):
    """Test delete rejection notes for given product."""
    new_rejection_notes = rejection_notes.RejectionNotes(
        release_approval_id=release_approval_data[0], table_name='release',
        comments='abc', corrected='N', date_added=datetime.today()
    )
    db.seed_models(new_rejection_notes)
    with mysql.db_session() as session:
        rejection_notes.delete(release_approval_data, session)
        result = session.query(rejection_notes.RejectionNotes).all()
    assert result == []