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
|
package parser
import (
"encoding/json"
"fmt"
"strings"
"testing"
"github.com/stretchr/testify/require"
)
func TestHighlight(t *testing.T) {
tests := []struct {
name string
language string
value string
want [][]token
}{
{
name: "go function definition",
language: "go",
value: "func main()",
want: [][]token{{{Class: "kd", Value: "func"}, {Value: " main()"}}},
},
{
name: "go struct definition",
language: "go",
value: "type Command struct",
want: [][]token{{{Class: "kd", Value: "type"}, {Value: " Command "}, {Class: "kd", Value: "struct"}}},
},
{
name: "go struct multiline definition",
language: "go",
value: `struct {\nConfig *Config\nReadWriter *ReadWriter\nEOFSent bool\n}`,
want: [][]token{
{{Class: "kd", Value: "struct"}, {Value: " {\n"}},
{{Value: "Config *Config\n"}},
{{Value: "ReadWriter *ReadWriter\n"}},
{{Value: "EOFSent "}, {Class: "kt", Value: "bool"}, {Value: "\n"}},
{{Value: "}"}},
},
},
{
name: "ruby method definition",
language: "ruby",
value: "def read(line)",
want: [][]token{{{Class: "k", Value: "def"}, {Value: " read(line)"}}},
},
{
name: "ruby multiline method definition",
language: "ruby",
value: `def read(line)\nend`,
want: [][]token{
{{Class: "k", Value: "def"}, {Value: " read(line)\n"}},
{{Class: "k", Value: "end"}},
},
},
{
name: "unknown/malicious language is passed",
language: "<lang> alert(1); </lang>",
value: `def a;\nend`,
want: [][]token(nil),
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
raw := []byte(fmt.Sprintf(`[{"language":"%s","value":"%s"}]`, tt.language, tt.value))
c, err := newCodeHovers(json.RawMessage(raw))
require.NoError(t, err)
require.Len(t, c, 1)
require.Equal(t, tt.want, c[0].Tokens)
})
}
}
func TestMarkdown(t *testing.T) {
value := `["This method reverses a string \n\n"]`
c, err := newCodeHovers(json.RawMessage(value))
require.NoError(t, err)
require.Len(t, c, 1)
require.Equal(t, "This method reverses a string \n\n", c[0].TruncatedValue.Value)
}
func TestMarkdownContentsFormat(t *testing.T) {
value := `{"kind":"markdown","value":"some _markdown_ **text**"}`
c, err := newCodeHovers(json.RawMessage(value))
require.NoError(t, err)
require.Len(t, c, 1)
require.Equal(t, [][]token(nil), c[0].Tokens)
require.Equal(t, "some _markdown_ **text**", c[0].TruncatedValue.Value)
}
func TestTruncatedValue(t *testing.T) {
value := strings.Repeat("a", 500)
rawValue, err := json.Marshal(value)
require.NoError(t, err)
c, err := newCodeHover(rawValue)
require.NoError(t, err)
require.Equal(t, value[0:maxValueSize], c.TruncatedValue.Value)
require.True(t, c.TruncatedValue.Truncated)
}
func TestTruncatingMultiByteChars(t *testing.T) {
value := strings.Repeat("ಅ", 500)
rawValue, err := json.Marshal(value)
require.NoError(t, err)
c, err := newCodeHover(rawValue)
require.NoError(t, err)
symbolSize := 3
require.Equal(t, value[0:maxValueSize*symbolSize], c.TruncatedValue.Value)
}
|