Function tallymanHandler

enterprise/coderd/usage/publisher_test.go:689–734

(t *testing.T, expectDeploymentID string, expectLicenseJWT string, handler func(req usagetypes.TallymanV1IngestRequest) any)

}

func tallymanHandler(t *testing.T, expectDeploymentID string, expectLicenseJWT string, handler func(req usagetypes.TallymanV1IngestRequest) any) http.Handler {
	t.Helper()
	return http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
		t.Helper()
		licenseJWT := r.Header.Get(usagetypes.TallymanCoderLicenseKeyHeader)
		if expectLicenseJWT != "" && !assert.Equal(t, expectLicenseJWT, licenseJWT, "license JWT in request did not match") {
			rw.WriteHeader(http.StatusUnauthorized)
			_ = json.NewEncoder(rw).Encode(usagetypes.TallymanV1Response{
				Message: "license JWT in request did not match",
			})
			return
		}

		deploymentID := r.Header.Get(usagetypes.TallymanCoderDeploymentIDHeader)
		if expectDeploymentID != "" && !assert.Equal(t, expectDeploymentID, deploymentID, "deployment ID in request did not match") {
			rw.WriteHeader(http.StatusUnauthorized)
			_ = json.NewEncoder(rw).Encode(usagetypes.TallymanV1Response{
				Message: "deployment ID in request did not match",
			})
			return
		}

		var req usagetypes.TallymanV1IngestRequest
		err := json.NewDecoder(r.Body).Decode(&req)
		if !assert.NoError(t, err, "could not decode request body") {
			rw.WriteHeader(http.StatusBadRequest)
			_ = json.NewEncoder(rw).Encode(usagetypes.TallymanV1Response{
				Message: "could not decode request body",
			})
			return
		}

		resp := handler(req)
		switch resp.(type) {
		case usagetypes.TallymanV1Response:
			rw.WriteHeader(http.StatusInternalServerError)
		default:
			rw.WriteHeader(http.StatusOK)
		}
		err = json.NewEncoder(rw).Encode(resp)
		if !assert.NoError(t, err, "could not encode response body") {
			rw.WriteHeader(http.StatusInternalServerError)
			return
		}
	})
}

func tallymanAcceptAllHandler(req usagetypes.TallymanV1IngestRequest) usagetypes.TallymanV1IngestResponse {
	acceptedEvents := make([]usagetypes.TallymanV1IngestAcceptedEvent, len(req.Events))