kvdb/etcd: replace defer cleanup with t.Cleanup

Signed-off-by: Eng Zer Jun <engzerjun@gmail.com>
This commit is contained in:
Eng Zer Jun committed 2022-10-13 17:47:03 +08:00
1 parent 945180f0ea
commit 889cb0af47
5 files changed
+17 -34

No files matched your search

+10 -15
View File
@@ -26,11 +26,10 @@ func TestPutToEmpty(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background())
txQueue := NewCommitQueue(ctx)
defer func() {
t.Cleanup(func() {
cancel()
f.Cleanup()
txQueue.Stop()
}()
})
db, err := newEtcdBackend(ctx, f.BackendConfig())
require.NoError(t, err)
@@ -54,11 +53,10 @@ func TestGetPutDel(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background())
txQueue := NewCommitQueue(ctx)
defer func() {
t.Cleanup(func() {
cancel()
f.Cleanup()
txQueue.Stop()
}()
})
testKeyValues := []KV{
{"a", "1"},
@@ -156,11 +154,10 @@ func testFirstLastNextPrev(t *testing.T, prefetchKeys []string,
ctx, cancel := context.WithCancel(context.Background())
txQueue := NewCommitQueue(ctx)
defer func() {
t.Cleanup(func() {
cancel()
f.Cleanup()
txQueue.Stop()
}()
})
testKeyValues := []KV{
{"kb", "1"},
@@ -331,11 +328,10 @@ func TestCommitError(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background())
txQueue := NewCommitQueue(ctx)
defer func() {
t.Cleanup(func() {
cancel()
f.Cleanup()
txQueue.Stop()
}()
})
db, err := newEtcdBackend(ctx, f.BackendConfig())
require.NoError(t, err)
@@ -381,11 +377,10 @@ func TestManualTxError(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background())
txQueue := NewCommitQueue(ctx)
defer func() {
t.Cleanup(func() {
cancel()
f.Cleanup()
txQueue.Stop()
}()
})
db, err := newEtcdBackend(ctx, f.BackendConfig())
require.NoError(t, err)