"""Test add product to queue logic."""

from unittest.mock import patch

from src.logic.add_product_to_queue import add_product_to_queue


@patch('src.logic.add_product_to_queue.LambdaGraphQLConnector')
def test_add_product_to_queue(mock_gql):
    """Test add product to queue."""
    gql_create_response = {
        'createReviewQueueItem': {
            'reviewQueueId': 1
        }
    }
    mock_gql.return_value.execute_add_to_queue_mutation.return_value = gql_create_response
    add_response = add_product_to_queue(1, '1')
    assert add_response == 1