76 lines
1.7 KiB
Go
76 lines
1.7 KiB
Go
// Copyright 2018 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 span_test
|
|
|
|
import (
|
|
"fmt"
|
|
"go/token"
|
|
"testing"
|
|
|
|
"golang.org/x/tools/internal/span"
|
|
)
|
|
|
|
var testdata = []struct {
|
|
uri string
|
|
content []byte
|
|
}{
|
|
{"/a.go", []byte(`
|
|
// file a.go
|
|
package test
|
|
`)},
|
|
{"/b.go", []byte(`
|
|
//
|
|
//
|
|
// file b.go
|
|
package test
|
|
`)},
|
|
}
|
|
|
|
var tokenTests = []span.Span{
|
|
span.New(span.FileURI("/a.go"), span.NewPoint(1, 1, 0), span.Point{}),
|
|
span.New(span.FileURI("/a.go"), span.NewPoint(3, 7, 20), span.NewPoint(3, 7, 20)),
|
|
span.New(span.FileURI("/b.go"), span.NewPoint(4, 9, 15), span.NewPoint(4, 13, 19)),
|
|
}
|
|
|
|
func TestToken(t *testing.T) {
|
|
fset := token.NewFileSet()
|
|
files := map[span.URI]*token.File{}
|
|
for _, f := range testdata {
|
|
file := fset.AddFile(f.uri, -1, len(f.content))
|
|
file.SetLinesForContent(f.content)
|
|
files[span.FileURI(f.uri)] = file
|
|
}
|
|
for _, test := range tokenTests {
|
|
f := files[test.URI()]
|
|
c := span.NewTokenConverter(fset, f)
|
|
checkToken(t, c, span.New(
|
|
test.URI(),
|
|
span.NewPoint(test.Start().Line(), test.Start().Column(), 0),
|
|
span.NewPoint(test.End().Line(), test.End().Column(), 0),
|
|
), test)
|
|
checkToken(t, c, span.New(
|
|
test.URI(),
|
|
span.NewPoint(0, 0, test.Start().Offset()),
|
|
span.NewPoint(0, 0, test.End().Offset()),
|
|
), test)
|
|
}
|
|
}
|
|
|
|
func checkToken(t *testing.T, c *span.TokenConverter, in, expect span.Span) {
|
|
rng, err := in.Range(c)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
gotLoc, err := rng.Span()
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
expected := fmt.Sprintf("%+v", expect)
|
|
got := fmt.Sprintf("%+v", gotLoc)
|
|
if expected != got {
|
|
t.Errorf("Expected %q got %q", expected, got)
|
|
}
|
|
}
|