awslambda测试_在Go中对AWSLambda进行单元测试

// ClusterInput represent input which will be given to the lambda
type ClusterInput struct {ClusterID string &＃96;json:"clusterID"&＃96;
}// awsService represents emr interface
type awsService struct {emr emriface.EMRAPI
}

// newAWSService returns a new instance of emr
func newAWSService () * awsService {awsConfig :&＃61; &aws.Config{Region: aws.String( "us-west-2" )}sess, err :&＃61; session.NewSession(awsConfig)if err !&＃61; nil {log.Errorf( "error while creating AWS session - %s" , err.Error())}return &awsService{emr: emr.New(sess),}
}

// getClusterStatus returns current cluster status along with an error
func (svc *awsService) getClusterStatus (input ClusterInput) ( string , error) {clusterID :&＃61; input.ClusterIDif clusterID &＃61;&＃61; "" {return "" , errors.New( "clusterID is empty" )}describeClusterInput :&＃61; &emr.DescribeClusterInput{ClusterId: aws.String(clusterID),}clusterDetails, err :&＃61; svc.emr.DescribeCluster(describeClusterInput)if err !&＃61; nil {log.Errorf( "DescribeCluster error - %s" , err)return "" , err}if clusterDetails &＃61;&＃61; nil {log.Errorf( "clusterID does not exist" )return "" , errors.New( "clusterID does not exist" )}clusterStatus :&＃61; *clusterDetails.Cluster.Status.Statereturn string (clusterStatus), nil
}

// mockEMR represents mock implementation of AWS EMR service
type mockEMR struct {emriface.EMRAPImock.Mock
}// DescribeCluster is a mocked method which return the cluster status
func (m *mockEMR) DescribeCluster (input *emr.DescribeClusterInput) (*emr.DescribeClusterOutput, error) {args :&＃61; m.Called(input)return args.Get( 0 ).(*emr.DescribeClusterOutput), args.Error( 1 )
}func setup () (*mockEMR, *awsService) {mockEMRClient :&＃61; new (mockEMR)mockEMR :&＃61; &awsService{emr: mockEMRClient,}return mockEMRClient, mockEMR
}

mockEMRClient, mockEMR :&＃61; setup()mockDescribeClusterInput :&＃61; &emr.DescribeClusterInput{ClusterId: aws.String(testCase.clusterID),
}mockDescribeClusterOutput :&＃61; &emr.DescribeClusterOutput{Cluster: &emr.Cluster{Status: &emr.ClusterStatus{State: aws.String(testCase.expectedClusterStatus),},},
}mockEMRClient.On("DescribeCluster" , mockDescribeClusterInput).Return(mockDescribeClusterOutput, testCase.emrError)
res, err :&＃61; mockEMR.getClusterStatus(testCase.expectedInput)assert.Equal(t, testCase.expectedClusterStatus, res, testCase.message)
assert.IsType(t, testCase.expectedError, err, testCase.message)