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
|
// Copyright 2024 syzkaller project authors. All rights reserved.
// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
package triage
import (
"testing"
"github.com/google/syzkaller/syz-cluster/pkg/api"
"github.com/stretchr/testify/assert"
)
func TestSelectTrees(t *testing.T) {
trees := []*api.Tree{
{
Name: "bpf",
EmailLists: []string{"bpf@list"},
},
{
Name: "wireless",
EmailLists: []string{"wireless@list"},
},
{
Name: "net",
EmailLists: []string{"net@list"},
},
{
Name: "test",
EmailLists: []string{"test@list"},
},
{
Name: "mainline",
EmailLists: nil,
},
}
tests := []struct {
testName string
result []string
series *api.Series
}{
{
testName: "only-net",
result: []string{"net", "mainline"},
series: &api.Series{Cc: []string{"net@list"}},
},
{
testName: "prefer-wireless",
result: []string{"wireless", "net", "mainline"},
series: &api.Series{Cc: []string{"net@list", "wireless@list"}},
},
{
testName: "fallback",
result: []string{"mainline"},
series: &api.Series{Cc: []string{"unknown@list"}},
},
{
testName: "prefer-direct-match",
result: []string{"test", "wireless", "net", "mainline"},
series: &api.Series{
Cc: []string{"net@list", "wireless@list"},
SubjectTags: []string{"test"},
},
},
}
for _, test := range tests {
t.Run(test.testName, func(t *testing.T) {
ret := SelectTrees(test.series, trees)
var retNames []string
for _, tree := range ret {
retNames = append(retNames, tree.Name)
}
assert.Equal(t, test.result, retNames)
})
}
}
func TestTreeFromBranch(t *testing.T) {
treeA, treeB := &api.Tree{Name: "a"}, &api.Tree{Name: "b"}
assert.Equal(t, treeA, TreeFromBranch([]*api.Tree{treeA, treeB}, "a/some_branch"))
assert.Equal(t, treeB, TreeFromBranch([]*api.Tree{treeA, treeB}, "b/some_branch"))
}
|
