// Copyright 2013 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 server

import (
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/chihaya/chihaya/drivers/backend"
	"github.com/chihaya/chihaya/drivers/tracker"
	"github.com/chihaya/chihaya/models"

	_ "github.com/chihaya/chihaya/drivers/backend/mock"
	_ "github.com/chihaya/chihaya/drivers/tracker/mock"
)

func TestAnnounce(t *testing.T) {
	s, err := newTestServer()
	if err != nil {
		t.Error(err)
	}

	err = s.Prime(func(t tracker.Pool, b backend.Conn) (err error) {
		conn, err := t.Get()
		if err != nil {
			return
		}

		err = conn.AddUser(&models.User{
			ID:      1,
			Passkey: "yby47f04riwpndba456rqxtmifenq5h6",
		})
		if err != nil {
			return
		}

		err = conn.WhitelistClient("TR2820")
		if err != nil {
			return
		}

		err = conn.AddTorrent(&models.Torrent{
			ID:       1,
			Infohash: string([]byte{0x89, 0xd4, 0xbc, 0x52, 0x11, 0x16, 0xca, 0x1d, 0x42, 0xa2, 0xf3, 0x0d, 0x1f, 0x27, 0x4d, 0x94, 0xe4, 0x68, 0x1d, 0xaf}),
			Seeders:  make(map[string]models.Peer),
			Leechers: make(map[string]models.Peer),
		})

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

	url := "http://localhost:34000/yby47f04riwpndba456rqxtmifenq5h6/announce?info_hash=%89%d4%bcR%11%16%ca%1dB%a2%f3%0d%1f%27M%94%e4h%1d%af&peer_id=-TR2820-l71jtqkl898b&port=51413&uploaded=0&downloaded=0&left=0&numwant=0&key=3c8e3319&compact=1&supportcrypto=1&event=stopped"
	r, err := http.NewRequest("GET", url, nil)
	if err != nil {
		t.Error(err)
	}

	w := httptest.NewRecorder()
	s.serveAnnounce(w, r)

	if w.Body.String() != "1:d8:completei0e10:incompletei0e8:intervali1800e12:min intervali900e1:e" {
		t.Errorf("improper response from server:\n%s", w.Body.String())
	}

}