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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
package syntaxhighlight
import (
"bytes"
"flag"
"io/ioutil"
"path/filepath"
"reflect"
"strings"
"testing"
"github.com/kr/pretty"
"github.com/sourcegraph/annotate"
)
var saveExp = flag.Bool("exp", false, "overwrite all expected output files with actual output (returning a failure)")
var match = flag.String("m", "", "only run tests whose name contains this string")
func testExpected(t *testing.T, fn, path, name, ext string, got []byte, err error) {
if err != nil {
t.Errorf("%s: %s: %s", fn, name, err)
return
}
expPath := path + ext
if *saveExp {
err = ioutil.WriteFile(expPath, got, 0700)
if err != nil {
t.Fatal(err)
}
return
}
want, err := ioutil.ReadFile(expPath)
if err != nil {
t.Fatal(err)
}
want = bytes.TrimSpace(want)
got = bytes.TrimSpace(got)
if !bytes.Equal(want, got) {
t.Errorf("%s: %s:\nwant ==========\n%q\ngot ===========\n%q", fn, name, want, got)
return
}
return
}
func TestAsHTML(t *testing.T) {
dir := "testdata"
tests, err := ioutil.ReadDir(dir)
if err != nil {
t.Fatal(err)
}
for _, test := range tests {
name := test.Name()
if !strings.Contains(name, *match) {
continue
}
if strings.HasSuffix(name, ".html") {
continue
}
if name == "net_http_client.go" {
// only use this file for benchmarking
continue
}
path := filepath.Join(dir, name)
input, err := ioutil.ReadFile(path)
if err != nil {
t.Fatal(err)
continue
}
got, err := AsHTML(input)
testExpected(t, "AsHTML", path, name, ".html", got, err)
got, err = AsHTML(input, OrderedList())
testExpected(t, "AsOrderedListHTML", path, name, ".ol.html", got, err)
}
if *saveExp {
t.Fatal("overwrote all expected output files with actual output (run tests again without -exp)")
}
}
func TestAnnotate(t *testing.T) {
src := []byte(`a:=2`)
want := annotate.Annotations{
{Start: 0, End: 1, Left: []byte(`<span class="pln">`), Right: []byte("</span>")},
{Start: 1, End: 2, Left: []byte(`<span class="pun">`), Right: []byte("</span>")},
{Start: 2, End: 3, Left: []byte(`<span class="pun">`), Right: []byte("</span>")},
{Start: 3, End: 4, Left: []byte(`<span class="dec">`), Right: []byte("</span>")},
}
got, err := Annotate(src, HTMLAnnotator(DefaultHTMLConfig))
if err != nil {
t.Fatal(err)
}
if !reflect.DeepEqual(want, got) {
t.Errorf("want %# v, got %# v\n\ndiff:\n%v", pretty.Formatter(want), pretty.Formatter(got), strings.Join(pretty.Diff(got, want), "\n"))
for _, g := range got {
t.Logf("%+v %q LEFT=%q RIGHT=%q", g, src[g.Start:g.End], g.Left, g.Right)
}
}
}
func BenchmarkAnnotate(b *testing.B) {
input, err := ioutil.ReadFile("testdata/net_http_client.go")
if err != nil {
b.Fatal(err)
}
b.ResetTimer()
for i := 0; i < b.N; i++ {
_, err := Annotate(input[:2000], HTMLAnnotator(DefaultHTMLConfig))
if err != nil {
b.Fatal(err)
}
}
}
|
