mirror of
https://github.com/hanzozt/zt.git
synced 2026-10-30 15:04:55 +00:00
zeekay
241625e7d9
- Repo names: ziti-* → zt-* - Go modules: github.com/hanzozt/ziti → github.com/hanzozt/zt - Docker images: hanzozt/ziti-* → hanzozt/zt-* - C/Java/JS symbols: ziti_* → zt_* - File and directory names updated
102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package smoke
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/hanzozt/fablab/kernel/model"
|
|
)
|
|
|
|
var hashes = map[string]string{
|
|
"1KB": "0f343b0931126a20f133d67c2b018a3b",
|
|
"100KB": "4c6426ac7ef186464ecbb0d81cbfcb1e",
|
|
"20MB": "8f4e33f3dc3e414ff94e5fb6905cba8c",
|
|
}
|
|
|
|
var timeouts = map[string]time.Duration{
|
|
"1KB": 10 * time.Second,
|
|
"100KB": 10 * time.Second,
|
|
"20MB": 40 * time.Second,
|
|
}
|
|
|
|
type HttpClient string
|
|
|
|
const (
|
|
ClientCurl HttpClient = "curl"
|
|
ClientWget HttpClient = "wget"
|
|
)
|
|
|
|
var FileSizes = []string{"1KB", "100KB", "20MB"}
|
|
var HttpClients = []HttpClient{ClientCurl, ClientWget}
|
|
|
|
func TestFileDownload(hostSelector string, client HttpClient, hostType string, encrypted bool, fileSize string) (string, error) {
|
|
host := model.GetModel().MustSelectHost("." + hostSelector + "-client")
|
|
|
|
urlExtra := ""
|
|
if !encrypted {
|
|
urlExtra = "-unencrypted"
|
|
}
|
|
|
|
url := fmt.Sprintf("http://files-%s%s.zt/%s.zip", hostType, urlExtra, fileSize)
|
|
|
|
filename := uuid.NewString() + ".tmp"
|
|
|
|
var cmds []string
|
|
cmds = append(cmds, fmt.Sprintf("echo '%s %s' > checksums", hashes[fileSize], filename))
|
|
|
|
var cmd string
|
|
if client == ClientCurl {
|
|
cmd = fmt.Sprintf(`set -o pipefail; rm -f *.tmp; curl -k --fail-early --fail-with-body -SL -o %s %s 2>&1`, filename, url)
|
|
} else if client == ClientWget {
|
|
cmd = fmt.Sprintf(`set -o pipefail; rm -f *.tmp; wget --no-check-certificate -O %s -t 5 -T 5 %s 2>&1`, filename, url)
|
|
}
|
|
cmds = append(cmds, cmd)
|
|
cmds = append(cmds, "md5sum -c checksums")
|
|
|
|
timeout := timeouts[fileSize]
|
|
return host.ExecLoggedWithTimeout(timeout, cmds...)
|
|
}
|
|
|
|
func TestIperf(clientHostSelector, hostType string, encrypted, reversed bool, run model.Run) (string, error) {
|
|
c, err := model.GetModel().SelectComponent(".iperf." + hostType)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if err = c.Type.Stop(run, c); err != nil {
|
|
return "", err
|
|
}
|
|
iperfServer := c.Type.(model.ServerComponent)
|
|
if err = iperfServer.Start(run, c); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
host, err := model.GetModel().SelectHost("." + clientHostSelector + "-client")
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
urlExtra := ""
|
|
if !encrypted {
|
|
urlExtra = "-unencrypted"
|
|
}
|
|
|
|
addr := fmt.Sprintf("iperf-%s%s.zt", hostType, urlExtra)
|
|
|
|
extraOptions := ""
|
|
if reversed {
|
|
extraOptions += " -R"
|
|
}
|
|
|
|
logFile := fmt.Sprintf("logs/iperf-%s%s%s.log", hostType, urlExtra, func() string {
|
|
if reversed {
|
|
return "-r"
|
|
}
|
|
return ""
|
|
}())
|
|
|
|
cmd := fmt.Sprintf(`set -o pipefail; iperf3 -c %s -P 5 -t 10 %s > %s 2>&1 `, addr, extraOptions, logFile)
|
|
|
|
return host.ExecLoggedWithTimeout(40*time.Second, cmd)
|
|
}
|