diff --git a/tlv/bench_test.go b/tlv/bench_test.go index 132ef784..6233c62a 100644 --- a/tlv/bench_test.go +++ b/tlv/bench_test.go @@ -10,6 +10,7 @@ import ( "github.com/lightningnetwork/lnd/tlv" "github.com/lightningnetwork/lnd/watchtower/blob" "github.com/lightningnetwork/lnd/watchtower/wtwire" + "github.com/stretchr/testify/require" ) // CreateSessionTLV mirrors the wtwire.CreateSession message, but uses TLV for @@ -104,7 +105,7 @@ func BenchmarkEncodeCreateSession(t *testing.B) { for i := 0; i < t.N; i++ { err = m.Encode(ioutil.Discard, 0) } - _ = err + require.NoError(t, err) } // BenchmarkEncodeCreateSessionTLV benchmarks encoding of the TLV CreateSession. @@ -118,7 +119,7 @@ func BenchmarkEncodeCreateSessionTLV(t *testing.B) { for i := 0; i < t.N; i++ { err = m.Encode(ioutil.Discard) } - _ = err + require.NoError(t, err) } // BenchmarkDecodeCreateSession benchmarks encoding of the non-TLV @@ -127,18 +128,19 @@ func BenchmarkDecodeCreateSession(t *testing.B) { m := &wtwire.CreateSession{} var b bytes.Buffer - m.Encode(&b, 0) + err := m.Encode(&b, 0) + require.NoError(t, err) + r := bytes.NewReader(b.Bytes()) t.ReportAllocs() t.ResetTimer() - var err error for i := 0; i < t.N; i++ { r.Seek(0, 0) err = m.Decode(r, 0) } - _ = err + require.NoError(t, err) } // BenchmarkDecodeCreateSessionTLV benchmarks decoding of the TLV CreateSession. @@ -146,8 +148,9 @@ func BenchmarkDecodeCreateSessionTLV(t *testing.B) { m := NewCreateSessionTLV() var b bytes.Buffer - var err error - m.Encode(&b) + err := m.Encode(&b) + require.NoError(t, err) + r := bytes.NewReader(b.Bytes()) t.ReportAllocs() @@ -157,5 +160,5 @@ func BenchmarkDecodeCreateSessionTLV(t *testing.B) { r.Seek(0, 0) err = m.Decode(r) } - _ = err + require.NoError(t, err) }