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
|
// Copyright 2023 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package main
import (
"strings"
"testing"
)
func TestJSONFilterRewritePackage(t *testing.T) {
const in = `{"Package":"abc"}
{"Field1":"1","Package":"abc","Field3":"3"}
{"Package":123}
{}
{"Package":"abc","Unexpected":[null,true,false,99999999999999999999]}
`
want := strings.ReplaceAll(in, `"Package":"abc"`, `"Package":"abc:variant"`)
checkJSONFilter(t, in, want)
}
func TestJSONFilterMalformed(t *testing.T) {
const in = `unexpected text
{"Package":"abc"}
more text
{"Package":"abc"}trailing text
{not json}
no newline`
const want = `unexpected text
{"Package":"abc:variant"}
more text
{"Package":"abc:variant"}trailing text
{not json}
no newline`
checkJSONFilter(t, in, want)
}
func TestJSONFilterBoundaries(t *testing.T) {
const in = `{"Package":"abc"}
{"Package":"def"}
{"Package":"ghi"}
`
want := strings.ReplaceAll(in, `"}`, `:variant"}`)
// Write one bytes at a time.
t.Run("bytes", func(t *testing.T) {
checkJSONFilterWith(t, want, func(f *testJSONFilter) {
for i := 0; i < len(in); i++ {
f.Write([]byte{in[i]})
}
})
})
// Write a block containing a whole line bordered by two partial lines.
t.Run("bytes", func(t *testing.T) {
checkJSONFilterWith(t, want, func(f *testJSONFilter) {
const b1 = 5
const b2 = len(in) - 5
f.Write([]byte(in[:b1]))
f.Write([]byte(in[b1:b2]))
f.Write([]byte(in[b2:]))
})
})
}
func checkJSONFilter(t *testing.T, in, want string) {
t.Helper()
checkJSONFilterWith(t, want, func(f *testJSONFilter) {
f.Write([]byte(in))
})
}
func checkJSONFilterWith(t *testing.T, want string, write func(*testJSONFilter)) {
t.Helper()
out := new(strings.Builder)
f := &testJSONFilter{w: out, variant: "variant"}
write(f)
f.Flush()
got := out.String()
if want != got {
t.Errorf("want:\n%s\ngot:\n%s", want, got)
}
}
|
