1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
package mock_gitlab
import (
"net/http"
"net/http/httptest"
"net/url"
"testing"
"github.com/hashicorp/go-retryablehttp"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/gitlab"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/tool/testing/testhelpers"
"go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp"
"go.opentelemetry.io/otel/propagation"
)
func SetupClient(t *testing.T, pattern string, handler func(http.ResponseWriter, *http.Request)) *gitlab.Client {
propagator := propagation.NewCompositeTextMapPropagator(propagation.TraceContext{}, propagation.Baggage{})
r := http.NewServeMux()
r.HandleFunc(pattern, handler)
h := otelhttp.NewHandler(r, "gitlab-request", otelhttp.WithPropagators(propagator))
s := httptest.NewServer(h)
t.Cleanup(s.Close)
u, err := url.Parse(s.URL)
require.NoError(t, err)
return gitlab.NewClient(u, []byte(testhelpers.AuthSecretKey),
gitlab.WithUserAgent(testhelpers.KasUserAgent),
gitlab.WithTextMapPropagator(propagator),
gitlab.WithRetryConfig(gitlab.RetryConfig{
CheckRetry: retryablehttp.DefaultRetryPolicy,
}), // disable retries
)
}
|
