Skip to content

Commit 3869f0e

Browse files
authored
Merge pull request #18752 from mmorel-35/tools/testifier/require-error
fix: use require.NoError instead of t.Fatal(err) in contrib and tools packages
2 parents c09b07f + 63dad79 commit 3869f0e

File tree

3 files changed

+21
-42
lines changed

3 files changed

+21
-42
lines changed

contrib/raftexample/kvstore_test.go

+5-6
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,8 @@ package main
1717
import (
1818
"reflect"
1919
"testing"
20+
21+
"github.com/stretchr/testify/require"
2022
)
2123

2224
func Test_kvstore_snapshot(t *testing.T) {
@@ -29,14 +31,11 @@ func Test_kvstore_snapshot(t *testing.T) {
2931
}
3032

3133
data, err := s.getSnapshot()
32-
if err != nil {
33-
t.Fatal(err)
34-
}
34+
require.NoError(t, err)
3535
s.kvStore = nil
3636

37-
if err := s.recoverFromSnapshot(data); err != nil {
38-
t.Fatal(err)
39-
}
37+
err = s.recoverFromSnapshot(data)
38+
require.NoError(t, err)
4039
v, _ = s.Lookup("foo")
4140
if v != "bar" {
4241
t.Fatalf("foo has unexpected value, got %s", v)

contrib/raftexample/raftexample_test.go

+8-15
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,8 @@ import (
2525
"testing"
2626
"time"
2727

28+
"github.com/stretchr/testify/require"
29+
2830
"go.etcd.io/raft/v3/raftpb"
2931
)
3032

@@ -96,9 +98,8 @@ func (clus *cluster) Close() (err error) {
9698

9799
func (clus *cluster) closeNoErrors(t *testing.T) {
98100
t.Log("closing cluster...")
99-
if err := clus.Close(); err != nil {
100-
t.Fatal(err)
101-
}
101+
err := clus.Close()
102+
require.NoError(t, err)
102103
t.Log("closing cluster [done]")
103104
}
104105

@@ -201,27 +202,19 @@ func TestPutAndGetKeyValue(t *testing.T) {
201202
cli := srv.Client()
202203

203204
req, err := http.NewRequest(http.MethodPut, url, body)
204-
if err != nil {
205-
t.Fatal(err)
206-
}
205+
require.NoError(t, err)
207206
req.Header.Set("Content-Type", "text/html; charset=utf-8")
208207
_, err = cli.Do(req)
209-
if err != nil {
210-
t.Fatal(err)
211-
}
208+
require.NoError(t, err)
212209

213210
// wait for a moment for processing message, otherwise get would be failed.
214211
<-time.After(time.Second)
215212

216213
resp, err := cli.Get(url)
217-
if err != nil {
218-
t.Fatal(err)
219-
}
214+
require.NoError(t, err)
220215

221216
data, err := io.ReadAll(resp.Body)
222-
if err != nil {
223-
t.Fatal(err)
224-
}
217+
require.NoError(t, err)
225218
defer resp.Body.Close()
226219

227220
if gotValue := string(data); wantValue != gotValue {

tools/etcd-dump-logs/etcd-dump-log_test.go

+8-21
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ import (
2323
"testing"
2424

2525
"github.com/stretchr/testify/assert"
26+
"github.com/stretchr/testify/require"
2627
"go.uber.org/zap/zaptest"
2728

2829
"go.etcd.io/etcd/api/v3/authpb"
@@ -36,9 +37,7 @@ import (
3637
func TestEtcdDumpLogEntryType(t *testing.T) {
3738
// directory where the command is
3839
binDir, err := os.Getwd()
39-
if err != nil {
40-
t.Fatal(err)
41-
}
40+
require.NoError(t, err)
4241

4342
// TODO(ptabor): The test does not run by default from ./scripts/test.sh.
4443
dumpLogsBinary := path.Join(binDir + "/etcd-dump-logs")
@@ -79,13 +78,9 @@ func TestEtcdDumpLogEntryType(t *testing.T) {
7978
t.Run(argtest.name, func(t *testing.T) {
8079
cmd := exec.Command(dumpLogsBinary, argtest.args...)
8180
actual, err := cmd.CombinedOutput()
82-
if err != nil {
83-
t.Fatal(err)
84-
}
81+
require.NoError(t, err)
8582
expected, err := os.ReadFile(path.Join(binDir, argtest.fileExpected))
86-
if err != nil {
87-
t.Fatal(err)
88-
}
83+
require.NoError(t, err)
8984

9085
assert.EqualValues(t, string(expected), string(actual))
9186
// The output files contains a lot of trailing whitespaces... difficult to diagnose without printing them explicitly.
@@ -98,21 +93,15 @@ func TestEtcdDumpLogEntryType(t *testing.T) {
9893
func mustCreateWALLog(t *testing.T, path string) {
9994
memberdir := filepath.Join(path, "member")
10095
err := os.Mkdir(memberdir, 0744)
101-
if err != nil {
102-
t.Fatal(err)
103-
}
96+
require.NoError(t, err)
10497
waldir := walDir(path)
10598
snapdir := snapDir(path)
10699

107100
w, err := wal.Create(zaptest.NewLogger(t), waldir, nil)
108-
if err != nil {
109-
t.Fatal(err)
110-
}
101+
require.NoError(t, err)
111102

112103
err = os.Mkdir(snapdir, 0744)
113-
if err != nil {
114-
t.Fatal(err)
115-
}
104+
require.NoError(t, err)
116105

117106
ents := make([]raftpb.Entry, 0)
118107

@@ -124,9 +113,7 @@ func mustCreateWALLog(t *testing.T, path string) {
124113

125114
// force commit newly appended entries
126115
err = w.Save(raftpb.HardState{}, ents)
127-
if err != nil {
128-
t.Fatal(err)
129-
}
116+
require.NoError(t, err)
130117
w.Close()
131118
}
132119

0 commit comments

Comments
 (0)