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

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_format


@db.test_schema
def test_get_distribution_format_media_format():
    """Test successful retrieval of distribution_format_media_format."""
    db.insert_distribution_format_media_formats()
    expected_response = [
        {'distribution_format_media_format_id': 1, 'name': 'Single'},
        {'distribution_format_media_format_id': 2, 'name': 'EP'},
        {'distribution_format_media_format_id': 3, 'name': 'LP'}]

    response = distribution_format_media_format.get_all_media_formats()

    assert response.status == 200
    assert response.message == expected_response


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

    response = distribution_format_media_format.get_all_media_formats()

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