-
Notifications
You must be signed in to change notification settings - Fork 0
/
token_writer_test.go
90 lines (76 loc) · 1.66 KB
/
token_writer_test.go
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
package main
import (
"bytes"
"encoding/json"
"fmt"
"io"
"math/rand"
"reflect"
"testing"
"testing/quick"
"github.com/stretchr/testify/require"
)
type testStruct struct {
Nested struct {
A safeString
B float64
C bool
D *bool
DoubleNested struct {
Stuff safeString
// Recursive *testStruct
}
}
Array []safeString
StructArray []struct {
X safeString
Y float64
Z float64
Contents struct {
Name safeString
}
}
}
type safeString string
func (s safeString) Generate(rand *rand.Rand, size int) reflect.Value {
buf := make([]byte, size)
_, err := rand.Read(buf)
if err != nil {
panic(err)
}
return reflect.ValueOf(safeString(fmt.Sprintf("%x", buf)))
}
func TestTokenWriter(t *testing.T) {
plain := func(ts testStruct) testStruct {
return ts
}
writeAndParse := func(ts testStruct) testStruct {
data, err := json.Marshal(ts)
require.NoError(t, err, "marshal")
buf := new(bytes.Buffer)
dec := json.NewDecoder(bytes.NewBuffer(data))
tw := NewTokenWriter(buf, dec)
token, err := dec.Token()
for err == nil {
err = tw.Write(token)
if err != nil {
err = fmt.Errorf("token write: %w", err)
continue
}
token, err = dec.Token()
}
if err != nil && err != io.EOF {
require.NoError(t, err, "token write")
}
var res testStruct
err = json.Unmarshal(buf.Bytes(), &res)
if synErr, ok := err.(*json.SyntaxError); err != nil && ok {
err = fmt.Errorf("syntax error: %w << %s >>", err, buf.Bytes()[synErr.Offset-10:synErr.Offset+10])
}
require.NoError(t, err, "unmarshal")
require.Equal(t, ts, res)
return res
}
err := quick.CheckEqual(plain, writeAndParse, nil)
require.NoError(t, err)
}