"""Test SQS."""

from unittest.mock import patch

from src.logic import sqs


@patch('src.logic.sqs.boto3')
def test_sqs_add_product(mock_boto):
    """Test add product."""
    sqs.get_sqs_client.cache_clear()
    sqs.add_product(4, 5)
    mock_boto.client.return_value.send_message.assert_called_once_with(
        QueueUrl='foo',
        MessageBody='{"operation": "add", "review_queue_id": 4, "product_id": 5}'
    )
    sqs.get_sqs_client.cache_clear()


@patch('src.logic.sqs.boto3')
def test_sqs_update_product(mock_boto):
    """Test update product."""
    sqs.get_sqs_client.cache_clear()
    sqs.update_product(4, 5)
    mock_boto.client.return_value.send_message.assert_called_once_with(
        QueueUrl='foo',
        MessageBody='{"operation": "update", "review_queue_id": 4, "product_id": 5}'
    )
    sqs.get_sqs_client.cache_clear()


@patch('src.logic.sqs.boto3')
def test_sqs_delete_product(mock_boto):
    """Test delete product."""
    sqs.get_sqs_client.cache_clear()
    sqs.delete_product(5)
    mock_boto.client.return_value.send_message.assert_called_once_with(
        QueueUrl='foo',
        MessageBody='{"operation": "delete", "product_id": 5}'
    )
    sqs.get_sqs_client.cache_clear()