lnwire: add benchmark for Warning

This commit is contained in:
Matt Morehouse committed 2022-11-11 11:56:49 -06:00
1 parent 404e18afab
commit 258cf6466d
1 file changed
+14
+14
View File
@@ -259,6 +259,7 @@ func BenchmarkReadMessage(b *testing.B) {
func makeAllMessages(t testing.TB, r *rand.Rand) []lnwire.Message {
msgAll := []lnwire.Message{}
msgAll = append(msgAll, newMsgWarning(t, r))
msgAll = append(msgAll, newMsgInit(t, r))
msgAll = append(msgAll, newMsgError(t, r))
msgAll = append(msgAll, newMsgPing(t, r))
@@ -293,6 +294,19 @@ func makeAllMessages(t testing.TB, r *rand.Rand) []lnwire.Message {
return msgAll
}
func newMsgWarning(tb testing.TB, r io.Reader) *lnwire.Warning {
tb.Helper()
msg := lnwire.NewWarning()
_, err := r.Read(msg.ChanID[:])
require.NoError(tb, err, "unable to generate chan id")
msg.Data = createExtraData(tb, r)
return msg
}
func newMsgInit(t testing.TB, r io.Reader) *lnwire.Init {
t.Helper()