"""Test RDS helpers."""
from unittest.mock import MagicMock, patch

from ddex_ingester_common.helpers.rds import run_rds_query


@patch('ddex_ingester_common.helpers.rds.mysql_connection')
def test_run_rds_query(mock_mysql_connection):
    """Test the run_rds_query helper."""
    query = 'SELECT * FROM table;'
    query_args = ('query', 'args')
    query_return = [{'some': 'thing'}, {'another': 'value'}]

    mock_rds_conn = MagicMock()
    mock_mysql_connection.return_value.__enter__.return_value = mock_rds_conn
    mock_cursor = MagicMock()
    mock_rds_conn.cursor.return_value.__enter__.return_value = mock_cursor
    mock_cursor.fetchall.return_value = query_return

    returned_value = run_rds_query(
        MagicMock(),
        'host',
        'db',
        'user',
        'pass',
        query,
        query_args,
    )

    assert returned_value == query_return
    mock_cursor.execute.assert_called_with(query, args=query_args)
    mock_cursor.fetchall.assert_called_with()