mirrors/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus.git synced 2025-08-06 14:17:12 +02:00
Code Issues Projects Releases Wiki Activity

Fix flaky TestReadClient/timeout test

Browse Source 
Signed-off-by: Patryk Prus <p@trykpr.us>
This commit is contained in:
Patryk Prus 2025-03-18 13:32:17 -04:00
parent d77d2a2d31
commit ebeeb1c823
No known key found for this signature in database
GPG Key ID: 795650115CA6A58F
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
storage/remote/client_test.go
View File

@ -332,9 +332,9 @@ func TestReadClient(t *testing.T) {
}, },
{ {
name: "timeout", name: "timeout",
httpHandler: sampledResponseHTTPHandler(t), httpHandler: delayedResponseHTTPHandler(t, 15*time.Millisecond),
timeout: time.Nanosecond, timeout: 5 * time.Millisecond,
expectedErrorContains: "context deadline exceeded: request timed out after 1ns", expectedErrorContains: "context deadline exceeded: request timed out after 5ms",
}, },
} }
@ -448,3 +448,16 @@ func sampledResponseHTTPHandler(t *testing.T) http.HandlerFunc {
require.NoError(t, err) require.NoError(t, err)
} }
} }
func delayedResponseHTTPHandler(t *testing.T, delay time.Duration) http.HandlerFunc {
return func(w http.ResponseWriter, _ *http.Request) {
time.Sleep(delay)
w.Header().Set("Content-Type", "application/x-protobuf")
b, err := proto.Marshal(&prompb.ReadResponse{})
require.NoError(t, err)
_, err = w.Write(snappy.Encode(nil, b))
require.NoError(t, err)
}
}