forked from LeanerCloud/AutoSpotting
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmock_test.go
139 lines (110 loc) · 3.96 KB
/
mock_test.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
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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
package autospotting
import (
"reflect"
"testing"
"github.com/aws/aws-sdk-go/service/autoscaling"
"github.com/aws/aws-sdk-go/service/autoscaling/autoscalingiface"
"github.com/aws/aws-sdk-go/service/ec2"
"github.com/aws/aws-sdk-go/service/ec2/ec2iface"
)
func CheckErrors(t *testing.T, err error, expected error) {
if err != nil && !reflect.DeepEqual(err, expected) {
t.Errorf("Error received: '%v' expected '%v'",
err.Error(), expected.Error())
}
}
// All fields are composed of the abbreviation of their method
// This is useful when methods are doing multiple calls to AWS API
type mockEC2 struct {
ec2iface.EC2API
// Create tags
cto *ec2.CreateTagsOutput
cterr error
// Wait Until Spot Instance Request Fulfilled
wusirferr error
// Describe Instance Request
dsiro *ec2.DescribeSpotInstanceRequestsOutput
dsirerr error
// Describe Spot Price History
dspho *ec2.DescribeSpotPriceHistoryOutput
dspherr error
// Error in DescribeInstancesPages
diperr error
// Terminate Instance
tio *ec2.TerminateInstancesOutput
tierr error
// Request Spot Instance
rsio *ec2.RequestSpotInstancesOutput
rsierr error
// Describe Spot Instance Requests
dspiro *ec2.DescribeSpotInstanceRequestsOutput
dspirerr error
// Describe Regions
dro *ec2.DescribeRegionsOutput
drerr error
// Cancel Spot instance request
csiro *ec2.CancelSpotInstanceRequestsOutput
csirerr error
}
func (m mockEC2) CreateTags(in *ec2.CreateTagsInput) (*ec2.CreateTagsOutput, error) {
return m.cto, m.cterr
}
func (m mockEC2) WaitUntilSpotInstanceRequestFulfilled(in *ec2.DescribeSpotInstanceRequestsInput) error {
return m.wusirferr
}
func (m mockEC2) DescribeSpotInstanceRequests(in *ec2.DescribeSpotInstanceRequestsInput) (*ec2.DescribeSpotInstanceRequestsOutput, error) {
return m.dsiro, m.dsirerr
}
func (m mockEC2) DescribeSpotPriceHistory(in *ec2.DescribeSpotPriceHistoryInput) (*ec2.DescribeSpotPriceHistoryOutput, error) {
return m.dspho, m.dspherr
}
func (m mockEC2) DescribeInstancesPages(in *ec2.DescribeInstancesInput, fn func(*ec2.DescribeInstancesOutput, bool) bool) error {
return m.diperr
}
func (m mockEC2) TerminateInstances(*ec2.TerminateInstancesInput) (*ec2.TerminateInstancesOutput, error) {
return m.tio, m.tierr
}
func (m mockEC2) RequestSpotInstances(*ec2.RequestSpotInstancesInput) (*ec2.RequestSpotInstancesOutput, error) {
return m.rsio, m.rsierr
}
func (m mockEC2) CancelSpotInstanceRequests(*ec2.CancelSpotInstanceRequestsInput) (*ec2.CancelSpotInstanceRequestsOutput, error) {
return m.csiro, m.csirerr
}
func (m mockEC2) DescribeRegions(*ec2.DescribeRegionsInput) (*ec2.DescribeRegionsOutput, error) {
return m.dro, m.drerr
}
// All fields are composed of the abbreviation of their method
// This is useful when methods are doing multiple calls to AWS API
type mockASG struct {
autoscalingiface.AutoScalingAPI
// Detach Instances
dio *autoscaling.DetachInstancesOutput
dierr error
// Attach Instances
aio *autoscaling.AttachInstancesOutput
aierr error
// Describe Launch Config
dlco *autoscaling.DescribeLaunchConfigurationsOutput
dlcerr error
// Update AutoScaling Group
uasgo *autoscaling.UpdateAutoScalingGroupOutput
uasgerr error
// Describe Tags
dto *autoscaling.DescribeTagsOutput
dterr error
}
func (m mockASG) DetachInstances(*autoscaling.DetachInstancesInput) (*autoscaling.DetachInstancesOutput, error) {
return m.dio, m.dierr
}
func (m mockASG) AttachInstances(*autoscaling.AttachInstancesInput) (*autoscaling.AttachInstancesOutput, error) {
return m.aio, m.aierr
}
func (m mockASG) DescribeLaunchConfigurations(*autoscaling.DescribeLaunchConfigurationsInput) (*autoscaling.DescribeLaunchConfigurationsOutput, error) {
return m.dlco, m.dlcerr
}
func (m mockASG) UpdateAutoScalingGroup(*autoscaling.UpdateAutoScalingGroupInput) (*autoscaling.UpdateAutoScalingGroupOutput, error) {
return m.uasgo, m.uasgerr
}
func (m mockASG) DescribeTags(*autoscaling.DescribeTagsInput) (*autoscaling.DescribeTagsOutput, error) {
return m.dto, m.dterr
}