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>
59 lines
1.7 KiB
Go
59 lines
1.7 KiB
Go
package services
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/service/ec2"
|
|
ec2types "github.com/aws/aws-sdk-go-v2/service/ec2/types"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend/log"
|
|
"github.com/grafana/grafana/pkg/tsdb/cloudwatch/mocks"
|
|
"github.com/grafana/grafana/pkg/tsdb/cloudwatch/models/resources"
|
|
"github.com/grafana/grafana/pkg/tsdb/cloudwatch/utils"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
var testLogger = log.New().With("logger", "test.logger")
|
|
|
|
func TestRegions(t *testing.T) {
|
|
t.Run("returns regions from the api and merges them with default regions", func(t *testing.T) {
|
|
mockRegions := &ec2.DescribeRegionsOutput{
|
|
Regions: []ec2types.Region{
|
|
{
|
|
RegionName: utils.Pointer("earth-1"),
|
|
},
|
|
},
|
|
}
|
|
ec2Mock := &mocks.EC2Mock{}
|
|
ec2Mock.On("DescribeRegions").Return(mockRegions, nil)
|
|
regions, err := NewRegionsService(ec2Mock, testLogger).GetRegions(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Contains(t, regions, resources.ResourceResponse[resources.Region]{
|
|
Value: resources.Region{
|
|
Name: "us-east-2",
|
|
},
|
|
})
|
|
assert.Contains(t, regions, resources.ResourceResponse[resources.Region]{
|
|
Value: resources.Region{
|
|
Name: "earth-1",
|
|
},
|
|
})
|
|
})
|
|
|
|
t.Run("always returns default regions, even if fetch fails", func(t *testing.T) {
|
|
ec2Mock := &mocks.EC2Mock{}
|
|
mockRegions := &ec2.DescribeRegionsOutput{
|
|
Regions: []ec2types.Region{},
|
|
}
|
|
ec2Mock.On("DescribeRegions").Return(mockRegions, assert.AnError)
|
|
regions, err := NewRegionsService(ec2Mock, testLogger).GetRegions(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Contains(t, regions, resources.ResourceResponse[resources.Region]{
|
|
Value: resources.Region{
|
|
Name: "us-east-2",
|
|
},
|
|
})
|
|
})
|
|
}
|