forked from evergreen-ci/evergreen
/
mock_poller.go
70 lines (58 loc) · 1.71 KB
/
mock_poller.go
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
package repotracker
import (
"context"
"github.com/evergreen-ci/evergreen/model"
)
// MockRepoPoller is a utility for testing the repotracker using a dummy
// project
type mockRepoPoller struct {
project *model.Project
revisions []model.Revision
ConfigGets uint
nextError error
badDistro string
}
// Creates a new MockRepo poller with the given project settings
func NewMockRepoPoller(mockProject *model.Project, mockRevisions []model.Revision) *mockRepoPoller {
return &mockRepoPoller{
project: mockProject,
revisions: mockRevisions,
}
}
func (d *mockRepoPoller) setNextError(err error) {
d.nextError = err
}
func (d *mockRepoPoller) addBadDistro(distro string) {
d.badDistro = distro
}
func (d *mockRepoPoller) clearError() (err error) {
err, d.nextError = d.nextError, nil
return err
}
func (d *mockRepoPoller) GetChangedFiles(_ context.Context, commitRevision string) ([]string, error) {
return nil, nil
}
func (d *mockRepoPoller) GetRemoteConfig(_ context.Context, revision string) (*model.Project, error) {
d.ConfigGets++
if d.nextError != nil {
return nil, d.clearError()
}
if d.badDistro != "" {
// change the target distros if we've called addBadDistro, creating a validation warning
d.project.BuildVariants[0].RunOn = append(d.project.BuildVariants[0].RunOn, d.badDistro)
d.badDistro = ""
}
return d.project, nil
}
func (d *mockRepoPoller) GetRevisionsSince(revision string, maxRevisionsToSearch int) ([]model.Revision, error) {
if d.nextError != nil {
return nil, d.clearError()
}
return d.revisions, nil
}
func (d *mockRepoPoller) GetRecentRevisions(maxRevisionsToSearch int) ([]model.Revision, error) {
if d.nextError != nil {
return nil, d.clearError()
}
return d.revisions, nil
}