aboutsummaryrefslogtreecommitdiff
path: root/internal/lsp/snippet/snippet_builder_test.go
blob: bc814b16d3ffebc5b06975641602037c77b5f504 (plain)
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
// Copyright 2019 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 snippet

import (
	"testing"
)

func TestSnippetBuilder(t *testing.T) {
	expect := func(expected string, fn func(*Builder)) {
		t.Helper()

		var b Builder
		fn(&b)
		if got := b.String(); got != expected {
			t.Errorf("got %q, expected %q", got, expected)
		}
	}

	expect("", func(b *Builder) {})

	expect(`hi { \} \$ | " , / \\`, func(b *Builder) {
		b.WriteText(`hi { } $ | " , / \`)
	})

	expect("${1:}", func(b *Builder) {
		b.WritePlaceholder(nil)
	})

	expect("hi ${1:there}", func(b *Builder) {
		b.WriteText("hi ")
		b.WritePlaceholder(func(b *Builder) {
			b.WriteText("there")
		})
	})

	expect(`${1:id=${2:{your id\}}}`, func(b *Builder) {
		b.WritePlaceholder(func(b *Builder) {
			b.WriteText("id=")
			b.WritePlaceholder(func(b *Builder) {
				b.WriteText("{your id}")
			})
		})
	})

	expect(`${1|one,{ \} \$ \| " \, / \\,three|}`, func(b *Builder) {
		b.WriteChoice([]string{"one", `{ } $ | " , / \`, "three"})
	})

	expect("$0 hello", func(b *Builder) {
		b.WriteFinalTabstop()
		b.WriteText(" hello")
	})

	expect(`prepended \$5 ${1:} hello`, func(b *Builder) {
		b.WritePlaceholder(nil)
		b.WriteText(" hello")
		b.PrependText("prepended $5 ")
	})
}