mirror of https://github.com/status-im/consul.git
115 lines
2.9 KiB
Go
115 lines
2.9 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
colpb "go.opentelemetry.io/proto/otlp/collector/metrics/v1"
|
|
metricpb "go.opentelemetry.io/proto/otlp/metrics/v1"
|
|
"google.golang.org/protobuf/proto"
|
|
|
|
"github.com/hashicorp/consul/version"
|
|
)
|
|
|
|
func TestNewMetricsClient(t *testing.T) {
|
|
for name, test := range map[string]struct {
|
|
wantErr string
|
|
cfg CloudConfig
|
|
ctx context.Context
|
|
}{
|
|
"success": {
|
|
cfg: &MockCloudCfg{},
|
|
ctx: context.Background(),
|
|
},
|
|
"failsWithoutCloudCfg": {
|
|
wantErr: "failed to init telemetry client: provide valid cloudCfg (Cloud Configuration for TLS)",
|
|
cfg: nil,
|
|
ctx: context.Background(),
|
|
},
|
|
"failsWithoutContext": {
|
|
wantErr: "failed to init telemetry client: provide a valid context",
|
|
cfg: MockCloudCfg{},
|
|
ctx: nil,
|
|
},
|
|
"failsHCPConfig": {
|
|
wantErr: "failed to init telemetry client",
|
|
cfg: MockCloudCfg{
|
|
ConfigErr: fmt.Errorf("test bad hcp config"),
|
|
},
|
|
ctx: context.Background(),
|
|
},
|
|
"failsBadResource": {
|
|
wantErr: "failed to init telemetry client",
|
|
cfg: MockCloudCfg{
|
|
ResourceErr: fmt.Errorf("test bad resource"),
|
|
},
|
|
ctx: context.Background(),
|
|
},
|
|
} {
|
|
t.Run(name, func(t *testing.T) {
|
|
client, err := NewMetricsClient(test.cfg, test.ctx)
|
|
if test.wantErr != "" {
|
|
require.Error(t, err)
|
|
require.Contains(t, err.Error(), test.wantErr)
|
|
return
|
|
}
|
|
|
|
require.Nil(t, err)
|
|
require.NotNil(t, client)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestExportMetrics(t *testing.T) {
|
|
for name, test := range map[string]struct {
|
|
wantErr string
|
|
status int
|
|
}{
|
|
"success": {
|
|
status: http.StatusOK,
|
|
},
|
|
"failsWithNonRetryableError": {
|
|
status: http.StatusBadRequest,
|
|
wantErr: "failed to export metrics: code 400",
|
|
},
|
|
} {
|
|
t.Run(name, func(t *testing.T) {
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
require.Equal(t, r.Header.Get("content-type"), "application/x-protobuf")
|
|
require.Equal(t, r.Header.Get("x-hcp-resource-id"), testResourceID)
|
|
require.Equal(t, r.Header.Get("x-channel"), fmt.Sprintf("consul/%s", version.GetHumanVersion()))
|
|
require.Equal(t, r.Header.Get("Authorization"), "Bearer test-token")
|
|
|
|
body := colpb.ExportMetricsServiceResponse{}
|
|
bytes, err := proto.Marshal(&body)
|
|
|
|
require.NoError(t, err)
|
|
|
|
w.Header().Set("Content-Type", "application/x-protobuf")
|
|
w.WriteHeader(test.status)
|
|
w.Write(bytes)
|
|
}))
|
|
defer srv.Close()
|
|
|
|
client, err := NewMetricsClient(MockCloudCfg{}, context.Background())
|
|
require.NoError(t, err)
|
|
|
|
ctx := context.Background()
|
|
metrics := &metricpb.ResourceMetrics{}
|
|
err = client.ExportMetrics(ctx, metrics, srv.URL)
|
|
|
|
if test.wantErr != "" {
|
|
require.Error(t, err)
|
|
require.Contains(t, err.Error(), test.wantErr)
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
})
|
|
}
|
|
}
|