-
Notifications
You must be signed in to change notification settings - Fork 124
/
mock_poller.go
80 lines (67 loc) · 2.04 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
71
72
73
74
75
76
77
78
79
80
package repotracker
import (
"context"
"github.com/evergreen-ci/evergreen/model"
"github.com/pkg/errors"
)
// MockRepoPoller is a utility for testing the repotracker using a dummy
// project
type mockRepoPoller struct {
parserProject *model.ParserProject
revisions []model.Revision
ConfigGets uint
nextError error
badDistro string
}
// Creates a new MockRepo poller with the given project settings
func NewMockRepoPoller(mockProject *model.ParserProject, mockRevisions []model.Revision) *mockRepoPoller {
return &mockRepoPoller{
parserProject: 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.ProjectInfo, error) {
d.ConfigGets++
if d.nextError != nil {
return model.ProjectInfo{}, d.clearError()
}
if d.badDistro != "" {
// change the target distros if we've called addBadDistro, creating a validation warning
d.parserProject.BuildVariants[0].RunOn = append(d.parserProject.BuildVariants[0].RunOn, d.badDistro)
d.badDistro = ""
}
p, err := model.TranslateProject(d.parserProject)
if err != nil {
return model.ProjectInfo{}, errors.Wrapf(err, model.TranslateProjectError)
}
return model.ProjectInfo{
Project: p,
IntermediateProject: d.parserProject,
Config: nil,
}, 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
}