"""Test for Placeholder UPC model."""

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

from sqlalchemy.exc import SQLAlchemyError

from product.models import placeholder_upc
from tests.testutils import db


@db.test_schema
def test_generate_upc():
    """Test insert query returns newly generated placeholder_upc."""
    expected_response = {'placeholder_upc': 1}
    inserted_row = placeholder_upc.generate_upc()

    assert inserted_row.status == 200
    assert inserted_row.message == expected_response


@patch('product.connectors.mysql._db_session')
def test_generate_upc_failure(db_session):
    """Test insert query returns an error when insert fails."""
    session = MagicMock()
    session.execute = MagicMock(side_effect=SQLAlchemyError())
    db_session.return_value = session

    inserted_row = placeholder_upc.generate_upc()

    assert inserted_row.status == 500