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
|
// 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 (
"fmt"
"testing"
"time"
"github.com/google/syzkaller/pkg/vcs"
"github.com/google/syzkaller/syz-cluster/pkg/api"
"github.com/stretchr/testify/assert"
)
func TestCommitSelector(t *testing.T) {
allApply := map[string]bool{"head": true, "build": true}
tests := []struct {
name string
ops TreeOps
series *api.Series
last *api.Build
commits []string
}{
{
name: "fresh series, no last build",
series: &api.Series{PublishedAt: date("2020-Jan-15")},
ops: newTestGitOps(&vcs.Commit{Hash: "head", CommitDate: date("2020-Jan-10")}, allApply),
commits: []string{"head"},
},
{
name: "fresh series with a fresh last build",
series: &api.Series{PublishedAt: date("2020-Jan-15")},
ops: newTestGitOps(&vcs.Commit{Hash: "head", CommitDate: date("2020-Jan-10")}, allApply),
last: &api.Build{CommitHash: "build", CommitDate: date("2020-Jan-06")},
commits: []string{"head", "build"},
},
{
name: "fresh series with a too old last build",
series: &api.Series{PublishedAt: date("2020-Jan-15")},
ops: newTestGitOps(&vcs.Commit{Hash: "head", CommitDate: date("2020-Jan-10")}, allApply),
last: &api.Build{CommitHash: "build", CommitDate: date("2019-Dec-20")},
commits: []string{"head"},
},
{
name: "slightly old series",
series: &api.Series{PublishedAt: date("2020-Jan-15")},
ops: newTestGitOps(&vcs.Commit{Hash: "head", CommitDate: date("2020-Jan-20")}, allApply),
commits: []string{"head"},
},
{
name: "a too old series",
series: &api.Series{PublishedAt: date("2020-Jan-15")},
ops: newTestGitOps(&vcs.Commit{Hash: "head", CommitDate: date("2020-Feb-15")}, allApply),
commits: nil,
},
{
name: "doesn't apply to the older build",
series: &api.Series{PublishedAt: date("2020-Jan-15")},
ops: newTestGitOps(
&vcs.Commit{Hash: "head", CommitDate: date("2020-Jan-13")},
map[string]bool{"head": true, "build": false},
),
last: &api.Build{CommitHash: "build", CommitDate: date("2020-Jan-10")},
commits: []string{"head"},
},
{
name: "doesn't apply anywhere",
series: &api.Series{PublishedAt: date("2020-Jan-15")},
ops: newTestGitOps(
&vcs.Commit{Hash: "head", CommitDate: date("2020-Jan-13")},
nil,
),
last: &api.Build{CommitHash: "build", CommitDate: date("2020-Jan-10")},
commits: nil,
},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
selector := NewCommitSelector(test.ops)
commits, err := selector.Select(test.series, testTree, test.last)
assert.NoError(t, err)
assert.Equal(t, test.commits, commits)
})
}
}
func date(date string) time.Time {
t, err := time.Parse("2006-Jan-02", date)
if err != nil {
panic(err)
}
return t
}
var testTree = &api.Tree{} // all tests will use the same tree
type testGitOps struct {
applies map[string]bool
head map[*api.Tree]*vcs.Commit
}
func newTestGitOps(head *vcs.Commit, applies map[string]bool) *testGitOps {
return &testGitOps{
applies: applies,
head: map[*api.Tree]*vcs.Commit{
testTree: head,
},
}
}
func (ops *testGitOps) HeadCommit(tree *api.Tree) (*vcs.Commit, error) {
return ops.head[tree], nil
}
func (ops *testGitOps) ApplySeries(commit string, _ [][]byte) error {
if ops.applies[commit] {
return nil
}
return fmt.Errorf("didn't apply")
}
|
