"""Test art_relation db util functions."""

from unittest.mock import Mock
from unittest.mock import patch

from api.utils import art_relations


@patch('api.utils.art_relations.MySQLdb')
def test__connection(mysql):
    """Test _connection function."""
    connection_object = Mock()
    mysql.connect.return_value = connection_object
    result = art_relations._connection()
    assert result == connection_object


@patch('api.utils.art_relations._connection')
@patch('api.utils.art_relations.database')
def test_context(database, _connection):
    """Test context function."""
    conn = Mock()
    _connection.return_value = conn

    art_relations.context(cursor_class='foo')
    assert _connection.called
    database.context.assert_called_once_with(conn, cursor_class='foo')