mirrors/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus.git synced 2025-08-05 21:57:09 +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",
httpHandler: sampledResponseHTTPHandler(t),
timeout: time.Nanosecond,
expectedErrorContains: "context deadline exceeded: request timed out after 1ns",
httpHandler: delayedResponseHTTPHandler(t, 15*time.Millisecond),
timeout: 5 * time.Millisecond,
expectedErrorContains: "context deadline exceeded: request timed out after 5ms",
},
}
@ -448,3 +448,16 @@ func sampledResponseHTTPHandler(t *testing.T) http.HandlerFunc {
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)
}
}