File: token_test.go

package info (click to toggle)
golang-golang-x-tools 1%3A0.5.0%2Bds-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, bookworm-backports
  • size: 16,592 kB
  • sloc: javascript: 2,011; asm: 1,635; sh: 192; yacc: 155; makefile: 52; ansic: 8
file content (57 lines) | stat: -rw-r--r-- 1,469 bytes parent folder | download
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
// Copyright 2022 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 lsppos_test

import (
	"go/token"
	"testing"

	. "golang.org/x/tools/gopls/internal/lsp/lsppos"
	"golang.org/x/tools/gopls/internal/lsp/protocol"
)

func makeTokenMapper(content []byte) (*TokenMapper, *token.File) {
	file := token.NewFileSet().AddFile("p.go", -1, len(content))
	file.SetLinesForContent(content)
	return NewTokenMapper(content, file), file
}

func TestInvalidPosition(t *testing.T) {
	content := []byte("a𐐀b\r\nx\ny")
	m, _ := makeTokenMapper(content)

	for _, pos := range []token.Pos{-1, 100} {
		posn, ok := m.Position(pos)
		if ok {
			t.Errorf("Position(%d) = %v, want error", pos, posn)
		}
	}
}

func TestTokenPosition(t *testing.T) {
	for _, test := range tests {
		m, f := makeTokenMapper([]byte(test.content))
		pos := token.Pos(f.Base() + test.offset())
		got, ok := m.Position(pos)
		if !ok {
			t.Error("invalid position for", test.substrOrOffset)
			continue
		}
		want := protocol.Position{Line: uint32(test.wantLine), Character: uint32(test.wantChar)}
		if got != want {
			t.Errorf("Position(%d) = %v, want %v", pos, got, want)
		}
		gotRange, err := m.Range(token.Pos(f.Base()), pos)
		if err != nil {
			t.Fatal(err)
		}
		wantRange := protocol.Range{
			End: want,
		}
		if gotRange != wantRange {
			t.Errorf("Range(%d) = %v, want %v", pos, got, want)
		}
	}
}