Merge pull request #19591 from mmorel-35/usetesting/etcdutl

etcdutl: address Go 1.24 usetesting issues
This commit is contained in:
Benjamin Wang 2025-03-14 17:29:12 +00:00 committed by GitHub
commit c571cff478
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -15,7 +15,6 @@
package snapshot package snapshot
import ( import (
"context"
"errors" "errors"
"path/filepath" "path/filepath"
"strconv" "strconv"
@ -136,7 +135,7 @@ func TestSnapshotStatusTotalKey(t *testing.T) {
Key: []byte(key), Key: []byte(key),
Value: val, Value: val,
} }
_, err := srv.Put(context.TODO(), &req) _, err := srv.Put(t.Context(), &req)
require.NoError(t, err) require.NoError(t, err)
} }
} }
@ -150,10 +149,10 @@ func TestSnapshotStatusTotalKey(t *testing.T) {
key := []byte("key1") key := []byte("key1")
for i := 0; i < 3; i++ { for i := 0; i < 3; i++ {
if i < 2 { if i < 2 {
_, err := srv.Put(context.TODO(), &etcdserverpb.PutRequest{Key: key, Value: []byte(strconv.Itoa(i))}) _, err := srv.Put(t.Context(), &etcdserverpb.PutRequest{Key: key, Value: []byte(strconv.Itoa(i))})
require.NoError(t, err) require.NoError(t, err)
} else { } else {
_, err := srv.DeleteRange(context.TODO(), &etcdserverpb.DeleteRangeRequest{Key: key}) _, err := srv.DeleteRange(t.Context(), &etcdserverpb.DeleteRangeRequest{Key: key})
require.NoError(t, err) require.NoError(t, err)
} }
} }
@ -166,9 +165,9 @@ func TestSnapshotStatusTotalKey(t *testing.T) {
// key1: create -> delete -> re-create -> delete // key1: create -> delete -> re-create -> delete
key := []byte("key1") key := []byte("key1")
for i := 0; i < 2; i++ { for i := 0; i < 2; i++ {
_, err := srv.Put(context.TODO(), &etcdserverpb.PutRequest{Key: key, Value: make([]byte, 1)}) _, err := srv.Put(t.Context(), &etcdserverpb.PutRequest{Key: key, Value: make([]byte, 1)})
require.NoError(t, err) require.NoError(t, err)
_, err = srv.DeleteRange(context.TODO(), &etcdserverpb.DeleteRangeRequest{Key: key}) _, err = srv.DeleteRange(t.Context(), &etcdserverpb.DeleteRangeRequest{Key: key})
require.NoError(t, err) require.NoError(t, err)
} }
}, },
@ -181,10 +180,10 @@ func TestSnapshotStatusTotalKey(t *testing.T) {
key := []byte("key1") key := []byte("key1")
for i := 0; i < 5; i++ { for i := 0; i < 5; i++ {
if i%2 == 0 { if i%2 == 0 {
_, err := srv.Put(context.TODO(), &etcdserverpb.PutRequest{Key: key, Value: make([]byte, 1)}) _, err := srv.Put(t.Context(), &etcdserverpb.PutRequest{Key: key, Value: make([]byte, 1)})
require.NoError(t, err) require.NoError(t, err)
} else { } else {
_, err := srv.DeleteRange(context.TODO(), &etcdserverpb.DeleteRangeRequest{Key: key}) _, err := srv.DeleteRange(t.Context(), &etcdserverpb.DeleteRangeRequest{Key: key})
require.NoError(t, err) require.NoError(t, err)
} }
} }
@ -195,11 +194,11 @@ func TestSnapshotStatusTotalKey(t *testing.T) {
name: "mixed deletions", name: "mixed deletions",
prepare: func(srv *etcdserver.EtcdServer) { prepare: func(srv *etcdserver.EtcdServer) {
// Put("key1") -> Put("key2")-> Delete("key1") // Put("key1") -> Put("key2")-> Delete("key1")
_, err := srv.Put(context.TODO(), &etcdserverpb.PutRequest{Key: []byte("key1"), Value: make([]byte, 1)}) _, err := srv.Put(t.Context(), &etcdserverpb.PutRequest{Key: []byte("key1"), Value: make([]byte, 1)})
require.NoError(t, err) require.NoError(t, err)
_, err = srv.Put(context.TODO(), &etcdserverpb.PutRequest{Key: []byte("key2"), Value: make([]byte, 1)}) _, err = srv.Put(t.Context(), &etcdserverpb.PutRequest{Key: []byte("key2"), Value: make([]byte, 1)})
require.NoError(t, err) require.NoError(t, err)
_, err = srv.DeleteRange(context.TODO(), &etcdserverpb.DeleteRangeRequest{Key: []byte("key1")}) _, err = srv.DeleteRange(t.Context(), &etcdserverpb.DeleteRangeRequest{Key: []byte("key1")})
require.NoError(t, err) require.NoError(t, err)
}, },
expected: 1, expected: 1,
@ -228,7 +227,7 @@ func insertKeys(t *testing.T, numKeys, valueSize int) func(*etcdserver.EtcdServe
Key: []byte(strconv.Itoa(i)), Key: []byte(strconv.Itoa(i)),
Value: val, Value: val,
} }
_, err := srv.Put(context.TODO(), &req) _, err := srv.Put(t.Context(), &req)
require.NoError(t, err) require.NoError(t, err)
} }
} }