Thomas Pelletier
145 lines
2.5 KiB
Go
145 lines
2.5 KiB
Go
package toml
|
|
|
|
import (
|
|
"bytes"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestDecodeError(t *testing.T) {
|
|
examples := []struct {
|
|
desc string
|
|
doc [3]string
|
|
msg string
|
|
expected string
|
|
}{
|
|
{
|
|
desc: "no context",
|
|
doc: [3]string{"", "morning", ""},
|
|
msg: "this is wrong",
|
|
expected: `
|
|
1| morning
|
|
| ~~~~~~~ this is wrong`,
|
|
},
|
|
{
|
|
desc: "one line",
|
|
doc: [3]string{"good ", "morning", " everyone"},
|
|
msg: "this is wrong",
|
|
expected: `
|
|
1| good morning everyone
|
|
| ~~~~~~~ this is wrong`,
|
|
},
|
|
{
|
|
desc: "exactly 3 lines",
|
|
doc: [3]string{`line1
|
|
line2
|
|
line3
|
|
before `, "highlighted", ` after
|
|
post line 1
|
|
post line 2
|
|
post line 3`},
|
|
msg: "this is wrong",
|
|
expected: `
|
|
1| line1
|
|
2| line2
|
|
3| line3
|
|
4| before highlighted after
|
|
| ~~~~~~~~~~~ this is wrong
|
|
5| post line 1
|
|
6| post line 2
|
|
7| post line 3`,
|
|
},
|
|
{
|
|
desc: "more than 3 lines",
|
|
doc: [3]string{`should not be seen1
|
|
should not be seen2
|
|
line1
|
|
line2
|
|
line3
|
|
before `, "highlighted", ` after
|
|
post line 1
|
|
post line 2
|
|
post line 3
|
|
should not be seen3
|
|
should not be seen4`},
|
|
msg: "this is wrong",
|
|
expected: `
|
|
3| line1
|
|
4| line2
|
|
5| line3
|
|
6| before highlighted after
|
|
| ~~~~~~~~~~~ this is wrong
|
|
7| post line 1
|
|
8| post line 2
|
|
9| post line 3`,
|
|
},
|
|
{
|
|
desc: "more than 10 total lines",
|
|
doc: [3]string{`should not be seen 0
|
|
should not be seen1
|
|
should not be seen2
|
|
should not be seen3
|
|
line1
|
|
line2
|
|
line3
|
|
before `, "highlighted", ` after
|
|
post line 1
|
|
post line 2
|
|
post line 3
|
|
should not be seen3
|
|
should not be seen4`},
|
|
msg: "this is wrong",
|
|
expected: `
|
|
5| line1
|
|
6| line2
|
|
7| line3
|
|
8| before highlighted after
|
|
| ~~~~~~~~~~~ this is wrong
|
|
9| post line 1
|
|
10| post line 2
|
|
11| post line 3`,
|
|
},
|
|
{
|
|
desc: "last line of more than 10",
|
|
doc: [3]string{`should not be seen
|
|
should not be seen
|
|
should not be seen
|
|
should not be seen
|
|
should not be seen
|
|
should not be seen
|
|
should not be seen
|
|
line1
|
|
line2
|
|
line3
|
|
before `, "highlighted", ``},
|
|
msg: "this is wrong",
|
|
expected: `
|
|
8| line1
|
|
9| line2
|
|
10| line3
|
|
11| before highlighted
|
|
| ~~~~~~~~~~~ this is wrong
|
|
`,
|
|
},
|
|
}
|
|
|
|
for _, e := range examples {
|
|
t.Run(e.desc, func(t *testing.T) {
|
|
b := bytes.Buffer{}
|
|
b.Write([]byte(e.doc[0]))
|
|
start := b.Len()
|
|
b.Write([]byte(e.doc[1]))
|
|
end := b.Len()
|
|
b.Write([]byte(e.doc[2]))
|
|
doc := b.Bytes()
|
|
hl := doc[start:end]
|
|
|
|
err := decodeErrorFromHighlight(doc, hl, e.msg)
|
|
derr := err.(*DecodeError)
|
|
assert.Equal(t, strings.Trim(e.expected, "\n"), derr.String())
|
|
})
|
|
}
|
|
}
|