"""Lambda test module."""

import pytest

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

import index


@pytest.mark.parametrize(
    'count_jobs',
    [(1), (0), (10)]
)
@patch('index.call_datadog_with_metric')
@patch('index.boto3')
@patch('index.get_dms_ids')
def test_handler(mock_get_dms_ids, mock_boto3, mock_datadog, count_jobs):
    """Test Handler."""
    boto3_client_mock = MagicMock()
    boto3_client_mock.invoke.return_value = None

    mock_boto3.client.return_value = boto3_client_mock

    mock_get_dms_ids.return_value = [
        {
            'dms_master_master_id': 1,
            'order_types': [
                {
                    'order_type': 'ringtone',
                    'count_ready_to_deliver': 0
                },
                {
                    'order_type': 'release',
                    'count_ready_to_deliver': count_jobs
                }
            ]
        }
    ]

    index.handler(None, None)

    expected_calls = 1 if count_jobs > 0 else 0
    assert boto3_client_mock.invoke.call_count == expected_calls