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
|
// Copyright 2023 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 tokeninternal_test
import (
"fmt"
"go/token"
"strings"
"testing"
"golang.org/x/tools/internal/tokeninternal"
)
func TestAddExistingFiles(t *testing.T) {
fset := token.NewFileSet()
check := func(descr, want string) {
t.Helper()
if got := fsetString(fset); got != want {
t.Errorf("%s: got %s, want %s", descr, got, want)
}
}
fileA := fset.AddFile("A", -1, 3)
fileB := fset.AddFile("B", -1, 5)
_ = fileB
check("after AddFile [AB]", "{A:1-4 B:5-10}")
tokeninternal.AddExistingFiles(fset, nil)
check("after AddExistingFiles []", "{A:1-4 B:5-10}")
fileC := token.NewFileSet().AddFile("C", 100, 5)
fileD := token.NewFileSet().AddFile("D", 200, 5)
tokeninternal.AddExistingFiles(fset, []*token.File{fileC, fileA, fileD, fileC})
check("after AddExistingFiles [CADC]", "{A:1-4 B:5-10 C:100-105 D:200-205}")
fileE := fset.AddFile("E", -1, 3)
_ = fileE
check("after AddFile [E]", "{A:1-4 B:5-10 C:100-105 D:200-205 E:206-209}")
}
func fsetString(fset *token.FileSet) string {
var buf strings.Builder
buf.WriteRune('{')
sep := ""
fset.Iterate(func(f *token.File) bool {
fmt.Fprintf(&buf, "%s%s:%d-%d", sep, f.Name(), f.Base(), f.Base()+f.Size())
sep = " "
return true
})
buf.WriteRune('}')
return buf.String()
}
|
