lnwire: add benchmark for Warning

This commit is contained in:
Matt Morehouse 2022-11-11 11:42:55 -06:00
parent 404e18afab
commit 258cf6466d
No known key found for this signature in database
GPG Key ID: CC8ECA224831C982

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()