"""Test handler."""
from unittest.mock import MagicMock, patch

from botocore.errorfactory import ClientError
import index
import pytest


@patch('index.get_current_logger')
@patch('index.boto3')
def test_handler(
        mock_boto3,
        mock_logger,
        mock_event):
    """Test handler."""
    mock_s3_client = MagicMock()
    mock_boto3.client.return_value = mock_s3_client

    with pytest.raises(index.LockfileExistsException):
        index.handler(mock_event, {})

    mock_s3_client.head_object.assert_called_with(
        Bucket='qa-bulk-metadata-ingester',
        Key='locks/78929201921003_____9913971029993.lock'
    )


@patch('index.get_current_logger')
@patch('index.boto3')
def test_handler_2(
        mock_boto3,
        mock_logger,
        mock_event):
    """Test handler."""
    mock_s3_client = MagicMock()
    mock_s3_client.head_object.side_effect = ClientError({}, '')
    mock_boto3.client.return_value = mock_s3_client

    event = index.handler(mock_event, {})

    assert event == mock_event

    mock_s3_client.head_object.assert_called_with(
        Bucket='qa-bulk-metadata-ingester',
        Key='locks/78929201921003_____9913971029993.lock'
    )
    mock_s3_client.put_object.assert_called_with(
        Bucket='qa-bulk-metadata-ingester',
        Key='locks/78929201921003_____9913971029993.lock'
    )