diff --git a/elkrem/elkrem_test.go b/elkrem/elkrem_test.go index e049c2cd..a2af11a8 100644 --- a/elkrem/elkrem_test.go +++ b/elkrem/elkrem_test.go @@ -10,7 +10,7 @@ import ( func TestElkremBig(t *testing.T) { sndr := NewElkremSender(63, wire.DoubleSha256SH([]byte("elktest"))) rcv := NewElkremReceiver(63) - SenderSerdesTest(t, sndr) + SenderSerdesTest(t, &sndr) for n := uint64(0); n < 10000; n++ { sha, err := sndr.Next() if err != nil { @@ -28,8 +28,8 @@ func TestElkremBig(t *testing.T) { } } } - SenderSerdesTest(t, sndr) - ReceiverSerdesTest(t, rcv) + SenderSerdesTest(t, &sndr) + ReceiverSerdesTest(t, &rcv) for n := uint64(0); n < 10000; n += 500 { sha, err := rcv.AtIndex(n) if err != nil { diff --git a/elkrem/serdes_test.go b/elkrem/serdes_test.go index 8dab5e33..3027ce59 100644 --- a/elkrem/serdes_test.go +++ b/elkrem/serdes_test.go @@ -5,19 +5,19 @@ import ( "testing" ) -func ReceiverSerdesTest(t *testing.T, er ElkremReceiver) { - b, err := er.ToBytes() +func ReceiverSerdesTest(t *testing.T, rcv *ElkremReceiver) { + b, err := rcv.ToBytes() if err != nil { t.Fatal(err) } t.Logf("Serialized receiver; %d bytes, hex:\n%x\n", len(b), b) - rcv2, err := ElkremReceiverFromBytes(b) + *rcv, err = ElkremReceiverFromBytes(b) if err != nil { t.Fatal(err) } - b2, err := rcv2.ToBytes() + b2, err := rcv.ToBytes() if err != nil { t.Fatal(err) } @@ -27,19 +27,19 @@ func ReceiverSerdesTest(t *testing.T, er ElkremReceiver) { } } -func SenderSerdesTest(t *testing.T, es ElkremSender) { - b, err := es.ToBytes() +func SenderSerdesTest(t *testing.T, sndr *ElkremSender) { + b, err := sndr.ToBytes() if err != nil { t.Fatal(err) } t.Logf("Serialized sender; %d bytes, hex:\n%x\n", len(b), b) - sndr2, err := ElkremSenderFromBytes(b) + *sndr, err = ElkremSenderFromBytes(b) if err != nil { t.Fatal(err) } - b2, err := sndr2.ToBytes() + b2, err := sndr.ToBytes() if err != nil { t.Fatal(err) }