1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
// Copyright 2021 CUE Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package astutil_test
import (
"fmt"
"path/filepath"
"testing"
"text/tabwriter"
"cuelang.org/go/cue/ast"
"cuelang.org/go/internal/astinternal"
"cuelang.org/go/internal/cuetxtar"
)
func TestResolve(t *testing.T) {
test := cuetxtar.TxTarTest{
Root: "./testdata/resolve",
Name: "resolve",
}
test.Run(t, func(t *cuetxtar.Test) {
// Use RawInstances because we want to allow imports
// without actually providing any dependencies. In general,
// Resolve is used just after parsing anyway, so lower level
// is more appropriate.
a := t.RawInstances()[0]
for _, f := range a.Files {
if filepath.Ext(f.Filename) != ".cue" {
continue
}
identMap := map[ast.Node]int{}
ast.Walk(f, func(n ast.Node) bool {
switch n.(type) {
case *ast.File, *ast.StructLit, *ast.Field, *ast.ImportSpec,
*ast.Ident, *ast.ForClause, *ast.LetClause, *ast.Alias:
identMap[n] = len(identMap) + 1
}
return true
}, nil)
// Resolve was already called.
base := filepath.Base(f.Filename)
b := t.Writer(base[:len(base)-len(".cue")])
w := tabwriter.NewWriter(b, 0, 4, 1, ' ', 0)
ast.Walk(f, func(n ast.Node) bool {
if x, ok := n.(*ast.Ident); ok {
fmt.Fprintf(w, "%d[%s]:\tScope: %d[%T]\tNode: %d[%s]\n",
identMap[x], astinternal.DebugStr(x),
identMap[x.Scope], x.Scope,
identMap[x.Node], astinternal.DebugStr(x.Node))
}
return true
}, nil)
w.Flush()
fmt.Fprint(b)
}
})
}
|
