Fix amazonec2 test nitpicks

Signed-off-by: Nathan LeClaire <nathan.leclaire@gmail.com>
This commit is contained in:
Nathan LeClaire 2016-03-04 12:20:27 -08:00
parent 1a0a740e27
commit 7452cb2c4f
1 changed files with 21 additions and 15 deletions

View File

@ -96,28 +96,34 @@ type fakeEC2SecurityGroupTestRecorder struct {
mock.Mock
}
func (f *fakeEC2SecurityGroupTestRecorder) DescribeSecurityGroups(
input *ec2.DescribeSecurityGroupsInput) (*ec2.DescribeSecurityGroupsOutput, error) {
func (f *fakeEC2SecurityGroupTestRecorder) DescribeSecurityGroups(input *ec2.DescribeSecurityGroupsInput) (*ec2.DescribeSecurityGroupsOutput, error) {
result := f.Called(input)
value, _ := result.Get(0).(*ec2.DescribeSecurityGroupsOutput)
return value, result.Error(1)
err := result.Error(1)
value, ok := result.Get(0).(*ec2.DescribeSecurityGroupsOutput)
if !ok && err == nil {
return nil, errors.New("Type assertion to DescribeSecurityGroupsOutput failed")
}
return value, err
}
func (f *fakeEC2SecurityGroupTestRecorder) CreateSecurityGroup(
input *ec2.CreateSecurityGroupInput) (*ec2.CreateSecurityGroupOutput, error) {
func (f *fakeEC2SecurityGroupTestRecorder) CreateSecurityGroup(input *ec2.CreateSecurityGroupInput) (*ec2.CreateSecurityGroupOutput, error) {
result := f.Called(input)
value, _ := result.Get(0).(*ec2.CreateSecurityGroupOutput)
return value, result.Error(1)
err := result.Error(1)
value, ok := result.Get(0).(*ec2.CreateSecurityGroupOutput)
if !ok && err == nil {
return nil, errors.New("Type assertion to CreateSecurityGroupOutput failed")
}
return value, err
}
func (f *fakeEC2SecurityGroupTestRecorder) AuthorizeSecurityGroupIngress(
input *ec2.AuthorizeSecurityGroupIngressInput) (*ec2.AuthorizeSecurityGroupIngressOutput, error) {
func (f *fakeEC2SecurityGroupTestRecorder) AuthorizeSecurityGroupIngress(input *ec2.AuthorizeSecurityGroupIngressInput) (*ec2.AuthorizeSecurityGroupIngressOutput, error) {
result := f.Called(input)
value, _ := result.Get(0).(*ec2.AuthorizeSecurityGroupIngressOutput)
return value, result.Error(1)
err := result.Error(1)
value, ok := result.Get(0).(*ec2.AuthorizeSecurityGroupIngressOutput)
if !ok && err == nil {
return nil, errors.New("Type assertion to AuthorizeSecurityGroupIngressInput failed")
}
return value, err
}
func NewTestDriver() *Driver {