Shabeeb Khalid
9dc6cac1f2
* Use context in aws DescribeRegionsWithContext In the current way, DescribeRegions is used which doesn't allow cancelling the request if the context changes. Using DescribeRegionsWithContext is the preferred way. * Fix context variable * Revert GetRegionsWithContext to GetRegions GetRegions is not an AWS SDK method. Hence, GetRegions should be enough as the name change is not needed for context implementation.
35 lines
1.0 KiB
Go
35 lines
1.0 KiB
Go
package mocks
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/aws/request"
|
|
"github.com/aws/aws-sdk-go/service/ec2"
|
|
"github.com/grafana/grafana/pkg/tsdb/cloudwatch/models/resources"
|
|
"github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
type RegionsService struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (r *RegionsService) GetRegions(ctx context.Context) (in []resources.ResourceResponse[resources.Region], e error) {
|
|
args := r.Called()
|
|
return args.Get(0).(([]resources.ResourceResponse[resources.Region])), args.Error(1)
|
|
}
|
|
|
|
type EC2Mock struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (e *EC2Mock) DescribeRegionsWithContext(ctx aws.Context, in *ec2.DescribeRegionsInput, opts ...request.Option) (*ec2.DescribeRegionsOutput, error) {
|
|
args := e.Called()
|
|
return args.Get(0).(*ec2.DescribeRegionsOutput), args.Error(1)
|
|
}
|
|
|
|
func (e *EC2Mock) DescribeInstancesPagesWithContext(ctx aws.Context, in *ec2.DescribeInstancesInput, fn func(*ec2.DescribeInstancesOutput, bool) bool, opts ...request.Option) error {
|
|
args := e.Called(in, fn)
|
|
return args.Error(0)
|
|
}
|