45 lines
847 B
Go
45 lines
847 B
Go
package bencode_test
|
|
|
|
import (
|
|
"net"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/anacrolix/dht/v2/krpc"
|
|
"github.com/anacrolix/torrent/bencode"
|
|
)
|
|
|
|
func marshalAndUnmarshal(tb testing.TB, orig krpc.Msg) (ret krpc.Msg) {
|
|
b, err := bencode.Marshal(orig)
|
|
if err != nil {
|
|
tb.Fatal(err)
|
|
}
|
|
err = bencode.Unmarshal(b, &ret)
|
|
if err != nil {
|
|
tb.Fatal(err)
|
|
}
|
|
// ret.Q = "what"
|
|
return
|
|
}
|
|
|
|
func BenchmarkMarshalThenUnmarshalKrpcMsg(tb *testing.B) {
|
|
orig := krpc.Msg{
|
|
T: "420",
|
|
Y: "r",
|
|
R: &krpc.Return{
|
|
Token: func() *string { t := "re-up"; return &t }(),
|
|
},
|
|
IP: krpc.NodeAddr{IP: net.ParseIP("1.2.3.4"), Port: 1337},
|
|
ReadOnly: true,
|
|
}
|
|
first := marshalAndUnmarshal(tb, orig)
|
|
if !reflect.DeepEqual(orig, first) {
|
|
tb.Fail()
|
|
}
|
|
tb.ReportAllocs()
|
|
tb.ResetTimer()
|
|
for i := 0; i < tb.N; i += 1 {
|
|
marshalAndUnmarshal(tb, orig)
|
|
}
|
|
}
|