"""Function test for get product distribution."""
import json

from ows_product_physical.constant import header
from ows_product_physical.constant import service_name


def test_get_product_distribution(client, valid_get_header, db_release_id,
                                  ows_request_mocker, db_with_data,
                                  db_product_distribution):
    """Test getting a product distribution."""
    expected_response = {'items': [{
        'product_distribution_id': 1,
        'product_id': db_release_id,
        'distribute_to': 'JP'
    }]}

    ows_request_mocker.mock_response(
        method='head',
        service=service_name.OWS_PRODUCT,
        path='/{}/{}/product/{}'.format(
            valid_get_header[header.GRASS_ACCOUNT_TYPE],
            valid_get_header[header.GRASS_ACCOUNT_ID],
            db_release_id
        ))

    client_response = client.get(
        '/product-distribution/{}'.format(db_release_id),
        headers=valid_get_header
    )

    assert client_response.status_code == 200
    actual_data = json.loads(client_response.data.decode('utf-8'))

    assert actual_data == expected_response