From 89f024a76bbebe03fd4b871ab5b5ed0d567b0652 Mon Sep 17 00:00:00 2001 From: Matt Joiner Date: Tue, 18 Aug 2015 02:08:08 +1000 Subject: [PATCH] dht: Revive the UnmarshalGetPeersResponse test --- dht/dht_test.go | 30 +++++++++++------------------- 1 file changed, 11 insertions(+), 19 deletions(-) diff --git a/dht/dht_test.go b/dht/dht_test.go index e1583460..6636ad6f 100644 --- a/dht/dht_test.go +++ b/dht/dht_test.go @@ -8,6 +8,7 @@ import ( "testing" "github.com/anacrolix/missinggo" + "github.com/stretchr/testify/assert" ) func TestSetNilBigInt(t *testing.T) { @@ -104,25 +105,16 @@ func TestClosestNodes(t *testing.T) { } } -// func TestUnmarshalGetPeersResponse(t *testing.T) { -// var gpr getPeersResponse -// if err != nil { -// t.Fatal(err) -// } -// err = dec.Decode(map[string]interface{}{ -// "values": []string{"\x01\x02\x03\x04\x05\x06", "\x07\x08\x09\x0a\x0b\x0c"}, -// "nodes": "\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d", -// }) -// if err != nil { -// t.Fatal(err) -// } -// if len(gpr.Values) != 2 { -// t.FailNow() -// } -// if len(gpr.Nodes) != 2 { -// t.FailNow() -// } -// } +func TestUnmarshalGetPeersResponse(t *testing.T) { + gpr := Msg{ + "r": map[string]interface{}{ + "values": []interface{}{"\x01\x02\x03\x04\x05\x06", "\x07\x08\x09\x0a\x0b\x0c"}, + "nodes": "\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x02\x03\x04\x05\x06\x07\x08\x09\x02\x03\x04\x05\x06\x07" + "\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x02\x03\x04\x05\x06\x07\x08\x09\x02\x03\x04\x05\x06\x07", + }, + } + assert.EqualValues(t, 2, len(gpr.Values())) + assert.EqualValues(t, 2, len(gpr.Nodes())) +} func TestDHTDefaultConfig(t *testing.T) { s, err := NewServer(nil) -- 2.48.1