| // 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" |
| "path" |
| "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`)}, |
| {"/c.go", []byte(` |
| // file c.go |
| package test`)}, |
| } |
| |
| var tokenTests = []span.Span{ |
| span.New(span.URIFromPath("/a.go"), span.NewPoint(1, 1, 0), span.Point{}), |
| span.New(span.URIFromPath("/a.go"), span.NewPoint(3, 7, 20), span.NewPoint(3, 7, 20)), |
| span.New(span.URIFromPath("/b.go"), span.NewPoint(4, 9, 15), span.NewPoint(4, 13, 19)), |
| span.New(span.URIFromPath("/c.go"), span.NewPoint(4, 1, 26), span.Point{}), |
| } |
| |
| 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.URIFromPath(f.uri)] = file |
| } |
| for _, test := range tokenTests { |
| f := files[test.URI()] |
| t.Run(path.Base(f.Name()), func(t *testing.T) { |
| checkToken(t, f, 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, f, span.New( |
| test.URI(), |
| span.NewPoint(0, 0, test.Start().Offset()), |
| span.NewPoint(0, 0, test.End().Offset()), |
| ), test) |
| }) |
| } |
| } |
| |
| func checkToken(t *testing.T, f *token.File, in, expect span.Span) { |
| rng, err := in.Range(f) |
| 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("For %v expected %q got %q", in, expected, got) |
| } |
| } |