"""Test `GET /distribution-format-media`."""
from unittest.mock import MagicMock
from unittest.mock import patch

# import sentry_sdk
from sqlalchemy.exc import SQLAlchemyError
from tests.testutils import db
from tests.testutils.test_helpers import parse_json


@db.test_schema
def test_get_distribution_format_media(
        mocker, client, distribution_format_media_types):
    """Test successful response for distribution format media types request."""
    db.insert_distribution_format_media_types()

    result = client.get('/distribution-format-media')
    assert result.status_code == 200

    result_data = parse_json(result)
    assert result_data.get('items') == distribution_format_media_types


@patch('product_configuration.connectors.mysql._db_session')
def test_get_distribution_format_media_db_failure(
        db_session, client, sentry_sdk_capture_exception_mock):
    """Test db failure response for GET /distribution-format-media."""
    session = MagicMock()
    session.query = MagicMock(side_effect=SQLAlchemyError())
    db_session.return_value = session

    result = client.get('/distribution-format-media')

    assert result.status_code == 500
    sentry_sdk_capture_exception_mock.assert_called_once()