"""Tests for GET /distribution-format-media-format."""
from tests.testutils import db
from tests.testutils.test_helpers import parse_json


@db.test_schema
def test_get_distribution_format_media_formats(client):
    """Test get_product_by_product_id is successful with valid headers."""
    db.insert_distribution_format_media_formats()
    server_response = client.get('/distribution-format-media-format')
    server_data = parse_json(server_response)

    assert server_response.status_code == 200
    assert len(server_data['items']) == 3
    assert server_data['pagination']['total_records'] == 3