"""Unit testcases for mysql utils."""

from unittest.mock import MagicMock

from adjustment_file_import.constants.mysql_queries import \
    INSERT_WORKSHEET_ADJUSTMENT_DETAIL
from adjustment_file_import.helpers.mysql_utils import bulk_insert_query
from adjustment_file_import.helpers.mysql_utils import execute_query


def test_execute_query():
    """Test execute_query function."""
    mock_mysql_conn = MagicMock()
    cursor_mock = MagicMock()
    mock_mysql_conn.cursor.return_value.__enter__.return_value = cursor_mock
    execute_query('select * from table', {'years': [2023]}, mock_mysql_conn)
    cursor_mock.fetchall.assert_called_once()


def test_bulk_insert_query():
    """Test bulk_insert_query function."""
    mock_mysql_conn = MagicMock()
    cursor_mock = MagicMock()
    mock_mysql_conn.cursor.return_value.__enter__.return_value = cursor_mock

    args = [1, '1234', '1234', 301, 302, 1, '-90.122',
            'USD', '1234567890', 'digital', None, None]
    bulk_insert_query(INSERT_WORKSHEET_ADJUSTMENT_DETAIL, args, mock_mysql_conn)
    cursor_mock.executemany.assert_called_once()