tracker/http/announce_private_test.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/models"
)

func TestPrivateAnnounce(t *testing.T) {
	cfg := config.DefaultConfig
	cfg.Private = true

	tkr, err := NewTracker(&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) 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)
}