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

context propagation: pkg/database/machines (#3248)

Browse source
This commit is contained in:
mmetc 2024-09-20 16:00:58 +02:00 • committed by GitHub
parent e2196bdd66
commit fee3debdcc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
11 changed files with 109 additions and 94 deletions
Download patch file Download diff file Expand all files Collapse all files

32
pkg/apiserver/usage_metrics_test.go
View file

@ -30,7 +30,7 @@ func TestLPMetrics(t *testing.T) {
name: "empty metrics for LP",
body: `{
}`,
expectedStatusCode: 400,
expectedStatusCode: http.StatusBadRequest,
expectedResponse: "Missing log processor data",
authType: PASSWORD,
},
@ -50,7 +50,7 @@ func TestLPMetrics(t *testing.T) {
}
]
}`,
expectedStatusCode: 201,
expectedStatusCode: http.StatusCreated,
expectedMetricsCount: 1,
expectedResponse: "",
expectedOSName: "foo",
@ -74,7 +74,7 @@ func TestLPMetrics(t *testing.T) {
}
]
}`,
expectedStatusCode: 201,
expectedStatusCode: http.StatusCreated,
expectedMetricsCount: 1,
expectedResponse: "",
expectedOSName: "foo",
@ -98,7 +98,7 @@ func TestLPMetrics(t *testing.T) {
}
]
}`,
expectedStatusCode: 400,
expectedStatusCode: http.StatusBadRequest,
expectedResponse: "Missing remediation component data",
authType: APIKEY,
},
@ -117,7 +117,7 @@ func TestLPMetrics(t *testing.T) {
}
]
}`,
expectedStatusCode: 201,
expectedStatusCode: http.StatusCreated,
expectedResponse: "",
expectedMetricsCount: 1,
expectedFeatureFlags: "a,b,c",
@ -138,7 +138,7 @@ func TestLPMetrics(t *testing.T) {
}
]
}`,
expectedStatusCode: 422,
expectedStatusCode: http.StatusUnprocessableEntity,
expectedResponse: "log_processors.0.datasources in body is required",
authType: PASSWORD,
},
@ -157,7 +157,7 @@ func TestLPMetrics(t *testing.T) {
}
]
}`,
expectedStatusCode: 201,
expectedStatusCode: http.StatusCreated,
expectedMetricsCount: 1,
expectedOSName: "foo",
expectedOSVersion: "42",
@ -179,7 +179,7 @@ func TestLPMetrics(t *testing.T) {
}
]
}`,
expectedStatusCode: 422,
expectedStatusCode: http.StatusUnprocessableEntity,
expectedResponse: "log_processors.0.os.name in body is required",
authType: PASSWORD,
},
@ -199,7 +199,7 @@ func TestLPMetrics(t *testing.T) {
assert.Equal(t, tt.expectedStatusCode, w.Code)
assert.Contains(t, w.Body.String(), tt.expectedResponse)
machine, _ := dbClient.QueryMachineByID("test")
machine, _ := dbClient.QueryMachineByID(ctx, "test")
metrics, _ := dbClient.GetLPUsageMetricsByMachineID(ctx, "test")
assert.Len(t, metrics, tt.expectedMetricsCount)
@ -233,7 +233,7 @@ func TestRCMetrics(t *testing.T) {
name: "empty metrics for RC",
body: `{
}`,
expectedStatusCode: 400,
expectedStatusCode: http.StatusBadRequest,
expectedResponse: "Missing remediation component data",
authType: APIKEY,
},
@ -251,7 +251,7 @@ func TestRCMetrics(t *testing.T) {
}
]
}`,
expectedStatusCode: 201,
expectedStatusCode: http.StatusCreated,
expectedMetricsCount: 1,
expectedResponse: "",
expectedOSName: "foo",
@ -273,7 +273,7 @@ func TestRCMetrics(t *testing.T) {
}
]
}`,
expectedStatusCode: 201,
expectedStatusCode: http.StatusCreated,
expectedMetricsCount: 1,
expectedResponse: "",
expectedOSName: "foo",
@ -295,7 +295,7 @@ func TestRCMetrics(t *testing.T) {
}
]
}`,
expectedStatusCode: 400,
expectedStatusCode: http.StatusBadRequest,
expectedResponse: "Missing log processor data",
authType: PASSWORD,
},
@ -312,7 +312,7 @@ func TestRCMetrics(t *testing.T) {
}
]
}`,
expectedStatusCode: 201,
expectedStatusCode: http.StatusCreated,
expectedResponse: "",
expectedMetricsCount: 1,
expectedFeatureFlags: "a,b,c",
@ -331,7 +331,7 @@ func TestRCMetrics(t *testing.T) {
}
]
}`,
expectedStatusCode: 201,
expectedStatusCode: http.StatusCreated,
expectedMetricsCount: 1,
expectedOSName: "foo",
expectedOSVersion: "42",
@ -351,7 +351,7 @@ func TestRCMetrics(t *testing.T) {
}
]
}`,
expectedStatusCode: 422,
expectedStatusCode: http.StatusUnprocessableEntity,
expectedResponse: "remediation_components.0.os.name in body is required",
authType: APIKEY,
},