"""Test for MySL Connection."""
from sqlalchemy.exc import NoSuchColumnError

from product_configuration.connectors import mysql


def test_wrap_db_errors():
    """Test wrap_db_errors with no errors."""
    @mysql.wrap_db_errors
    def no_exception():
        return True

    response = no_exception()
    assert response


def test_wrap_db_errors_with_error():
    """Test wrap_db_errors with a function that raises an exception."""
    @mysql.wrap_db_errors
    def has_exception():
        raise NoSuchColumnError('column not found')

    response = has_exception()
    assert response.status == 500