"""Model Tests for Distribution Format Media."""

from unittest.mock import MagicMock
from unittest.mock import patch

from sqlalchemy.exc import SQLAlchemyError
from tests.testutils import db

from product_configuration.models import distribution_format_media


@db.test_schema
def test_get_distribution_format_media(distribution_format_media_types):
    """Test successful retrieval of distribution_format_media types."""
    db.insert_distribution_format_media_types()
    response = distribution_format_media.get_all_media_types()

    assert response.status == 200
    assert response.message == distribution_format_media_types


@patch('product_configuration.connectors.mysql._db_session')
def test_get_distribution_format_media_db_failure(
        db_session, sentry_sdk_capture_exception_mock):
    """Test fatal db error when performing query for media types."""
    session = MagicMock()
    session.query = MagicMock(side_effect=SQLAlchemyError())
    db_session.return_value = session

    response = distribution_format_media.get_all_media_types()

    assert response.status == 500
    sentry_sdk_capture_exception_mock.assert_called_once()