"""Unit tests for the MySQL connector."""

import pytest
from sqlalchemy.orm import Session

from contributor.connectors.mysql import ar_db_session


def test_ar_db_session_yields_session() -> None:
    with ar_db_session() as session:
        assert isinstance(session, Session)


def test_ar_db_session_rolls_back_on_exception() -> None:
    with pytest.raises(ValueError):
        with ar_db_session() as session:
            session.execute(__import__("sqlalchemy").text("SELECT 1"))
            raise ValueError("test error")