Skip to content

Commit 39d9cd7

Browse files
authored
Merge pull request #18968 from mmorel-35/testifier/contrib
fix: use testify instead of t.Fatal or t.Error in contrib package
2 parents 854bdd6 + 7f9efcf commit 39d9cd7

File tree

3 files changed

+8
-16
lines changed

3 files changed

+8
-16
lines changed

contrib/raftexample/kvstore_test.go

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,7 @@ func Test_kvstore_snapshot(t *testing.T) {
2626
s := &kvstore{kvStore: tm}
2727

2828
v, _ := s.Lookup("foo")
29-
if v != "bar" {
30-
t.Fatalf("foo has unexpected value, got %s", v)
31-
}
29+
require.Equalf(t, "bar", v, "foo has unexpected value, got %s", v)
3230

3331
data, err := s.getSnapshot()
3432
require.NoError(t, err)
@@ -37,10 +35,6 @@ func Test_kvstore_snapshot(t *testing.T) {
3735
err = s.recoverFromSnapshot(data)
3836
require.NoError(t, err)
3937
v, _ = s.Lookup("foo")
40-
if v != "bar" {
41-
t.Fatalf("foo has unexpected value, got %s", v)
42-
}
43-
if !reflect.DeepEqual(s.kvStore, tm) {
44-
t.Fatalf("store expected %+v, got %+v", tm, s.kvStore)
45-
}
38+
require.Equalf(t, "bar", v, "foo has unexpected value, got %s", v)
39+
require.Truef(t, reflect.DeepEqual(s.kvStore, tm), "store expected %+v, got %+v", tm, s.kvStore)
4640
}

contrib/raftexample/raft_test.go

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,8 @@ import (
1818
"reflect"
1919
"testing"
2020

21+
"github.com/stretchr/testify/require"
22+
2123
"go.etcd.io/raft/v3/raftpb"
2224
)
2325

@@ -121,10 +123,7 @@ func TestProcessMessages(t *testing.T) {
121123
}
122124

123125
outputMessages := rn.processMessages(tc.InputMessages)
124-
125-
if !reflect.DeepEqual(outputMessages, tc.ExpectedMessages) {
126-
t.Fatalf("Unexpected messages, expected: %v, got %v", tc.ExpectedMessages, outputMessages)
127-
}
126+
require.Truef(t, reflect.DeepEqual(outputMessages, tc.ExpectedMessages), "Unexpected messages, expected: %v, got %v", tc.ExpectedMessages, outputMessages)
128127
})
129128
}
130129
}

contrib/raftexample/raftexample_test.go

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -217,9 +217,8 @@ func TestPutAndGetKeyValue(t *testing.T) {
217217
require.NoError(t, err)
218218
defer resp.Body.Close()
219219

220-
if gotValue := string(data); wantValue != gotValue {
221-
t.Fatalf("expect %s, got %s", wantValue, gotValue)
222-
}
220+
gotValue := string(data)
221+
require.Equalf(t, wantValue, gotValue, "expect %s, got %s", wantValue, gotValue)
223222
}
224223

225224
// TestAddNewNode tests adding new node to the existing cluster.

0 commit comments

Comments
 (0)