mirror of
https://github.com/zalando-incubator/kube-metrics-adapter.git
synced 2024-12-22 11:06:04 +00:00
102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package collector
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
autoscalingv2 "k8s.io/api/autoscaling/v2"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
)
|
|
|
|
type testExternalMetricsHandler struct {
|
|
values []int64
|
|
test *testing.T
|
|
}
|
|
|
|
func (t testExternalMetricsHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
response, err := json.Marshal(testMetricResponse{t.values})
|
|
require.NoError(t.test, err)
|
|
w.Header().Set("Content-Type", "application/json")
|
|
_, err = w.Write(response)
|
|
require.NoError(t.test, err)
|
|
}
|
|
|
|
func makeHTTPTestServer(t *testing.T, values []int64) string {
|
|
server := httptest.NewServer(&testExternalMetricsHandler{values: values, test: t})
|
|
return server.URL
|
|
}
|
|
|
|
func TestHTTPCollector(t *testing.T) {
|
|
for _, tc := range []struct {
|
|
name string
|
|
values []int64
|
|
output int
|
|
aggregator string
|
|
}{
|
|
{
|
|
name: "basic",
|
|
values: []int64{3},
|
|
output: 3,
|
|
aggregator: "sum",
|
|
},
|
|
{
|
|
name: "sum",
|
|
values: []int64{3, 5, 6},
|
|
aggregator: "sum",
|
|
output: 14,
|
|
},
|
|
{
|
|
name: "average",
|
|
values: []int64{3, 5, 6},
|
|
aggregator: "sum",
|
|
output: 14,
|
|
},
|
|
} {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
testServer := makeHTTPTestServer(t, tc.values)
|
|
plugin, err := NewHTTPCollectorPlugin()
|
|
require.NoError(t, err)
|
|
testConfig := makeTestHTTPCollectorConfig(testServer, tc.aggregator)
|
|
hpa := &autoscalingv2.HorizontalPodAutoscaler{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Namespace: "default",
|
|
},
|
|
}
|
|
collector, err := plugin.NewCollector(context.Background(), hpa, testConfig, testInterval)
|
|
require.NoError(t, err)
|
|
metrics, err := collector.GetMetrics(context.Background())
|
|
require.NoError(t, err)
|
|
require.NotNil(t, metrics)
|
|
require.Len(t, metrics, 1)
|
|
require.EqualValues(t, metrics[0].External.Value.Value(), tc.output)
|
|
})
|
|
}
|
|
}
|
|
|
|
func makeTestHTTPCollectorConfig(endpoint, aggregator string) *MetricConfig {
|
|
config := &MetricConfig{
|
|
MetricTypeName: MetricTypeName{
|
|
Type: autoscalingv2.ExternalMetricSourceType,
|
|
Metric: autoscalingv2.MetricIdentifier{
|
|
Name: "test-metric",
|
|
Selector: &v1.LabelSelector{
|
|
MatchLabels: map[string]string{"type": HTTPJSONPathType},
|
|
},
|
|
},
|
|
},
|
|
Config: map[string]string{
|
|
HTTPJsonPathAnnotationKey: "$.values",
|
|
HTTPEndpointAnnotationKey: endpoint,
|
|
},
|
|
}
|
|
if aggregator != "" {
|
|
config.Config["aggregator"] = aggregator
|
|
}
|
|
return config
|
|
}
|