Yuriy Tseretyan
74 lines
2.5 KiB
Go
74 lines
2.5 KiB
Go
package routes
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/grafana/grafana/pkg/tsdb/cloudwatch/models"
|
|
"github.com/grafana/grafana/pkg/tsdb/cloudwatch/services"
|
|
)
|
|
|
|
func Test_Namespaces_Route(t *testing.T) {
|
|
customNamespaces := ""
|
|
factoryFunc := func(pluginCtx backend.PluginContext, region string) (reqCtx models.RequestContext, err error) {
|
|
return models.RequestContext{
|
|
Settings: &models.CloudWatchSettings{
|
|
Namespace: customNamespaces,
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
t.Run("calls GetHardCodedNamespaces", func(t *testing.T) {
|
|
origGetHardCodedNamespaces := services.GetHardCodedNamespaces
|
|
t.Cleanup(func() {
|
|
services.GetHardCodedNamespaces = origGetHardCodedNamespaces
|
|
})
|
|
haveBeenCalled := false
|
|
services.GetHardCodedNamespaces = func() []string {
|
|
haveBeenCalled = true
|
|
return []string{}
|
|
}
|
|
rr := httptest.NewRecorder()
|
|
req := httptest.NewRequest("GET", "/namespaces", nil)
|
|
handler := http.HandlerFunc(ResourceRequestMiddleware(NamespacesHandler, logger, factoryFunc))
|
|
handler.ServeHTTP(rr, req)
|
|
assert.True(t, haveBeenCalled)
|
|
})
|
|
|
|
t.Run("returns merges hardcoded namespaces and custom namespaces", func(t *testing.T) {
|
|
origGetHardCodedNamespaces := services.GetHardCodedNamespaces
|
|
t.Cleanup(func() {
|
|
services.GetHardCodedNamespaces = origGetHardCodedNamespaces
|
|
})
|
|
services.GetHardCodedNamespaces = func() []string {
|
|
return []string{"AWS/EC2", "AWS/ELB"}
|
|
}
|
|
rr := httptest.NewRecorder()
|
|
req := httptest.NewRequest("GET", "/namespaces", nil)
|
|
customNamespaces = "customNamespace1,customNamespace2"
|
|
handler := http.HandlerFunc(ResourceRequestMiddleware(NamespacesHandler, logger, factoryFunc))
|
|
handler.ServeHTTP(rr, req)
|
|
assert.JSONEq(t, `["AWS/EC2", "AWS/ELB", "customNamespace1", "customNamespace2"]`, rr.Body.String())
|
|
})
|
|
|
|
t.Run("sorts result", func(t *testing.T) {
|
|
origGetHardCodedNamespaces := services.GetHardCodedNamespaces
|
|
t.Cleanup(func() {
|
|
services.GetHardCodedNamespaces = origGetHardCodedNamespaces
|
|
})
|
|
services.GetHardCodedNamespaces = func() []string {
|
|
return []string{"AWS/XYZ", "AWS/ELB"}
|
|
}
|
|
rr := httptest.NewRecorder()
|
|
req := httptest.NewRequest("GET", "/namespaces", nil)
|
|
customNamespaces = "DCustomNamespace1,ACustomNamespace2"
|
|
handler := http.HandlerFunc(ResourceRequestMiddleware(NamespacesHandler, logger, factoryFunc))
|
|
handler.ServeHTTP(rr, req)
|
|
assert.JSONEq(t, `["ACustomNamespace2", "AWS/ELB", "AWS/XYZ", "DCustomNamespace1"]`, rr.Body.String())
|
|
})
|
|
}
|