-
Notifications
You must be signed in to change notification settings - Fork 70
/
mock_discoverer.go
48 lines (37 loc) · 988 Bytes
/
mock_discoverer.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
package discovery
import (
"context"
"github.com/newrelic/newrelic-cli/internal/install/types"
)
type MockDiscoverer struct {
DiscoveryManifest *types.DiscoveryManifest
}
func NewMockDiscoverer() *MockDiscoverer {
m := &types.DiscoveryManifest{
OS: "linux",
Platform: "",
PlatformVersion: "",
KernelArch: "",
}
return &MockDiscoverer{
DiscoveryManifest: m,
}
}
func (d *MockDiscoverer) SetOs(os string) {
d.DiscoveryManifest.OS = os
}
func (d *MockDiscoverer) SetPlatform(p string) {
d.DiscoveryManifest.Platform = p
}
func (d *MockDiscoverer) SetPlatformVersion(pf string) {
d.DiscoveryManifest.PlatformVersion = pf
}
func (d *MockDiscoverer) SetKernelArch(arch string) {
d.DiscoveryManifest.KernelArch = arch
}
func (d *MockDiscoverer) GetManifest() *types.DiscoveryManifest {
return d.DiscoveryManifest
}
func (d *MockDiscoverer) Discover(context.Context) (*types.DiscoveryManifest, error) {
return d.DiscoveryManifest, nil
}