forked from cloudfoundry-attic/bosh-init
-
Notifications
You must be signed in to change notification settings - Fork 0
/
fake_disk_deployer.go
55 lines (47 loc) · 1.25 KB
/
fake_disk_deployer.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
package fakes
import (
bicloud "github.com/cloudfoundry/bosh-init/cloud"
bidisk "github.com/cloudfoundry/bosh-init/deployment/disk"
bideplmanifest "github.com/cloudfoundry/bosh-init/deployment/manifest"
bivm "github.com/cloudfoundry/bosh-init/deployment/vm"
biui "github.com/cloudfoundry/bosh-init/ui"
)
type FakeDiskDeployer struct {
DeployInputs []DeployInput
deployOutputs deployOutput
}
type DeployInput struct {
DiskPool bideplmanifest.DiskPool
Cloud bicloud.Cloud
VM bivm.VM
EventLoggerStage biui.Stage
}
type deployOutput struct {
disks []bidisk.Disk
err error
}
func NewFakeDiskDeployer() *FakeDiskDeployer {
return &FakeDiskDeployer{
DeployInputs: []DeployInput{},
}
}
func (d *FakeDiskDeployer) Deploy(
diskPool bideplmanifest.DiskPool,
cloud bicloud.Cloud,
vm bivm.VM,
eventLoggerStage biui.Stage,
) ([]bidisk.Disk, error) {
d.DeployInputs = append(d.DeployInputs, DeployInput{
DiskPool: diskPool,
Cloud: cloud,
VM: vm,
EventLoggerStage: eventLoggerStage,
})
return d.deployOutputs.disks, d.deployOutputs.err
}
func (d *FakeDiskDeployer) SetDeployBehavior(disks []bidisk.Disk, err error) {
d.deployOutputs = deployOutput{
disks: disks,
err: err,
}
}