"""Tests for physical product logic."""

from unittest.mock import MagicMock

from ows_product_physical.logic import product_physical
from ows_product_physical.models import product_physical_supply_chain_info


def test_insert_supply_chain_defaults_success(
        monkeypatch, supply_chain_defults_mock_response,
        product_physical_supply_chain_info_record):
    """Assert Supply Chain default is set for product."""
    product_id = 123
    distribution_format_media_id = 2
    supply_chain_defaults = supply_chain_defults_mock_response
    monkeypatch.setattr(
        product_physical_supply_chain_info,
        'create',
        value=MagicMock(
            return_value=product_physical_supply_chain_info_record))

    result = product_physical._insert_supply_chain_defaults(
        distribution_format_media_id, supply_chain_defaults, product_id)

    assert result == [product_physical_supply_chain_info_record.message]


def test_insert_supply_chain_defaults_success_for_default(
        monkeypatch, supply_chain_defults_mock_response,
        product_physical_supply_chain_info_default_record):
    """Assert Supply Chain default is set for product with default values."""
    product_id = 123
    distribution_format_media_id = 7
    supply_chain_defaults = supply_chain_defults_mock_response
    monkeypatch.setattr(
        product_physical_supply_chain_info,
        'create',
        value=MagicMock(
            return_value=product_physical_supply_chain_info_default_record))

    result = product_physical._insert_supply_chain_defaults(
        distribution_format_media_id, supply_chain_defaults, product_id)

    assert result ==\
        [product_physical_supply_chain_info_default_record.message]