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 mock.Mock
} }
func (f *fakeEC2SecurityGroupTestRecorder) DescribeSecurityGroups( func (f *fakeEC2SecurityGroupTestRecorder) DescribeSecurityGroups(input *ec2.DescribeSecurityGroupsInput) (*ec2.DescribeSecurityGroupsOutput, error) {
input *ec2.DescribeSecurityGroupsInput) (*ec2.DescribeSecurityGroupsOutput, error) {
result := f.Called(input) result := f.Called(input)
value, _ := result.Get(0).(*ec2.DescribeSecurityGroupsOutput) err := result.Error(1)
return value, 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( func (f *fakeEC2SecurityGroupTestRecorder) CreateSecurityGroup(input *ec2.CreateSecurityGroupInput) (*ec2.CreateSecurityGroupOutput, error) {
input *ec2.CreateSecurityGroupInput) (*ec2.CreateSecurityGroupOutput, error) {
result := f.Called(input) result := f.Called(input)
value, _ := result.Get(0).(*ec2.CreateSecurityGroupOutput) err := result.Error(1)
return value, 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( func (f *fakeEC2SecurityGroupTestRecorder) AuthorizeSecurityGroupIngress(input *ec2.AuthorizeSecurityGroupIngressInput) (*ec2.AuthorizeSecurityGroupIngressOutput, error) {
input *ec2.AuthorizeSecurityGroupIngressInput) (*ec2.AuthorizeSecurityGroupIngressOutput, error) {
result := f.Called(input) result := f.Called(input)
value, _ := result.Get(0).(*ec2.AuthorizeSecurityGroupIngressOutput) err := result.Error(1)
return value, 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 { func NewTestDriver() *Driver {