"""Unit tests for deletion lambda."""

import mock
import index  # noqa


@mock.patch('streams_totals.delete')
@mock.patch('s3.delete_key')
@mock.patch('util.object_is_file')
@mock.patch('index.logger.info')
@mock.patch('util.extract_triggered_key')
@mock.patch('s3.key_exists')
def test_deletion_handler(
        s3_key_exists_mock, extract_triggered_key_mock, logger_info_mock,
        object_is_file_mock, s3_delete_key_mock, streams_totals_delete_mock):
    """Test deletion handler."""
    event_mock = mock.MagicMock()
    context_mock = mock.MagicMock()

    test_bucket = 'test_bucket'
    test_key = 'test_key'
    extract_triggered_key_mock.return_value = (test_bucket, test_key)
    object_is_file_mock.return_value = True
    s3_key_exists_mock.return_value = True

    index.handler(event_mock, context_mock)

    object_is_file_mock.assert_called_once_with(test_key)
    s3_delete_key_mock.assert_called_once_with(test_bucket, test_key)
    streams_totals_delete_mock.assert_called_once_with(
        's3://test_bucket/test_key')
    assert logger_info_mock.call_count == 3