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

from ddex_ingester_common.schemas.s3_schema import S3Schema

import index


@patch('index.load_ddex_json')
@patch('index.logger')
@patch('index.graphql_gateway.execute')
def test_handler(
        mock_graphql_execute,
        mock_current_logger,
        mock_load_ddex_json,
        mock_event):
    """Test the main handler."""
    mock_logger = MagicMock(name='info')
    mock_current_logger.return_value = mock_logger
    mock_load_ddex_json.return_value = mock_event
    mock_graphql_execute.return_value = True

    response = index.handler(mock_event, None)

    assert response


def test_parse_delivery_history(mock_event):
    """Test parse_delivery_history."""
    mock_context = S3Schema().load(mock_event)

    data = index.parse_delivery_history(
        mock_context.product.stores,
        '245234523454',
        25,
    )

    assert data == [
        {
            'storeId': 286,
            'dateDelivered': '2020-12-03',
            'encoderId': 25,
            'upc': '245234523454',
        },
        {
            'storeId': 1,
            'dateDelivered': '2020-12-02',
            'encoderId': 25,
            'upc': '245234523454',
        },
        {
            'storeId': 446,
            'dateDelivered': '2020-12-01',
            'encoderId': 25,
            'upc': '245234523454',
        }
    ]