aws lambda 测试
当我开始在Go和AWS Lambda中工作时&＃xff0c;我面临的困难之一是单元测试。 我对什么是单元测试有一个不错的想法&＃xff0c;并且知道如何在Ruby中进行测试&＃xff0c;但是在Go中&＃xff0c;我不知道是什么原因&＃xff0c;因为我是一个初学者。
学习围棋本身就是一个挑战。 主要是因为Go不是一种OOP语言。 我开始在Go上阅读文章&＃xff0c;并在YouTube上观看了许多视频系列。 几天后&＃xff0c;我逐渐好起来&＃xff0c;并且能够理解事物。 但是我想学习如何进行单元测试&＃xff0c;不幸的是&＃xff0c;没有很多好的博客来说明如何使用Go进行AWS单元测试。 因此&＃xff0c;此博客旨在解释如何使用Go正确地对AWS服务进行单元测试。
在博客中&＃xff0c;我将演示如何对在Go中使用EMR服务的Lambda进行单元测试。 代码很简单&＃xff0c;给出了集群ID&＃xff0c;我必须获取集群状态。
永远记住&＃xff0c;如果要在Go中进行单元测试&＃xff0c;则必须使用接口&＃xff0c;并尽可能避免使用具体的API或函数。 所以对于aws-sdk-go &＃xff0c;我们有一些接口&＃xff0c;例如dynamodb
dynamodbiface您可以看到aws-sdk-go来查看服务的iface名称是什么。 通常&＃xff0c;其service-nameiface. 。
现在开始编码
首先&＃xff0c;我将创建将集群ID作为输入并将emr接口作为API的结构
// 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
}
接下来&＃xff0c;我将创建一个函数&＃xff0c;其功能是创建一个新的AWS会话并创建一个新的emr服务
// 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),}
}
现在&＃xff0c;来了肉的部分。 我将进行输入验证&＃xff0c;并准备输入DescribeCluster emr API方法。 休息就很简单。
// 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
}
要了解的重点是我如何在DescribeClusterInput上使用&emr 。 如果您想使用任何其他AWS服务&＃xff0c;那么您应该做类似的事情。
现在开始测试
对于测试&＃xff0c;我将使用Stretcher / Testify&＃xff0c;因为它提供了模拟和断言功能。 尤其是模拟非常重要。 当您编写单元测试时&＃xff0c;它不应该调用真实服务是至关重要的。 它应该始终调用模拟服务。
首先&＃xff0c;我将创建模拟emr并创建DescribeCluster方法的模拟实现。 之后&＃xff0c;我将创建setup方法
// 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
}
现在&＃xff0c;该编写表驱动测试并调用原始函数了。 一旦调用了原始函数&＃xff0c;我就可以断言预期结果是否与实际结果匹配。
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)
而已&＃xff01; 我希望阅读此博客后&＃xff0c;您可以了解Go中的一点单元测试AWS Lambda。
检出aws-unit-test-golang以获取完整代码
翻译自: https://hackernoon.com/unit-test-aws-lambda-in-go-h85l3ymz
aws lambda 测试
京公网安备 11010802041100号