Function TestSession_ElicitationRequestSchema

go/session_test.go:1135–1217

(t *testing.T)

}

func TestSession_ElicitationRequestSchema(t *testing.T) {
	t.Run("nil content values are allowed", func(t *testing.T) {
		value, err := toRPCContent(nil)
		if err != nil {
			t.Fatalf("Expected nil content to be accepted, got %v", err)
		}
		if value != nil {
			t.Fatalf("Expected nil RPC content, got %T", value)
		}
	})

	t.Run("elicitation.requested passes full schema to handler", func(t *testing.T) {
		// Verify the schema extraction logic from handleBroadcastEvent
		// preserves type, properties, and required.
		properties := map[string]any{
			"name": map[string]any{"type": "string"},
			"age":  map[string]any{"type": "number"},
		}
		required := []string{"name", "age"}

		requestedSchema := ElicitationSchema{
			Properties: properties,
			Required:   required,
		}

		props := requestedSchema.Properties
		if props == nil {
			t.Fatal("Expected schema properties map")
		}
		if len(props) != 2 {
			t.Errorf("Expected 2 properties, got %d", len(props))
		}
		if len(requestedSchema.Required) != 2 {
			t.Errorf("Expected required [name, age], got %v", requestedSchema.Required)
		}
	})

	t.Run("schema without required omits required key", func(t *testing.T) {
		properties := map[string]any{
			"optional_field": map[string]any{"type": "string"},
		}

		requestedSchema := ElicitationSchema{
			Properties: properties,
		}

		if requestedSchema.Required != nil {
			t.Error("Expected Required to be nil when omitted")
		}
	})

	t.Run("schema conversion adds object type", func(t *testing.T) {
		requestedSchema := ElicitationSchema{
			Properties: map[string]any{
				"name": map[string]any{"type": "string"},
			},
		}