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
|
// Copyright ©2021 The Gonum 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 encoding
import (
"sort"
"testing"
)
var setAttributesTests = []struct {
attr *Attributes
opName string
op func(AttributeSetter) error
want *Attributes
}{
{
attr: &Attributes{},
opName: "noop",
op: func(a AttributeSetter) error {
return a.SetAttribute(Attribute{Key: "", Value: "bar"})
},
want: &Attributes{},
},
{
attr: &Attributes{},
opName: "add attr to empty",
op: func(a AttributeSetter) error {
return a.SetAttribute(Attribute{Key: "foo", Value: "bar"})
},
want: &Attributes{{Key: "foo", Value: "bar"}},
},
{
attr: &Attributes{},
opName: "remove attr from empty",
op: func(a AttributeSetter) error {
return a.SetAttribute(Attribute{Key: "foo", Value: ""})
},
want: &Attributes{},
},
{
attr: &Attributes{{Key: "foo", Value: "bar"}},
opName: "add attr to non-empty",
op: func(a AttributeSetter) error {
return a.SetAttribute(Attribute{Key: "bif", Value: "fud"})
},
want: &Attributes{{Key: "foo", Value: "bar"}, {Key: "bif", Value: "fud"}},
},
{
attr: &Attributes{{Key: "foo", Value: "bar"}},
opName: "remove attr from singleton",
op: func(a AttributeSetter) error {
return a.SetAttribute(Attribute{Key: "foo", Value: ""})
},
want: &Attributes{},
},
{
attr: &Attributes{{Key: "foo", Value: "bar"}, {Key: "bif", Value: "fud"}},
opName: "remove first attr from pair",
op: func(a AttributeSetter) error {
return a.SetAttribute(Attribute{Key: "foo", Value: ""})
},
want: &Attributes{{Key: "bif", Value: "fud"}},
},
{
attr: &Attributes{{Key: "foo", Value: "bar"}, {Key: "bif", Value: "fud"}},
opName: "remove second attr from pair",
op: func(a AttributeSetter) error {
return a.SetAttribute(Attribute{Key: "bif", Value: ""})
},
want: &Attributes{{Key: "foo", Value: "bar"}},
},
{
attr: &Attributes{{Key: "foo", Value: "bar"}, {Key: "bif", Value: "fud"}},
opName: "replace first attr in pair",
op: func(a AttributeSetter) error {
return a.SetAttribute(Attribute{Key: "foo", Value: "not bar"})
},
want: &Attributes{{Key: "foo", Value: "not bar"}, {Key: "bif", Value: "fud"}},
},
{
attr: &Attributes{{Key: "foo", Value: "bar"}, {Key: "bif", Value: "fud"}},
opName: "replace second attr in pair",
op: func(a AttributeSetter) error {
return a.SetAttribute(Attribute{Key: "bif", Value: "not fud"})
},
want: &Attributes{{Key: "foo", Value: "bar"}, {Key: "bif", Value: "not fud"}},
},
}
func TestSetAttributes(t *testing.T) {
for _, test := range setAttributesTests {
err := test.op(test.attr)
if err != nil {
t.Errorf("unexpected error for %q: %v", test.opName, err)
continue
}
if !sameAttributes(test.attr, test.want) {
t.Errorf("unexpected result from %q:\ngot: %+v\nwant:%+v", test.opName, test.attr, test.want)
}
}
}
func sameAttributes(a, b Attributer) bool {
aAttr := a.Attributes()
bAttr := b.Attributes()
if len(aAttr) != len(bAttr) {
return false
}
aAttr = append(aAttr[:0:0], aAttr...)
sort.Slice(aAttr, func(i, j int) bool { return aAttr[i].Key < aAttr[j].Key })
bAttr = append(bAttr[:0:0], bAttr...)
sort.Slice(bAttr, func(i, j int) bool { return bAttr[i].Key < bAttr[j].Key })
for i, a := range aAttr {
if bAttr[i] != a {
return false
}
}
return true
}
|
