tracker/http_test.go

   1 package tracker
   2
   3 import (
   4         "testing"
   5
   6         "github.com/stretchr/testify/assert"
   7         "github.com/stretchr/testify/require"
   8
   9         "github.com/anacrolix/torrent/bencode"
  10 )
  11
  12 func TestUnmarshalHTTPResponsePeerDicts(t *testing.T) {
  13         var hr HttpResponse
  14         require.NoError(t, bencode.Unmarshal(
  15                 []byte("d5:peersl"+
  16                         "d2:ip7:1.2.3.47:peer id20:thisisthe20bytepeeri4:porti9999ee"+
  17                         "d7:peer id20:thisisthe20bytepeeri2:ip39:2001:0db8:85a3:0000:0000:8a2e:0370:73344:porti9998ee"+
  18                         "e"+
  19                         "6:peers618:123412341234123456"+
  20                         "e"),
  21                 &hr))
  22
  23         require.Len(t, hr.Peers, 2)
  24         assert.Equal(t, []byte("thisisthe20bytepeeri"), hr.Peers[0].ID)
  25         assert.EqualValues(t, 9999, hr.Peers[0].Port)
  26         assert.EqualValues(t, 9998, hr.Peers[1].Port)
  27         assert.NotNil(t, hr.Peers[0].IP)
  28         assert.NotNil(t, hr.Peers[1].IP)
  29
  30         assert.Len(t, hr.Peers6, 1)
  31         assert.EqualValues(t, "1234123412341234", hr.Peers6[0].IP)
  32         assert.EqualValues(t, 0x3536, hr.Peers6[0].Port)
  33 }
  34
  35 func TestUnmarshalHttpResponseNoPeers(t *testing.T) {
  36         var hr HttpResponse
  37         require.NoError(t, bencode.Unmarshal(
  38                 []byte("d6:peers618:123412341234123456e"),
  39                 &hr,
  40         ))
  41         require.Len(t, hr.Peers, 0)
  42         assert.Len(t, hr.Peers6, 1)
  43 }
  44
  45 func TestUnmarshalHttpResponsePeers6NotCompact(t *testing.T) {
  46         var hr HttpResponse
  47         require.Error(t, bencode.Unmarshal(
  48                 []byte("d6:peers6lee"),
  49                 &hr,
  50         ))
  51 }