"""Unit testcases for mysql utils."""
from datetime import datetime
from unittest.mock import MagicMock

from json_contract_file_import.constants.mysql_queries import \
    INSERT_JSON_CONTRACT_FILE_IMPORT_STATUS
from json_contract_file_import.helpers.mysql_utils import bulk_insert_query


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 = [
        datetime.now(),
        1,
        'Name',
        's3_path',
        's3_version',
        's3_report',
        'SUCCESS',
        'failed_entities',
        'errors',
        True,
        23,
        None
    ]
    bulk_insert_query(
        INSERT_JSON_CONTRACT_FILE_IMPORT_STATUS,
        [args],
        mock_mysql_conn
    )
    cursor_mock.executemany.assert_called_once_with(
        INSERT_JSON_CONTRACT_FILE_IMPORT_STATUS,
        [args]
    )