requesting_test.go

   1 package torrent
   2
   3 import (
   4         "testing"
   5
   6         pp "github.com/anacrolix/torrent/peer_protocol"
   7         qt "github.com/frankban/quicktest"
   8 )
   9
  10 func keysAsSlice(m map[Request]struct{}) (sl []Request) {
  11         for k := range m {
  12                 sl = append(sl, k)
  13         }
  14         return
  15 }
  16
  17 func makeTypicalRequests() map[Request]struct{} {
  18         m := make(map[Request]struct{})
  19         for p := pp.Integer(0); p < 4; p++ {
  20                 for c := pp.Integer(0); c < 16; c++ {
  21                         m[Request{p, ChunkSpec{c * defaultChunkSize, defaultChunkSize}}] = struct{}{}
  22                 }
  23         }
  24         return m
  25 }
  26
  27 func TestLogExampleRequestMapOrdering(t *testing.T) {
  28         for k := range makeTypicalRequests() {
  29                 t.Log(k)
  30         }
  31
  32 }
  33
  34 func TestRequestMapOrderingPersistent(t *testing.T) {
  35         m := makeTypicalRequests()
  36         // Shows that map order is persistent across separate range statements.
  37         qt.Assert(t, keysAsSlice(m), qt.ContentEquals, keysAsSlice(m))
  38 }
  39
  40 func TestRequestMapOrderAcrossInstances(t *testing.T) {
  41         // This shows that different map instances with the same contents can have the same range order.
  42         qt.Assert(t, keysAsSlice(makeTypicalRequests()), qt.ContentEquals, keysAsSlice(makeTypicalRequests()))
  43 }