package torrent import ( "sync" "testing" "time" "github.com/anacrolix/torrent/peer_protocol" ) func r(i, b, l peer_protocol.Integer) request { return request{i, chunkSpec{b, l}} } // Check the given Request is correct for various torrent offsets. func TestTorrentRequest(t *testing.T) { const s = 472183431 // Length of torrent. for _, _case := range []struct { off int64 // An offset into the torrent. req request // The expected Request. The zero value means !ok. }{ // Invalid offset. {-1, request{}}, {0, r(0, 0, 16384)}, // One before the end of a piece. {1<<18 - 1, r(0, 1<<18-16384, 16384)}, // Offset beyond torrent length. {472 * 1 << 20, request{}}, // One before the end of the torrent. Complicates the chunk length. {s - 1, r((s-1)/(1<<18), (s-1)%(1<<18)/(16384)*(16384), 12935)}, {1, r(0, 0, 16384)}, // One before end of chunk. {16383, r(0, 0, 16384)}, // Second chunk. {16384, r(0, 16384, 16384)}, } { req, ok := torrentOffsetRequest(472183431, 1<<18, 16384, _case.off) if (_case.req == request{}) == ok { t.Fatalf("expected %v, got %v", _case.req, req) } if req != _case.req { t.Fatalf("expected %v, got %v", _case.req, req) } } } func TestTorrentDoubleClose(t *testing.T) { tt, err := newTorrent(InfoHash{}) tt.pruneTimer = time.NewTimer(0) if err != nil { t.Fatal(err) } wg := sync.WaitGroup{} for i := 0; i < 2; i++ { wg.Add(1) go func() { tt.close() wg.Done() }() } wg.Wait() } func TestAppendToCopySlice(t *testing.T) { orig := []int{1, 2, 3} dupe := append([]int{}, orig...) dupe[0] = 4 if orig[0] != 1 { t.FailNow() } } func TestTorrentString(t *testing.T) { tor := &torrent{} s := tor.InfoHash.HexString() if s != "0000000000000000000000000000000000000000" { t.FailNow() } }