composiohq__composio/python/tests/test_openapi.py

"""Tests for :func:`composio.utils.openapi.function_signature_from_jsonschema`.

This builder is what the ``google_adk`` provider uses to attach a
``__signature__`` to wrapped tools, so a crash here fails tool wrapping. The
top-level "no type" property already falls back to ``Any``; these tests cover
the nested combiner options (``oneOf``/``anyOf``/``allOf``) that used to raise.
"""

import typing as t

import pytest

from composio.exceptions import InvalidSchemaError
from composio.utils.openapi import function_signature_from_jsonschema


def _annotation(prop: t.Dict[str, t.Any]) -> t.Any:
    params = function_signature_from_jsonschema({"properties": {"x": prop}})
    return params[0].annotation


class TestTypelessCombinerOptions:
    """A combiner option without a ``type`` key should default to Any, not raise."""

    @pytest.mark.unit
    @pytest.mark.schema
    def test_typeless_any_of_option_defaults_to_any(self):
        annotation = _annotation(
            {"anyOf": [{"description": "free form"}, {"type": "string"}]}
        )
        assert annotation == t.Union[t.Any, str]

    @pytest.mark.unit
    @pytest.mark.schema
    def test_typeless_one_of_option_defaults_to_any(self):
        annotation = _annotation(
            {"oneOf": [{"type": "string"}, {"description": "free form"}]}
        )
        assert annotation == t.Union[str, t.Any]

    @pytest.mark.unit
    @pytest.mark.schema
    def test_empty_all_of_resolves_to_any(self):
        assert _annotation({"allOf": []}) is t.Any


class TestEmptyCombinerLists:
    """An empty oneOf/anyOf has no options to union; fall back to Any."""

    @pytest.mark.unit
    @pytest.mark.schema
    def test_empty_any_of(self):
        assert _annotation({"anyOf": []}) is t.Any

    @pytest.mark.unit
    @pytest.mark.schema
    def test_empty_one_of(self):
        assert _annotation({"oneOf": []}) is t.Any


class TestExistingShapesUnchanged:
    """Shapes that already worked must keep resolving identically."""

    @pytest.mark.unit
    @pytest.mark.schema
    def test_nullable_any_of_is_optional(self):
        assert (
            _annotation({"anyOf": [{"type": "string"}, {"type": "null"}]})
            == t.Optional[str]
        )

    @pytest.mark.unit
    @pytest.mark.schema
    def test_multi_member_one_of(self):
        annotation = _annotation(
            {
                "oneOf": [
                    {"type": "string"},
                    {"type": "integer"},
                    {"type": "boolean"},
                    {"type": "number"},
                ]
            }
        )
        assert annotation == t.Union[str, int, bool, float]

    @pytest.mark.unit
    @pytest.mark.schema
    def test_enum(self):
        assert (
            _annotation({"type": "string", "enum": ["a", "b"]}) == t.Literal["a", "b"]
        )

    @pytest.mark.unit
    @pytest.mark.schema
    def test_array(self):
        assert (
            _annotation({"type": "array", "items": {"type": "string"}}) == t.List[str]
        )

    @pytest.mark.unit
    @pytest.mark.schema
    def test_object(self):
        assert _annotation({"type": "object"}) == t.Dict[str, t.Any]

    @pytest.mark.unit
    @pytest.mark.schema
    def test_typeless_property_is_any(self):
        assert _annotation({"description": "free form"}) is t.Any


class TestListValuedType:
    """A list-valued ``type`` (e.g. ``["string", "null"]``) resolves to a Union.

    JSON Schema Draft 2020-12 / OpenAPI 3.1 express a nullable field as a list of
    types rather than an ``anyOf``. A list is unhashable, so the older
    ``p_type in OPENAPI_TO_PYTHON`` membership test raised
    ``TypeError: unhashable type: 'list'`` and crashed tool wrapping.
    """

    @pytest.mark.unit
    @pytest.mark.schema
    def test_nullable_list_is_optional(self):
        assert _annotation({"type": ["string", "null"]}) == t.Optional[str]

    @pytest.mark.unit
    @pytest.mark.schema
    def test_single_member_list_resolves_to_that_type(self):
        assert _annotation({"type": ["integer"]}) is int

    @pytest.mark.unit
    @pytest.mark.schema
    def test_multiple_non_null_members_build_a_union(self):
        assert _annotation({"type": ["string", "integer"]}) == t.Union[str, int]

    @pytest.mark.unit
    @pytest.mark.schema
    def test_nullable_array_preserves_items_schema(self):
        assert (
            _annotation({"type": ["array", "null"], "items": {"type": "string"}})
            == t.Optional[t.List[str]]
        )

    @pytest.mark.unit
    @pytest.mark.schema
    @pytest.mark.parametrize("property_type", ["file", ["file", "null"]])
    def test_unrecognized_member_raises_invalid_schema(self, property_type: t.Any):
        with pytest.raises(InvalidSchemaError, match="Invalid property type file"):
            _annotation({"type": property_type})

    @pytest.mark.unit
    @pytest.mark.schema
    def test_empty_list_resolves_to_any(self):
        assert _annotation({"type": []}) is t.Any