Function TestRunHealthCheck

cmd/tempo/health_test.go:12–34

(t *testing.T)

)

func TestRunHealthCheck(t *testing.T) {
	tests := []struct {
		name       string
		statusCode int
		body       string
		wantExit   int
	}{
		{"healthy", http.StatusOK, "", 0},
		{"unhealthy", http.StatusServiceUnavailable, "not ready", 1},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
				w.WriteHeader(tt.statusCode)
				fmt.Fprint(w, tt.body)
			}))
			defer server.Close()

			assert.Equal(t, tt.wantExit, RunHealthCheck(server.URL+"/ready"))
		})
	}
}

func TestRunHealthCheckDefaultURL(t *testing.T) {
	assert.Equal(t, 1, RunHealthCheck(""))