mirrors/crowdsec
1
0
Fork 0
mirror of https://github.com/crowdsecurity/crowdsec.git synced 2025-05-11 20:36:12 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

context propagation: apic, unit tests (#3271)

Browse source 
* context propagation: apic

* context propagation: unit tests
This commit is contained in:
mmetc 2024-10-03 14:51:53 +02:00 • committed by GitHub
parent af3116d1a7
commit 06adbe031d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 189 additions and 179 deletions
Download patch file Download diff file Expand all files Collapse all files

12
pkg/apiserver/usage_metrics_test.go
View file

@ -187,14 +187,14 @@ func TestLPMetrics(t *testing.T) {
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
lapi := SetupLAPITest(t)
lapi := SetupLAPITest(t, ctx)
dbClient, err := database.NewClient(context.Background(), lapi.DBConfig)
dbClient, err := database.NewClient(ctx, lapi.DBConfig)
if err != nil {
t.Fatalf("unable to create database client: %s", err)
}
w := lapi.RecordResponse(t, http.MethodPost, "/v1/usage-metrics", strings.NewReader(tt.body), tt.authType)
w := lapi.RecordResponse(t, ctx, http.MethodPost, "/v1/usage-metrics", strings.NewReader(tt.body), tt.authType)
assert.Equal(t, tt.expectedStatusCode, w.Code)
assert.Contains(t, w.Body.String(), tt.expectedResponse)
@ -359,14 +359,14 @@ func TestRCMetrics(t *testing.T) {
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
lapi := SetupLAPITest(t)
lapi := SetupLAPITest(t, ctx)
dbClient, err := database.NewClient(context.Background(), lapi.DBConfig)
dbClient, err := database.NewClient(ctx, lapi.DBConfig)
if err != nil {
t.Fatalf("unable to create database client: %s", err)
}
w := lapi.RecordResponse(t, http.MethodPost, "/v1/usage-metrics", strings.NewReader(tt.body), tt.authType)
w := lapi.RecordResponse(t, ctx, http.MethodPost, "/v1/usage-metrics", strings.NewReader(tt.body), tt.authType)
assert.Equal(t, tt.expectedStatusCode, w.Code)
assert.Contains(t, w.Body.String(), tt.expectedResponse)