Fix status code handling

castai · Dec 24, 2021 · 8413b3f · 8413b3f
1 parent d72d86a
commit 8413b3f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/client.go b/client.go
@@ -118,7 +118,7 @@ func (p *Client) Write(ctx context.Context, req *WriteRequest, options ...WriteO
 	}
 	defer httpResp.Body.Close()
 
-	if st := httpResp.StatusCode; st != http.StatusOK {
+	if st := httpResp.StatusCode; st/100 != 2 {
 		msg, _ := ioutil.ReadAll(httpResp.Body)
 		return nil, &WriteError{
 			err:  fmt.Errorf("promwrite: expected status %d, got %d: %s", http.StatusOK, st, string(msg)),

diff --git a/client_test.go b/client_test.go
@@ -119,7 +119,7 @@ func TestClient(t *testing.T) {
 			r.NoError(err)
 			receivedWriteRequest <- parsed
 			receivedHeaders <- req.Header
-			w.WriteHeader(http.StatusOK)
+			w.WriteHeader(http.StatusAccepted)
 		}))
 		defer srv.Close()