p2p/discover/v5wire: fix test

This commit is contained in:
Felix Lange 2025-04-02 17:16:58 +02:00
parent 31e7ab4be6
commit 64a054b06b

View file

@ -280,8 +280,9 @@ func TestEncodeWhoareyouResend(t *testing.T) {
IDNonce: testIDnonce, IDNonce: testIDnonce,
RecordSeq: 0, RecordSeq: 0,
} }
whoareyou, _ := net.nodeA.encode(t, net.nodeB, challenge) enc, _ := net.nodeA.encode(t, net.nodeB, challenge)
net.nodeB.expectDecode(t, WhoareyouPacket, whoareyou) net.nodeB.expectDecode(t, WhoareyouPacket, enc)
whoareyou1 := bytes.Clone(enc)
if len(challenge.ChallengeData) == 0 { if len(challenge.ChallengeData) == 0 {
t.Fatal("ChallengeData not assigned by encode") t.Fatal("ChallengeData not assigned by encode")
@ -290,8 +291,9 @@ func TestEncodeWhoareyouResend(t *testing.T) {
// A -> B WHOAREYOU // A -> B WHOAREYOU
// Send the same challenge again. This should produce exactly // Send the same challenge again. This should produce exactly
// the same bytes as the first send. // the same bytes as the first send.
whoareyou2, _ := net.nodeA.encode(t, net.nodeB, challenge) enc, _ = net.nodeA.encode(t, net.nodeB, challenge)
if !bytes.Equal(whoareyou2, whoareyou) { whoareyou2 := bytes.Clone(enc)
if !bytes.Equal(whoareyou2, whoareyou1) {
t.Fatal("re-encoded challenge not equal to first") t.Fatal("re-encoded challenge not equal to first")
} }
} }