mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
Merge pull request #17235 from fuweid/fix-17223
integration: deflake TestMaintenanceSnapshotCancel
This commit is contained in:
commit
c8b4b168af
@ -26,6 +26,7 @@ import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
"go.uber.org/zap/zaptest"
|
||||
"google.golang.org/grpc"
|
||||
@ -176,6 +177,12 @@ func TestMaintenanceSnapshotCancel(t *testing.T) {
|
||||
}
|
||||
defer rc1.Close()
|
||||
|
||||
// read 16 bytes to ensure that server opens snapshot
|
||||
buf := make([]byte, 16)
|
||||
n, err := rc1.Read(buf)
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, 16, n)
|
||||
|
||||
cancel()
|
||||
_, err = io.Copy(io.Discard, rc1)
|
||||
if err != context.Canceled {
|
||||
|
Loading…
x
Reference in New Issue
Block a user