aboutsummaryrefslogtreecommitdiff
path: root/formatter_bench_test.go
blob: 53f607b9e2f91c8555e2e50cec5264d7d74b1c90 (plain) (blame)
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
package main

import (
	"strings"
	"testing"
)

func BenchmarkFormat_Small(b *testing.B) {
	input := []byte(`package main
func main() {
	x := 1
	y := 2
	if x > 0 {
		z := 3
	}
	a := 4
}
`)
	f := &Formatter{CommentMode: CommentsFollow}

	for b.Loop() {
		_, _ = f.Format(input)
	}
}

func BenchmarkFormat_Large(b *testing.B) {
	var sb strings.Builder

	sb.WriteString("package main\n\n")

	for i := range 100 {
		sb.WriteString("func foo")
		sb.WriteString(string(rune('A' + i%26)))
		sb.WriteString("() {\n")
		sb.WriteString("\tx := 1\n")
		sb.WriteString("\tif x > 0 {\n")
		sb.WriteString("\t\ty := 2\n")
		sb.WriteString("\t}\n")
		sb.WriteString("\tz := 3\n")
		sb.WriteString("}\n\n")
	}

	input := []byte(sb.String())
	f := &Formatter{CommentMode: CommentsFollow}

	for b.Loop() {
		_, _ = f.Format(input)
	}
}