"""Test file for release_approval_queue model."""

from datetime import datetime
from unittest.mock import MagicMock

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


@db.test_schema
def test_get_release_approval_ids(product_id):
    """Test get release_approval_queue data for given product."""
    new_release_approval_data = release_approval_queue.ReleaseApprovalQueue(
        release_id=product_id, status='rejected', admin_approval='N',
        last_updated=datetime.today(), date_submitted=datetime.today()
    )
    db.seed_models(new_release_approval_data)

    with mysql.db_session() as session:
        result = release_approval_queue.get_release_approval_ids(product_id, session)

    assert str(result[0].release_id) == product_id


@db.test_schema
def test_delete(product_id, mocker):
    """Test delete release_approval entry of the product."""
    mock_validation_response = MagicMock(status_code=200)
    mocker.patch.object(
        rejection_notes, 'delete',
        return_value=mock_validation_response
    )
    new_release_approval_data = release_approval_queue.ReleaseApprovalQueue(
        release_id=product_id, status='rejected', admin_approval='N',
        last_updated=datetime.today(), date_submitted=datetime.today()
    )
    db.seed_models(new_release_approval_data)

    with mysql.db_session() as session:
        release_approval_queue.delete(product_id, session)
        result = session.query(release_approval_queue.ReleaseApprovalQueue).all()

    assert rejection_notes.delete.called
    assert result == []