grafana/pkg/tsdb/cloudwatch/routes/external_id_test.go
Sarah Zinger 09d5483c6c
Cloudwatch: Upgrade aws-sdk and display external ids for temporary credentials (#72821)
(under a feature toggle, not yet ready for public testing)
2023-08-14 13:42:30 -04:00

37 lines
1017 B
Go

package routes
import (
"net/http"
"net/http/httptest"
"testing"
"github.com/stretchr/testify/assert"
)
func Test_external_id_route(t *testing.T) {
t.Run("successfully returns an external id from the env", func(t *testing.T) {
t.Setenv("AWS_AUTH_EXTERNAL_ID", "mock-external-id")
rr := httptest.NewRecorder()
handler := http.HandlerFunc(ResourceRequestMiddleware(ExternalIdHandler, logger, nil))
req := httptest.NewRequest("GET", "/external-id", nil)
handler.ServeHTTP(rr, req)
assert.Equal(t, http.StatusOK, rr.Code)
assert.JSONEq(t, `{"externalId":"mock-external-id"}`, rr.Body.String())
})
t.Run("returns an empty string if there is no external id", func(t *testing.T) {
rr := httptest.NewRecorder()
handler := http.HandlerFunc(ResourceRequestMiddleware(ExternalIdHandler, logger, nil))
req := httptest.NewRequest("GET", "/external-id", nil)
handler.ServeHTTP(rr, req)
assert.Equal(t, http.StatusOK, rr.Code)
assert.JSONEq(t, `{"externalId":""}`, rr.Body.String())
})
}