104 lines
2.3 KiB
Go
104 lines
2.3 KiB
Go
// Copyright 2014 The Chihaya Authors. All rights reserved.
|
|
// Use of this source code is governed by the BSD 2-Clause license,
|
|
// which can be found in the LICENSE file.
|
|
|
|
package http
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/chihaya/bencode"
|
|
"github.com/chihaya/chihaya/config"
|
|
"github.com/chihaya/chihaya/tracker"
|
|
"github.com/chihaya/chihaya/tracker/models"
|
|
)
|
|
|
|
func TestPrivateAnnounce(t *testing.T) {
|
|
cfg := config.DefaultConfig
|
|
cfg.Private = true
|
|
|
|
tkr, err := tracker.New(&cfg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = loadTestData(tkr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
srv, err := createServer(tkr, &cfg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
defer srv.Close()
|
|
baseURL := srv.URL
|
|
|
|
peer := makePeerParams("-TR2820-peer1", false)
|
|
expected := makeResponse(0, 1, bencode.List{})
|
|
srv.URL = baseURL + "/users/vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv1"
|
|
checkAnnounce(peer, expected, srv, t)
|
|
|
|
peer = makePeerParams("-TR2820-peer2", false)
|
|
expected = makeResponse(0, 2, bencode.List{
|
|
makePeerResponse("-TR2820-peer1"),
|
|
})
|
|
srv.URL = baseURL + "/users/vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv2"
|
|
checkAnnounce(peer, expected, srv, t)
|
|
|
|
peer = makePeerParams("-TR2820-peer3", true)
|
|
expected = makeResponse(1, 2, bencode.List{
|
|
makePeerResponse("-TR2820-peer1"),
|
|
makePeerResponse("-TR2820-peer2"),
|
|
})
|
|
srv.URL = baseURL + "/users/vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv3"
|
|
checkAnnounce(peer, expected, srv, t)
|
|
|
|
peer = makePeerParams("-TR2820-peer1", false)
|
|
expected = makeResponse(1, 2, bencode.List{
|
|
makePeerResponse("-TR2820-peer2"),
|
|
makePeerResponse("-TR2820-peer3"),
|
|
})
|
|
srv.URL = baseURL + "/users/vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv1"
|
|
checkAnnounce(peer, expected, srv, t)
|
|
}
|
|
|
|
func loadTestData(tkr *tracker.Tracker) error {
|
|
conn, err := tkr.Pool.Get()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
users := []string{
|
|
"vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv1",
|
|
"vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv2",
|
|
"vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv3",
|
|
}
|
|
|
|
for i, passkey := range users {
|
|
err = conn.PutUser(&models.User{
|
|
ID: uint64(i + 1),
|
|
Passkey: passkey,
|
|
})
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
err = conn.PutClient("TR2820")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
torrent := &models.Torrent{
|
|
ID: 1,
|
|
Infohash: infoHash,
|
|
Seeders: make(map[string]models.Peer),
|
|
Leechers: make(map[string]models.Peer),
|
|
}
|
|
|
|
return conn.PutTorrent(torrent)
|
|
}
|