From 2fcd34360fb3a520f7545e205be4608ca2dec894 Mon Sep 17 00:00:00 2001
From: Jimmy Zelinskie <jimmyzelinskie@gmail.com>
Date: Mon, 11 May 2015 00:57:54 -0400
Subject: [PATCH] udp: basic announce test

---
 udp/announce_test.go | 87 ++++++++++++++++++++++++++++++++++++++++++++
 udp/scrape_test.go   | 29 +++++++++------
 2 files changed, 104 insertions(+), 12 deletions(-)
 create mode 100644 udp/announce_test.go

diff --git a/udp/announce_test.go b/udp/announce_test.go
new file mode 100644
index 0000000..cff1939
--- /dev/null
+++ b/udp/announce_test.go
@@ -0,0 +1,87 @@
+// Copyright 2015 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 udp
+
+import (
+	"bytes"
+	"encoding/binary"
+	"net"
+	"testing"
+
+	"github.com/chihaya/chihaya/config"
+)
+
+func requestAnnounce(sock *net.UDPConn, connID []byte, hash string) ([]byte, error) {
+	txID := makeTransactionID()
+	peerID := []byte("-UT2210-b4a2h9a9f5c4")
+
+	var request []byte
+	request = append(request, connID...)
+	request = append(request, announceAction...)
+	request = append(request, txID...)
+	request = append(request, []byte(hash)...)
+	request = append(request, peerID...)
+	request = append(request, make([]byte, 8)...) // Downloaded
+	request = append(request, make([]byte, 8)...) // Left
+	request = append(request, make([]byte, 8)...) // Uploaded
+	request = append(request, make([]byte, 4)...) // Event
+	request = append(request, make([]byte, 4)...) // IP
+	request = append(request, make([]byte, 4)...) // Key
+	request = append(request, make([]byte, 4)...) // NumWant
+	request = append(request, make([]byte, 2)...) // Port
+
+	return doRequest(sock, request, txID)
+}
+
+func TestAnnounce(t *testing.T) {
+	srv, done, err := setupTracker(&config.DefaultConfig)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	_, sock, err := setupSocket()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	connID, err := requestConnectionID(sock)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	announce, err := requestAnnounce(sock, connID, "aaaaaaaaaaaaaaaaaaaa")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// Parse the response.
+	var action, txID, interval, leechers, seeders uint32
+	buf := bytes.NewReader(announce)
+	binary.Read(buf, binary.BigEndian, &action)
+	binary.Read(buf, binary.BigEndian, &txID)
+	binary.Read(buf, binary.BigEndian, &interval)
+	binary.Read(buf, binary.BigEndian, &leechers)
+	binary.Read(buf, binary.BigEndian, &seeders)
+
+	if action != uint32(announceActionID) {
+		t.Fatal("expected announce action")
+	}
+
+	if interval != uint32(config.DefaultConfig.Announce.Seconds()) {
+		t.Fatal("incorrect interval")
+	}
+
+	if leechers != uint32(0) {
+		t.Fatal("incorrect leecher count")
+	}
+
+	// We're the only seeder.
+	if seeders != uint32(1) {
+		t.Fatal("incorrect seeder count")
+	}
+
+	srv.Stop()
+	<-done
+}
diff --git a/udp/scrape_test.go b/udp/scrape_test.go
index 79482e0..eca4fd0 100644
--- a/udp/scrape_test.go
+++ b/udp/scrape_test.go
@@ -13,19 +13,9 @@ import (
 	"github.com/chihaya/chihaya/config"
 )
 
-func requestScrape(sock *net.UDPConn, connID []byte, hashes []string) ([]byte, error) {
-	txID := makeTransactionID()
-	request := []byte{}
-
-	request = append(request, connID...)
-	request = append(request, scrapeAction...)
-	request = append(request, txID...)
-
-	for _, hash := range hashes {
-		request = append(request, []byte(hash)...)
-	}
-
+func doRequest(sock *net.UDPConn, request, txID []byte) ([]byte, error) {
 	response := make([]byte, 1024)
+
 	n, err := sendRequest(sock, request, response)
 	if err != nil {
 		return nil, err
@@ -38,6 +28,21 @@ func requestScrape(sock *net.UDPConn, connID []byte, hashes []string) ([]byte, e
 	return response[:n], nil
 }
 
+func requestScrape(sock *net.UDPConn, connID []byte, hashes []string) ([]byte, error) {
+	txID := makeTransactionID()
+
+	var request []byte
+	request = append(request, connID...)
+	request = append(request, scrapeAction...)
+	request = append(request, txID...)
+
+	for _, hash := range hashes {
+		request = append(request, []byte(hash)...)
+	}
+
+	return doRequest(sock, request, txID)
+}
+
 func TestScrapeEmpty(t *testing.T) {
 	srv, done, err := setupTracker(&config.DefaultConfig)
 	if err != nil {