1
0
mirror of https://github.com/golang/go synced 2024-11-05 17:26:11 -07:00
go/internal/lsp/cmd/test/call_hierarchy.go
Danish Dua b793a1359e internal/lsp: add outgoing calls call hierarchy
* Adds outgoing calls call hierarchy for function declarations to gopls. Returns all call ranges and call items for functions/literals being called.
* Adds tests for outgoing call.
* Updates cmd to account for call ranges and call items being in different files for outgoing calls.
* Updates prepare call hierarchy to return declaration as root instead of cursor position.

Example:
Example shows https://github.com/golang/tools/blob/master/internal/lsp/source/call_hierarchy.go

Show Call Hierarchy View: https://imgur.com/a/DA5vc6l
Peek Call Hierarchy View: https://imgur.com/a/fuiG0Be

Note:
* While incoming calls for a function defined in an interface return references to that function, outgoing calls don't return anything since we don't know what implementation to return outgoing calls for.* Outgoing calls to function literals show as variable name used to define the literal, compared to <scope>.func() for incoming calls.

Change-Id: Ib8afbd8617675d12952db0b80170ada5988e90ab
Reviewed-on: https://go-review.googlesource.com/c/tools/+/248537
Run-TryBot: Danish Dua <danishdua@google.com>
TryBot-Result: Gobot Gobot <gobot@golang.org>
Reviewed-by: Heschi Kreinick <heschi@google.com>
2020-08-20 01:08:01 +00:00

86 lines
2.8 KiB
Go

// Copyright 2020 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package cmdtest
import (
"fmt"
"sort"
"strings"
"testing"
"golang.org/x/tools/internal/lsp/protocol"
"golang.org/x/tools/internal/lsp/tests"
"golang.org/x/tools/internal/span"
)
func (r *runner) CallHierarchy(t *testing.T, spn span.Span, expectedCalls *tests.CallHierarchyResult) {
collectCallSpansString := func(callItems []protocol.CallHierarchyItem) string {
var callSpans []string
for _, call := range callItems {
mapper, err := r.data.Mapper(call.URI.SpanURI())
if err != nil {
t.Fatal(err)
}
callSpan, err := mapper.Span(protocol.Location{URI: call.URI, Range: call.Range})
if err != nil {
t.Fatal(err)
}
callSpans = append(callSpans, fmt.Sprint(callSpan))
}
// to make tests deterministic
sort.Strings(callSpans)
return r.Normalize(strings.Join(callSpans, "\n"))
}
expectIn, expectOut := collectCallSpansString(expectedCalls.IncomingCalls), collectCallSpansString(expectedCalls.OutgoingCalls)
expectIdent := r.Normalize(fmt.Sprint(spn))
uri := spn.URI()
filename := uri.Filename()
target := filename + fmt.Sprintf(":%v:%v", spn.Start().Line(), spn.Start().Column())
got, stderr := r.NormalizeGoplsCmd(t, "call_hierarchy", target)
if stderr != "" {
t.Fatalf("call_hierarchy failed for %s: %s", target, stderr)
}
gotIn, gotIdent, gotOut := cleanCallHierarchyCmdResult(got)
if expectIn != gotIn {
t.Errorf("incoming calls call_hierarchy failed for %s expected:\n%s\ngot:\n%s", target, expectIn, gotIn)
}
if expectIdent != gotIdent {
t.Errorf("call_hierarchy failed for %s expected:\n%s\ngot:\n%s", target, expectIdent, gotIdent)
}
if expectOut != gotOut {
t.Errorf("outgoing calls call_hierarchy failed for %s expected:\n%s\ngot:\n%s", target, expectOut, gotOut)
}
}
// parses function URI and Range from call hierarchy cmd output to
// incoming, identifier and outgoing calls (returned in that order)
// ex: "identifier: function d at .../callhierarchy/callhierarchy.go:19:6-7" -> ".../callhierarchy/callhierarchy.go:19:6-7"
func cleanCallHierarchyCmdResult(output string) (incoming, ident, outgoing string) {
var incomingCalls, outgoingCalls []string
for _, out := range strings.Split(output, "\n") {
if out == "" {
continue
}
callLocation := out[strings.LastIndex(out, " ")+1:]
if strings.HasPrefix(out, "caller") {
incomingCalls = append(incomingCalls, callLocation)
} else if strings.HasPrefix(out, "callee") {
outgoingCalls = append(outgoingCalls, callLocation)
} else {
ident = callLocation
}
}
sort.Strings(incomingCalls)
sort.Strings(outgoingCalls)
incoming, outgoing = strings.Join(incomingCalls, "\n"), strings.Join(outgoingCalls, "\n")
return
}