Nathan Vērzemnieks
560be84dc8
* CloudWatch: Backport aws-sdk-go-v2 update from external plugin (#107136) (cherry picked froma18ea34688) * Datasources: Update grafana-aws-sdk for new sigv4 middleware and aws-sdk-go v1 removal (#107522) (cherry picked from commit66d9a33cc9) * CloudWatch: Fix proxy transport issue (#107807) (cherry picked fromc3eeb1fcd9) * CloudWatch: Fix http client handling + assume role bug (#107893) (cherry picked from commitf34a9fc0c2) * CloudWatch: Use default region when query region is unset (#109089) (cherry picked from commit5f4097a159) * CloudWatch: Fix handling region for legacy alerts (#109217) (cherry picked from commit2bf9aea8ef) * Update go.mod owners --------- Co-authored-by: Isabella Siu <Isabella.siu@grafana.com>
66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
package mocks
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/aws"
|
|
"github.com/aws/aws-sdk-go-v2/service/cloudwatch"
|
|
cloudwatchtypes "github.com/aws/aws-sdk-go-v2/service/cloudwatch/types"
|
|
|
|
"github.com/grafana/grafana/pkg/tsdb/cloudwatch/models"
|
|
|
|
"github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
type FakeMetricsAPI struct {
|
|
models.CWClient
|
|
|
|
Metrics []cloudwatchtypes.Metric
|
|
OwningAccounts []string
|
|
MetricsPerPage int
|
|
|
|
cursor int
|
|
}
|
|
|
|
func (c *FakeMetricsAPI) ListMetrics(_ context.Context, _ *cloudwatch.ListMetricsInput, _ ...func(*cloudwatch.Options)) (*cloudwatch.ListMetricsOutput, error) {
|
|
if c.MetricsPerPage == 0 {
|
|
c.MetricsPerPage = 1000
|
|
}
|
|
var metrics []cloudwatchtypes.Metric
|
|
nextToken := aws.String("yes")
|
|
if c.cursor < len(c.Metrics) {
|
|
end := c.cursor + c.MetricsPerPage
|
|
if end > len(c.Metrics) {
|
|
end = len(c.Metrics)
|
|
nextToken = nil
|
|
}
|
|
metrics = c.Metrics[c.cursor:end]
|
|
}
|
|
c.cursor += c.MetricsPerPage
|
|
|
|
return &cloudwatch.ListMetricsOutput{
|
|
Metrics: metrics,
|
|
OwningAccounts: c.OwningAccounts,
|
|
NextToken: nextToken,
|
|
}, nil
|
|
}
|
|
|
|
type MetricsAPI struct {
|
|
mock.Mock
|
|
models.CWClient
|
|
|
|
Metrics []cloudwatchtypes.Metric
|
|
}
|
|
|
|
func (m *MetricsAPI) GetMetricData(ctx context.Context, input *cloudwatch.GetMetricDataInput, optFns ...func(*cloudwatch.Options)) (*cloudwatch.GetMetricDataOutput, error) {
|
|
args := m.Called(ctx, input, optFns)
|
|
|
|
return args.Get(0).(*cloudwatch.GetMetricDataOutput), args.Error(1)
|
|
}
|
|
|
|
func (m *MetricsAPI) ListMetrics(_ context.Context, _ *cloudwatch.ListMetricsInput, _ ...func(*cloudwatch.Options)) (*cloudwatch.ListMetricsOutput, error) {
|
|
return &cloudwatch.ListMetricsOutput{
|
|
Metrics: m.Metrics,
|
|
}, m.Called().Error(0)
|
|
}
|