"""Unit testcases for db connector."""

from unittest.mock import MagicMock
from unittest.mock import patch

import pymysql

from adjustment_file_import.connectors.db import mysql_connection


@patch('adjustment_file_import.connectors.db.pymysql.connect')
def test_mysql_connection(mock_connect):
    """Test mysql_connection function."""
    mock_connection = MagicMock()
    mock_connect.return_value = mock_connection
    host = 'localhost'
    user = 'test_user'
    password = 'test_password'
    database = 'test_db'

    with mysql_connection(host, user, password, database):
        mock_connect.assert_called_once_with(
            host=host,
            user=user,
            passwd=password,
            db=database,
            connect_timeout=5,
            port=3306,
            cursorclass=pymysql.cursors.DictCursor,
            autocommit=False
        )
    mock_connect.return_value.close.assert_called()