From 4c2e712581a04553db653883ddb3da185f1bbb5f Mon Sep 17 00:00:00 2001 From: Matt Morehouse Date: Fri, 19 May 2023 12:07:49 -0500 Subject: [PATCH] wtwire: use require package for fuzz tests Simplify code by using the require package instead of t.Fatal(). --- watchtower/wtwire/fuzz_test.go | 22 ++++++---------------- 1 file changed, 6 insertions(+), 16 deletions(-) diff --git a/watchtower/wtwire/fuzz_test.go b/watchtower/wtwire/fuzz_test.go index 06b0b3ffd..5a03b84e3 100644 --- a/watchtower/wtwire/fuzz_test.go +++ b/watchtower/wtwire/fuzz_test.go @@ -3,8 +3,9 @@ package wtwire import ( "bytes" "encoding/binary" - "reflect" "testing" + + "github.com/stretchr/testify/require" ) // prefixWithMsgType takes []byte and adds a wire protocol prefix @@ -43,26 +44,15 @@ func harness(t *testing.T, data []byte, emptyMsg Message) { // We will serialize the message into a new bytes buffer. var b bytes.Buffer - if _, err := WriteMessage(&b, msg, 0); err != nil { - // Could not serialize message into bytes buffer, panic. - t.Fatal(err) - } + _, err = WriteMessage(&b, msg, 0) + require.NoError(t, err) // Deserialize the message from the serialized bytes buffer, and then // assert that the original message is equal to the newly deserialized // message. newMsg, err := ReadMessage(&b, 0) - if err != nil { - // Could not deserialize message from bytes buffer, panic. - t.Fatal(err) - } - - if !reflect.DeepEqual(msg, newMsg) { - // Deserialized message and original message are not - // deeply equal. - t.Fatal("deserialized message and original message " + - "are not deeply equal.") - } + require.NoError(t, err) + require.Equal(t, msg, newMsg) } func FuzzCreateSessionReply(f *testing.F) {