"""Tests for publishing obligation get endpoint."""

from oto import response

from ows_product_physical.constant import header
from ows_product_physical.logic import product_physical_tracks


def test_handler_get_publishing_obligations_for_product_success(
        client, valid_get_header, mocker):
    """Test get publishing-obligation handler success state."""
    mocker.patch.object(
        product_physical_tracks, 'get_publishing_obligations_for_product',
        return_value=response.Response({}),
        autospec=True)

    result = client.get(
        '/product/1234/tracks/publishing-obligation',
        headers=valid_get_header)

    assert result.status_code == 200
    assert (
        product_physical_tracks.get_publishing_obligations_for_product.called)


def test_handler_get_publishing_obligations_for_product_header_error(
        client, valid_get_header):
    """Test get publishing-obligation handler fails."""
    invalid_header = valid_get_header
    invalid_header[header.GRASS_ACCOUNT_TYPE] = 'banana'

    result = client.get(
        '/product/1234/tracks/publishing-obligation', headers=invalid_header)

    assert result.status_code == 400