internal/worker: use TenantChannelMiddleware

This commit is contained in:
Sanne Raymaekers 2023-06-23 00:37:01 +02:00
parent 06038b2af6
commit 9594156baf
2 changed files with 8 additions and 2 deletions

View file

@ -176,6 +176,7 @@ func runNextJob(t *testing.T, jobs []uuid.UUID, workerHandler http.Handler, orgI
Method: http.MethodPatch,
Path: job.Location,
RequestBody: test.JSONRequestBody(`{"result": {"job_result":{}}}`),
Context: reqContext(orgID),
ExpectedStatus: http.StatusOK,
}.Do(t)
}

View file

@ -100,8 +100,13 @@ func (s *Server) Handler() http.Handler {
server: s,
}
statusMW := prometheus.StatusMiddleware(prometheus.WorkerSubsystem)
api.RegisterHandlers(e.Group(api.BasePath, statusMW), &handler)
mws := []echo.MiddlewareFunc{
prometheus.StatusMiddleware(prometheus.WorkerSubsystem),
}
if s.config.JWTEnabled {
mws = append(mws, auth.TenantChannelMiddleware(s.config.TenantProviderFields, api.HTTPError(api.ErrorTenantNotFound)))
}
api.RegisterHandlers(e.Group(api.BasePath, mws...), &handler)
return e
}