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>
91 lines
2.6 KiB
Go
91 lines
2.6 KiB
Go
package cloudwatch
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/aws"
|
|
"github.com/aws/aws-sdk-go-v2/service/cloudwatch"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
|
"github.com/grafana/grafana/pkg/tsdb/cloudwatch/models"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestQuery_AnnotationQuery(t *testing.T) {
|
|
ds := newTestDatasource()
|
|
origNewCWClient := NewCWClient
|
|
t.Cleanup(func() {
|
|
NewCWClient = origNewCWClient
|
|
})
|
|
|
|
var client fakeCWAnnotationsClient
|
|
NewCWClient = func(aws.Config) models.CWClient {
|
|
return &client
|
|
}
|
|
|
|
t.Run("DescribeAlarmsForMetric is called with minimum parameters", func(t *testing.T) {
|
|
client = fakeCWAnnotationsClient{describeAlarmsForMetricOutput: &cloudwatch.DescribeAlarmsForMetricOutput{}}
|
|
|
|
_, err := ds.QueryData(context.Background(), &backend.QueryDataRequest{
|
|
PluginContext: backend.PluginContext{
|
|
DataSourceInstanceSettings: &backend.DataSourceInstanceSettings{},
|
|
},
|
|
Queries: []backend.DataQuery{
|
|
{
|
|
JSON: json.RawMessage(`{
|
|
"type": "annotationQuery",
|
|
"region": "us-east-1",
|
|
"namespace": "custom",
|
|
"metricName": "CPUUtilization",
|
|
"statistic": "Average"
|
|
}`),
|
|
},
|
|
},
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
require.Len(t, client.calls.describeAlarmsForMetric, 1)
|
|
assert.Equal(t, &cloudwatch.DescribeAlarmsForMetricInput{
|
|
Namespace: aws.String("custom"),
|
|
MetricName: aws.String("CPUUtilization"),
|
|
Statistic: "Average",
|
|
Period: aws.Int32(300),
|
|
}, client.calls.describeAlarmsForMetric[0])
|
|
})
|
|
|
|
t.Run("DescribeAlarms is called when prefixMatching is true", func(t *testing.T) {
|
|
client = fakeCWAnnotationsClient{describeAlarmsOutput: &cloudwatch.DescribeAlarmsOutput{}}
|
|
|
|
_, err := ds.QueryData(context.Background(), &backend.QueryDataRequest{
|
|
PluginContext: backend.PluginContext{
|
|
DataSourceInstanceSettings: &backend.DataSourceInstanceSettings{},
|
|
},
|
|
Queries: []backend.DataQuery{
|
|
{
|
|
JSON: json.RawMessage(`{
|
|
"type": "annotationQuery",
|
|
"region": "us-east-1",
|
|
"namespace": "custom",
|
|
"metricName": "CPUUtilization",
|
|
"statistic": "Average",
|
|
"prefixMatching": true,
|
|
"actionPrefix": "some_action_prefix",
|
|
"alarmNamePrefix": "some_alarm_name_prefix"
|
|
}`),
|
|
},
|
|
},
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
require.Len(t, client.calls.describeAlarms, 1)
|
|
assert.Equal(t, &cloudwatch.DescribeAlarmsInput{
|
|
MaxRecords: aws.Int32(100),
|
|
ActionPrefix: aws.String("some_action_prefix"),
|
|
AlarmNamePrefix: aws.String("some_alarm_name_prefix"),
|
|
}, client.calls.describeAlarms[0])
|
|
})
|
|
}
|