from app.ai.tool_schemas import get_all_tool_schemas, get_tool_schemas


def test_new_user_tools_are_onboarding_only():
    names = {schema["function"]["name"] for schema in get_tool_schemas("new_user")}
    assert names == {
        "about_falzh",
        "create_driver_account",
        "switch_to_driver",
        "switch_to_passenger",
    }


def test_driver_tools_exclude_passenger_booking_tools():
    names = {schema["function"]["name"] for schema in get_tool_schemas("driver")}
    assert "search_trips" not in names
    assert "select_trip" not in names
    assert "initiate_trip_action" in names
    assert "update_trip_field" in names
    assert "switch_to_passenger" in names


def test_passenger_tools_exclude_driver_management_tools():
    names = {schema["function"]["name"] for schema in get_tool_schemas("passenger")}
    assert "add_trip_by_driver" not in names
    assert "check_driver_info" not in names
    assert "switch_to_driver" in names
    assert "create_driver_account" in names


def test_all_tool_schemas_include_every_tool():
    names = {schema["function"]["name"] for schema in get_all_tool_schemas()}
    assert len(names) == 12