metainfo: Ensure that nodes encode correctly

diff --git a/metainfo/nodes_test.go b/metainfo/nodes_test.go
index 9a76e1377dc687a2a4c867166eb5860cc500e471..585120b76f23f9405fa13b071524c738252497da 100644 (file)
--- a/metainfo/nodes_test.go
+++ b/metainfo/nodes_test.go
@@ -5,6 +5,8 @@ import (
 
        "github.com/stretchr/testify/assert"
        "github.com/stretchr/testify/require"
+
+       "github.com/anacrolix/torrent/bencode"
 )
 
 func testFileNodesMatch(t *testing.T, file string, nodes []Node) {
@@ -38,3 +40,16 @@ func TestNodesListPairsBEP5(t *testing.T) {
                "83.128.223.71:23865",
        })
 }
+
+func testMarshalMetainfo(t *testing.T, expected string, mi MetaInfo) {
+       b, err := bencode.Marshal(mi)
+       assert.NoError(t, err)
+       assert.EqualValues(t, expected, string(b))
+}
+
+func TestMarshalMetainfoNodes(t *testing.T) {
+       testMarshalMetainfo(t, "d4:infod4:name0:12:piece lengthi0e6:piecesleee", MetaInfo{})
+       testMarshalMetainfo(t, "d4:infod4:name0:12:piece lengthi0e6:pieceslee5:nodesl12:1.2.3.4:555514:not a hostportee", MetaInfo{
+               Nodes: []Node{"1.2.3.4:5555", "not a hostport"},
+       })
+}