dbg

replicate · Dec 12, 2024 · 20f8db3 · 20f8db3
1 parent c3c9779
commit 20f8db3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/internal/tests/prediction_test.go b/internal/tests/prediction_test.go
@@ -7,6 +7,7 @@ import (
 	"io"
 	"net/http"
 	"testing"
+	"time"
 
 	"github.com/replicate/go/must"
 
@@ -97,6 +98,7 @@ func TestPredictionCrash(t *testing.T) {
 		assert.Equal(t, http.StatusInternalServerError, resp.StatusCode)
 		body := string(must.Get(io.ReadAll(resp.Body)))
 		assert.Equal(t, "Internal Server Error", body)
+		time.Sleep(time.Second)
 		assert.Equal(t, "DEFUNCT", ct.HealthCheck().Status)
 	} else {
 		resp := ct.Prediction(map[string]any{"i": 1, "s": "bar"})