"""Tests for Placeholder UPC logic."""

from oto import response

from product.logic import placeholder_upc
from product.models import placeholder_upc as placeholder_upc_model


def test_generate_placeholder_upc(mocker):
    """Test successful response from Placeholder UPC model."""
    expected_message = {'placeholder_upc': 1}
    generated_upc_response = response.Response(message=expected_message)
    mocker.patch.object(
        placeholder_upc_model, 'generate_upc',
        return_value=generated_upc_response)

    generated_upc = placeholder_upc.generate_placeholder_upc()

    assert generated_upc.status == 200
    assert generated_upc.message == expected_message


def test_generate_placeholder_upc_failure(mocker):
    """Test unsuccessful response from Placeholder UPC model."""
    mocker.patch.object(
        placeholder_upc_model, 'generate_upc',
        return_value=response.create_fatal_response())

    generated_upc = placeholder_upc.generate_placeholder_upc()

    assert generated_upc.status == 500