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

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

from adjustment_file_import.connectors import s3


@patch('adjustment_file_import.connectors.s3.NamedTemporaryFile')
def test_download_file(mock_named_temporary_file):
    """Test download_file function."""
    bucket_name = 'bucket_name'
    file_path = 'file_path'

    mock_s3_client = MagicMock()
    s3.s3_client = mock_s3_client

    s3.download_file(bucket_name, file_path)

    mock_s3_client.download_file.assert_called_once_with(
        bucket_name,
        file_path,
        mock_named_temporary_file().name,
        ExtraArgs={
            'ExpectedBucketOwner': '437795906767'
        }
    )