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
122
123
124
125
126
127
128
129
130
131
132
|
package binproto
import (
"bytes"
"io"
"testing"
)
func TestRegularIdKVMap(t *testing.T) {
// We assume that we are already in the map
testdata := []byte{
0x09, 0x01, // UKey(1)
0x05, 0x2a, 0, 0, 0, 0, 0, 0, 0, // Number(42)
0x09, 0x02, // UKey(2)
0x0c, 0x00, // Bool(false)
0x09, 0x03, // UKey(3)
0x06, // List, for testing skipping unknown fields
0x05, 0x01, 0, 0, 0, 0, 0, 0, 0, // Number(1)
0x05, 0x02, 0, 0, 0, 0, 0, 0, 0, // Number(2)
0x05, 0x03, 0, 0, 0, 0, 0, 0, 0, // Number(3)
0x0b, // Term
0x09, 0x04, // UKey(4)
0x04, 0x02, 0, 0, 0, 'h', 'i', // Bin("hi")
0x0b} // Term
r := bytes.NewReader(testdata)
ur := NewSimpleUnitReader(r)
var n1 int64
var b2 bool
var bs4 []byte
var seen1, seen2, seen4, seen5 bool
err := ScanIdKVMap(ur, map[byte]UKeyGetter{
1: {UTNumber, false, ActionStoreNumber(&n1), &seen1},
2: {UTBool, false, ActionStoreBool(&b2), &seen2},
4: {UTBin, false, ActionStoreBin(&bs4), &seen4},
5: {UTNil, true, ActionSkip(UTNil), &seen5}}, false)
if err != nil {
t.Errorf("Did not expect error, got: %s", err)
}
if n1 != 42 {
t.Errorf("n1 wrong, got %d, want 42", n1)
}
if b2 != false {
t.Error("b2 wrong, want false")
}
if !bytes.Equal(bs4, []byte("hi")) {
t.Errorf("bs4 wrong: %v", bs4)
}
if !(seen1 && seen2 && seen4 && (!seen5)) {
t.Errorf("Unexpected values for seen* vars: %v %v %v %v", seen1, seen2, seen4, seen5)
}
if b, err := r.ReadByte(); err != io.EOF {
t.Errorf("Expected EOF for reader r, got %x, %v", b, err)
}
}
func TestFailOnUnknown(t *testing.T) {
// We assume that we are already in the map
testdata := []byte{
0x09, 0x01, // UKey(1)
0x05, 0x2a, 0, 0, 0, 0, 0, 0, 0, // Number(42)
0x09, 0x02, // UKey(2)
0x0c, 0x00, // Bool(false)
0x09, 0x03, // UKey(3)
0x06, // List, for testing skipping unknown fields
0x05, 0x01, 0, 0, 0, 0, 0, 0, 0, // Number(1)
0x05, 0x02, 0, 0, 0, 0, 0, 0, 0, // Number(2)
0x05, 0x03, 0, 0, 0, 0, 0, 0, 0, // Number(3)
0x0b, // Term
0x09, 0x04, // UKey(4)
0x04, 0x02, 0, 0, 0, 'h', 'i', // Bin("hi")
0x0b} // Term
r := bytes.NewReader(testdata)
ur := NewSimpleUnitReader(r)
err := ScanIdKVMap(ur, map[byte]UKeyGetter{
1: {UTNumber, false, ActionSkip(UTNumber), nil},
2: {UTBool, false, ActionSkip(UTBool), nil},
4: {UTBin, false, ActionSkip(UTBin), nil},
5: {UTNil, true, ActionSkip(UTNil), nil}}, true)
if err != UnknownKey {
t.Errorf("Got wrong error: %s", err)
}
if b, err := r.ReadByte(); err != io.EOF {
t.Errorf("Expected EOF for reader r, got %x, %v", b, err)
}
}
func TestMissingMandatory(t *testing.T) {
// We assume that we are already in the map
testdata := []byte{
0x09, 0x01, // UKey(1)
0x05, 0x2a, 0, 0, 0, 0, 0, 0, 0, // Number(42)
0x09, 0x03, // UKey(3)
0x06, // List, for testing skipping unknown fields
0x05, 0x01, 0, 0, 0, 0, 0, 0, 0, // Number(1)
0x05, 0x02, 0, 0, 0, 0, 0, 0, 0, // Number(2)
0x05, 0x03, 0, 0, 0, 0, 0, 0, 0, // Number(3)
0x0b, // Term
0x09, 0x04, // UKey(4)
0x04, 0x02, 0, 0, 0, 'h', 'i', // Bin("hi")
0x0b} // Term
r := bytes.NewReader(testdata)
ur := NewSimpleUnitReader(r)
err := ScanIdKVMap(ur, map[byte]UKeyGetter{
1: {UTNumber, false, ActionSkip(UTNumber), nil},
2: {UTBool, false, ActionSkip(UTBool), nil},
4: {UTBin, false, ActionSkip(UTBin), nil},
5: {UTNil, true, ActionSkip(UTNil), nil}}, false)
if err != KeyMissing {
t.Errorf("Got wrong error: %s", err)
}
if b, err := r.ReadByte(); err != io.EOF {
t.Errorf("Expected EOF for reader r, got %x, %v", b, err)
}
}
|
