tracker/http/announce_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 (
	"io/ioutil"
	"net/http"
	"net/http/httptest"
	"net/url"
	"reflect"
	"testing"

	"github.com/chihaya/bencode"
	"github.com/chihaya/chihaya/config"

	_ "github.com/chihaya/chihaya/drivers/backend/noop"
	_ "github.com/chihaya/chihaya/drivers/tracker/memory"
)

type params map[string]string

const infoHash = "%89%d4%bcR%11%16%ca%1dB%a2%f3%0d%1f%27M%94%e4h%1d%af"

func TestPublicAnnounce(t *testing.T) {
	srv, _ := setupTracker(&config.DefaultConfig)
	defer srv.Close()

	// Add one seeder.
	peer := basePeer("peer1", true)
	expected := baseResponse(1, 0, bencode.List{})
	checkResponse(peer, expected, srv, t)

	// Add another seeder.
	peer = basePeer("peer2", true)
	expected = baseResponse(2, 0, bencode.List{})
	checkResponse(peer, expected, srv, t)

	// Add a leecher.
	peer = basePeer("peer3", false)
	expected = baseResponse(2, 1, bencode.List{
		bencode.Dict{
			"ip":      "127.0.0.1",
			"peer id": "peer1",
			"port":    int64(1234),
		},
		bencode.Dict{
			"ip":      "127.0.0.1",
			"peer id": "peer2",
			"port":    int64(1234),
		},
	})

	checkResponse(peer, expected, srv, t)

	// Remove seeder.
	peer = basePeer("peer1", true)
	peer["event"] = "stopped"
	expected = baseResponse(1, 1, nil)
	checkResponse(peer, expected, srv, t)

	// Check seeders.
	peer = basePeer("peer3", false)
	expected = baseResponse(1, 1, bencode.List{
		bencode.Dict{
			"ip":      "127.0.0.1",
			"peer id": "peer2",
			"port":    int64(1234),
		},
	})

	checkResponse(peer, expected, srv, t)
}

func basePeer(id string, seed bool) params {
	left := "1"
	if seed {
		left = "0"
	}

	return params{
		"info_hash":  infoHash,
		"peer_id":    id,
		"port":       "1234",
		"uploaded":   "0",
		"downloaded": "0",
		"left":       left,
		"compact":    "0",
		"numwant":    "50",
	}
}

func baseResponse(seeders, leechers int64, peers bencode.List) bencode.Dict {
	dict := bencode.Dict{
		"complete":     seeders,
		"incomplete":   leechers,
		"interval":     int64(1800),
		"min interval": int64(900),
	}

	if peers != nil {
		dict["peers"] = peers
	}
	return dict
}

func checkResponse(p params, expected interface{}, srv *httptest.Server, t *testing.T) bool {
	values := &url.Values{}
	for k, v := range p {
		values.Add(k, v)
	}

	response, err := http.Get(srv.URL + "/announce?" + values.Encode())
	if err != nil {
		t.Error(err)
		return false
	}

	body, err := ioutil.ReadAll(response.Body)
	response.Body.Close()

	if err != nil {
		t.Error(err)
		return false
	}

	got, err := bencode.Unmarshal(body)
	if !reflect.DeepEqual(got, expected) {
		t.Errorf("\ngot:    %#v\nwanted: %#v", got, expected)
		return false
	}
	return true
}

func setupTracker(cfg *config.Config) (*httptest.Server, error) {
	tkr, err := NewTracker(cfg)
	if err != nil {
		return nil, err
	}

	return httptest.NewServer(setupRoutes(tkr, cfg)), nil
}