"""Lambda test module."""

import mock


@mock.patch('pymysql.connect')
def test_reportsdd_connection(mock_connect):
    """Test util.reportsdd_connection context manager."""
    import util
    with util.reportsdd_connection('db_host', 'user', 'passwd', 'db'):
        mock_connect.assert_called_with(
            'db_host', user='user', passwd='passwd', db='db',
            connect_timeout=5)
    mock_connect.return_value.close.assert_called_with()