"""Lambda test module."""

import pytest
import config
from gql import Client, gql
from gql.transport.requests import RequestsHTTPTransport


@pytest.fixture
def client():
    """Create graphQL test client."""
    transport = RequestsHTTPTransport(
        url=config.GRAPHQL_URL,
        use_json=True,
        headers={
            'Orchard-User-Id': config.OA_USER_ID
        },
        verify=True
    )

    client = Client(
        retries=3,
        transport=transport,
        fetch_schema_from_transport=False,
    )
    return client


def test_product_with_metadata(client):
    """Test query for a product with metadata."""
    query = gql(
        """
        {
            product(productId:"2749347", cached: false) {
                productId
                format
            }
        }
    """
    )
    expected = {'productId': 2749347, 'format': 'Full Length'}
    result = client.execute(query)
    product_metadata = result['product']
    assert product_metadata == expected


def test_product_no_metadata(client):
    """Test query for a product with no metadata."""
    query = gql(
        """
        {
            product(productId:"0", cached: false) {
                productId
                format
            }
        }
    """
    )
    expected = {'productId': 0, 'format': None}
    result = client.execute(query)
    product_metadata = result['product']
    assert product_metadata == expected