Implemented Commandline Download Speedtest (#2064)
Added the net/speedtest package that contains code for starting up a speedtest server and a client. The speedtest command for starting a client takes in a duration for the speedtest as well as the host and port of the speedtest server to connect to. The speedtest command for starting a server takes in a host:port pair to listen on. Signed-off-by: Aaditya Chaudhary <32117362+AadityaChaudhary@users.noreply.github.com>
This commit is contained in:

committed by
GitHub

parent
3ebe16558c
commit
4f89fe17a2
81
net/speedtest/speedtest_test.go
Normal file
81
net/speedtest/speedtest_test.go
Normal file
@ -0,0 +1,81 @@
|
||||
// Copyright (c) 2021 Tailscale Inc & AUTHORS All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package speedtest
|
||||
|
||||
import (
|
||||
"net"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestDownload(t *testing.T) {
|
||||
// start a listener and find the port where the server will be listening.
|
||||
l, err := net.Listen("tcp", ":0")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
t.Cleanup(func() { l.Close() })
|
||||
|
||||
serverIP := l.Addr().String()
|
||||
t.Log("server IP found:", serverIP)
|
||||
|
||||
type state struct {
|
||||
err error
|
||||
}
|
||||
displayResult := func(t *testing.T, r Result) {
|
||||
t.Helper()
|
||||
t.Logf("{ Megabytes: %.2f, Start: %.1f, End: %.1f, Total: %t }", r.MegaBytes(), r.IntervalStart.Seconds(), r.IntervalEnd.Seconds(), r.Total)
|
||||
}
|
||||
stateChan := make(chan state, 1)
|
||||
|
||||
go func() {
|
||||
err := Serve(l)
|
||||
stateChan <- state{err: err}
|
||||
}()
|
||||
|
||||
// ensure that the test returns an appropriate number of Result structs
|
||||
expectedLen := int(DefaultDuration.Seconds()) + 1
|
||||
|
||||
t.Run("download test", func(t *testing.T) {
|
||||
// conduct a download test
|
||||
results, err := RunClient(Download, DefaultDuration, serverIP)
|
||||
|
||||
if err != nil {
|
||||
t.Fatal("download test failed:", err)
|
||||
}
|
||||
|
||||
if len(results) < expectedLen {
|
||||
t.Fatalf("download results: expected length: %d, actual length: %d", expectedLen, len(results))
|
||||
}
|
||||
|
||||
for _, result := range results {
|
||||
displayResult(t, result)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("upload test", func(t *testing.T) {
|
||||
// conduct an upload test
|
||||
results, err := RunClient(Upload, DefaultDuration, serverIP)
|
||||
|
||||
if err != nil {
|
||||
t.Fatal("upload test failed:", err)
|
||||
}
|
||||
|
||||
if len(results) < expectedLen {
|
||||
t.Fatalf("upload results: expected length: %d, actual length: %d", expectedLen, len(results))
|
||||
}
|
||||
|
||||
for _, result := range results {
|
||||
displayResult(t, result)
|
||||
}
|
||||
})
|
||||
|
||||
// causes the server goroutine to finish
|
||||
l.Close()
|
||||
|
||||
testState := <-stateChan
|
||||
if testState.err != nil {
|
||||
t.Error("server error:", err)
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user