Mirror/mochi
1
0
Fork 0
mirror of https://github.com/sot-tech/mochi.git synced 2026-05-10 14:19:09 -07:00
Code Issues Packages Projects Releases Wiki Activity

udp: basic announce test

Browse Source
This commit is contained in:
Jimmy Zelinskie
2015-05-11 00:57:54 -04:00
parent de18172952
commit 2fcd34360f
2 changed files with 104 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
udp/scrape_test.go
View File

@@ -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 {