
- replaced s.GetContainerByName("xyz") with s.Containers.Xyz in tests and suites - same thing for interfaces - each suite has its own structs with containers/interfaces - structs are initialized in SetupSuite Type: test Change-Id: I5bd99605b40921b7b8c844e8650f6fb0915e9e99 Signed-off-by: Adrian Villin <avillin@cisco.com>
407 lines
10 KiB
Go
407 lines
10 KiB
Go
package hst
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"net"
|
|
"net/http"
|
|
"net/http/httputil"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
const networkTopologyDir string = "topo-network/"
|
|
const containerTopologyDir string = "topo-containers/"
|
|
|
|
type Stanza struct {
|
|
content string
|
|
pad int
|
|
}
|
|
|
|
type ActionResult struct {
|
|
Err error
|
|
Desc string
|
|
ErrOutput string
|
|
StdOutput string
|
|
}
|
|
|
|
type JsonResult struct {
|
|
Code int
|
|
Desc string
|
|
ErrOutput string
|
|
StdOutput string
|
|
}
|
|
|
|
type IPerfResult struct {
|
|
Start struct {
|
|
Timestamp struct {
|
|
Time string `json:"time"`
|
|
} `json:"timestamp"`
|
|
Connected []struct {
|
|
Socket int `json:"socket"`
|
|
LocalHost string `json:"local_host"`
|
|
LocalPort int `json:"local_port"`
|
|
RemoteHost string `json:"remote_host"`
|
|
RemotePort int `json:"remote_port"`
|
|
} `json:"connected"`
|
|
Version string `json:"version"`
|
|
Details struct {
|
|
Protocol string `json:"protocol"`
|
|
} `json:"test_start"`
|
|
} `json:"start"`
|
|
End struct {
|
|
TcpSent *struct {
|
|
MbitsPerSecond float64 `json:"bits_per_second"`
|
|
MBytes float64 `json:"bytes"`
|
|
} `json:"sum_sent,omitempty"`
|
|
TcpReceived *struct {
|
|
MbitsPerSecond float64 `json:"bits_per_second"`
|
|
MBytes float64 `json:"bytes"`
|
|
} `json:"sum_received,omitempty"`
|
|
Udp *struct {
|
|
MbitsPerSecond float64 `json:"bits_per_second"`
|
|
JitterMs float64 `json:"jitter_ms,omitempty"`
|
|
LostPackets int `json:"lost_packets,omitempty"`
|
|
Packets int `json:"packets,omitempty"`
|
|
LostPercent float64 `json:"lost_percent,omitempty"`
|
|
MBytes float64 `json:"bytes"`
|
|
} `json:"sum,omitempty"`
|
|
} `json:"end"`
|
|
}
|
|
|
|
func AssertFileSize(f1, f2 string) error {
|
|
fi1, err := os.Stat(f1)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fi2, err1 := os.Stat(f2)
|
|
if err1 != nil {
|
|
return err1
|
|
}
|
|
|
|
if fi1.Size() != fi2.Size() {
|
|
return fmt.Errorf("file sizes differ (%d vs %d)", fi1.Size(), fi2.Size())
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *Stanza) NewStanza(name string) *Stanza {
|
|
c.Append("\n" + name + " {")
|
|
c.pad += 2
|
|
return c
|
|
}
|
|
|
|
func (c *Stanza) Append(name string) *Stanza {
|
|
c.content += strings.Repeat(" ", c.pad)
|
|
c.content += name + "\n"
|
|
return c
|
|
}
|
|
|
|
func (c *Stanza) Close() *Stanza {
|
|
c.content += "}\n"
|
|
c.pad -= 2
|
|
return c
|
|
}
|
|
|
|
func (s *Stanza) ToString() string {
|
|
return s.content
|
|
}
|
|
|
|
func (s *Stanza) SaveToFile(fileName string) error {
|
|
fo, err := os.Create(fileName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer fo.Close()
|
|
|
|
_, err = io.Copy(fo, strings.NewReader(s.content))
|
|
return err
|
|
}
|
|
|
|
// NewHttpClient creates [http.Client] with disabled proxy and redirects, it also sets timeout to 30seconds.
|
|
func NewHttpClient(timeout time.Duration) *http.Client {
|
|
transport := http.DefaultTransport
|
|
transport.(*http.Transport).Proxy = nil
|
|
transport.(*http.Transport).DisableKeepAlives = true
|
|
transport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
|
|
client := &http.Client{
|
|
Transport: transport,
|
|
Timeout: timeout,
|
|
CheckRedirect: func(req *http.Request, via []*http.Request) error {
|
|
return http.ErrUseLastResponse
|
|
}}
|
|
return client
|
|
}
|
|
|
|
func DumpHttpResp(resp *http.Response, body bool) string {
|
|
dump, err := httputil.DumpResponse(resp, body)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
return string(dump)
|
|
}
|
|
|
|
func TcpSendReceive(address, data string) (string, error) {
|
|
conn, err := net.DialTimeout("tcp", address, time.Second*30)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer conn.Close()
|
|
err = conn.SetDeadline(time.Now().Add(time.Second * 30))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
_, err = conn.Write([]byte(data))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
reply := make([]byte, 1024)
|
|
_, err = conn.Read(reply)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(reply), nil
|
|
}
|
|
|
|
/*
|
|
RunCurlContainer execute curl command with given args.
|
|
Container with name "curl" must be available.
|
|
Curl runs in verbose mode and progress meter switch off by default.
|
|
*/
|
|
func (s *HstSuite) RunCurlContainer(curlCont *Container, args string) (string, string) {
|
|
cmd := fmt.Sprintf("curl -v -s %s", args)
|
|
s.Log(cmd)
|
|
curlCont.ExtraRunningArgs = cmd
|
|
curlCont.Run()
|
|
stdout, stderr := curlCont.GetOutput()
|
|
s.Log(stderr)
|
|
s.Log(stdout)
|
|
return stdout, stderr
|
|
}
|
|
|
|
/*
|
|
CollectNginxLogs save access and error logs to the test execution directory.
|
|
Nginx logging need to be set following way:
|
|
|
|
- error_log <default-work-dir>/{{.LogPrefix}}-error.log;
|
|
- access_log <default-work-dir>/{{.LogPrefix}}-access.log;
|
|
|
|
where LogPrefix is set to nginxContainer.Name
|
|
*/
|
|
func (s *HstSuite) CollectNginxLogs(nginxContainer *Container) {
|
|
targetDir := nginxContainer.Suite.getLogDirPath()
|
|
source := nginxContainer.GetHostWorkDir() + "/" + nginxContainer.Name + "-"
|
|
cmd := exec.Command("cp", "-t", targetDir, source+"error.log", source+"access.log")
|
|
s.Log(cmd.String())
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
s.Log(fmt.Sprint(err))
|
|
}
|
|
}
|
|
|
|
/*
|
|
CollectEnvoyLogs save access logs to the test execution directory.
|
|
Envoy access log path need to be set following way:
|
|
<default-work-dir>/{{.LogPrefix}}-access.log
|
|
where LogPrefix is set to envoyContainer.Name
|
|
*/
|
|
func (s *HstSuite) CollectEnvoyLogs(envoyContainer *Container) {
|
|
targetDir := envoyContainer.Suite.getLogDirPath()
|
|
source := envoyContainer.GetHostWorkDir() + "/" + envoyContainer.Name + "-"
|
|
cmd := exec.Command("cp", "-t", targetDir, source+"access.log")
|
|
s.Log(cmd.String())
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
s.Log(fmt.Sprint(err))
|
|
}
|
|
}
|
|
|
|
func (s *HstSuite) StartIperfServerApp(running chan error, done chan struct{}, env []string) {
|
|
cmd := exec.Command("iperf3", "-4", "-s", "-p", s.GetPortFromPpid())
|
|
if env != nil {
|
|
cmd.Env = env
|
|
}
|
|
s.Log(cmd)
|
|
err := cmd.Start()
|
|
if err != nil {
|
|
msg := fmt.Errorf("failed to start iperf server: %v", err)
|
|
running <- msg
|
|
return
|
|
}
|
|
running <- nil
|
|
<-done
|
|
cmd.Process.Kill()
|
|
}
|
|
|
|
func (s *HstSuite) StartIperfClientApp(ipAddress string, env []string, clnCh chan error, clnRes chan string) {
|
|
defer func() {
|
|
clnCh <- nil
|
|
}()
|
|
|
|
nTries := 0
|
|
|
|
for {
|
|
cmd := exec.Command("iperf3", "-c", ipAddress, "-u", "-l", "1460", "-b", "10g", "-p", s.GetPortFromPpid())
|
|
if env != nil {
|
|
cmd.Env = env
|
|
}
|
|
s.Log(cmd)
|
|
o, err := cmd.CombinedOutput()
|
|
if err != nil {
|
|
if nTries > 5 {
|
|
clnRes <- ""
|
|
clnCh <- fmt.Errorf("failed to start client app '%s'.\n%s", err, o)
|
|
return
|
|
}
|
|
time.Sleep(1 * time.Second)
|
|
nTries++
|
|
continue
|
|
} else {
|
|
clnRes <- fmt.Sprintf("Client output: %s", o)
|
|
}
|
|
break
|
|
}
|
|
}
|
|
|
|
func (s *HstSuite) StartHttpServer(running chan struct{}, done chan struct{}, addressPort, netNs string) {
|
|
cmd := newCommand([]string{"./http_server", addressPort, s.Ppid, s.ProcessIndex}, netNs)
|
|
err := cmd.Start()
|
|
s.Log(cmd)
|
|
if err != nil {
|
|
s.Log("Failed to start http server: " + fmt.Sprint(err))
|
|
return
|
|
}
|
|
running <- struct{}{}
|
|
<-done
|
|
cmd.Process.Kill()
|
|
}
|
|
|
|
func (s *HstSuite) StartWget(finished chan error, server_ip, port, query, netNs string) {
|
|
defer func() {
|
|
finished <- errors.New("wget error")
|
|
}()
|
|
|
|
cmd := newCommand([]string{"wget", "--timeout=10", "--no-proxy", "--tries=5", "-O", "/dev/null", server_ip + ":" + port + "/" + query},
|
|
netNs)
|
|
s.Log(cmd)
|
|
o, err := cmd.CombinedOutput()
|
|
s.Log(string(o))
|
|
if err != nil {
|
|
finished <- fmt.Errorf("wget error: '%v\n\n%s'", err, o)
|
|
return
|
|
} else if !strings.Contains(string(o), "200 OK") {
|
|
finished <- fmt.Errorf("wget error: response not 200 OK")
|
|
return
|
|
}
|
|
finished <- nil
|
|
}
|
|
|
|
// Start a server app. 'processName' is used to check whether the app started correctly.
|
|
func (s *HstSuite) StartServerApp(c *Container, processName string, cmd string,
|
|
running chan error, done chan struct{}) {
|
|
|
|
s.Log("starting server")
|
|
c.ExecServer(true, cmd)
|
|
cmd2 := exec.Command("docker", "exec", c.Name, "pidof", processName)
|
|
err := cmd2.Run()
|
|
if err != nil {
|
|
msg := fmt.Errorf("failed to start server app: %v", err)
|
|
running <- msg
|
|
<-done
|
|
return
|
|
}
|
|
running <- nil
|
|
<-done
|
|
}
|
|
|
|
func (s *HstSuite) StartClientApp(c *Container, cmd string,
|
|
clnCh chan error, clnRes chan []byte) {
|
|
defer func() {
|
|
close(clnCh)
|
|
close(clnRes)
|
|
}()
|
|
|
|
s.Log("starting client app, please wait")
|
|
cmd2 := exec.Command("/bin/sh", "-c", "docker exec "+c.getEnvVarsAsCliOption()+" "+
|
|
c.Name+" "+cmd)
|
|
s.Log(cmd2)
|
|
o, err := cmd2.CombinedOutput()
|
|
|
|
if err != nil {
|
|
s.Log(err)
|
|
s.Log(string(o))
|
|
clnRes <- nil
|
|
clnCh <- fmt.Errorf("failed to start client app '%s'", err)
|
|
s.AssertNil(err, fmt.Sprint(err))
|
|
} else {
|
|
clnRes <- o
|
|
}
|
|
}
|
|
|
|
func (s *HstSuite) ParseJsonIperfOutput(jsonResult []byte) IPerfResult {
|
|
var result IPerfResult
|
|
// remove iperf warning line if present
|
|
if strings.Contains(string(jsonResult), "warning") {
|
|
index := strings.Index(string(jsonResult), "\n")
|
|
jsonResult = jsonResult[index+1:]
|
|
}
|
|
|
|
err := json.Unmarshal(jsonResult, &result)
|
|
s.AssertNil(err)
|
|
|
|
if result.Start.Details.Protocol == "TCP" {
|
|
result.End.TcpSent.MbitsPerSecond = result.End.TcpSent.MbitsPerSecond / 1000000
|
|
result.End.TcpSent.MBytes = result.End.TcpSent.MBytes / 1000000
|
|
result.End.TcpReceived.MbitsPerSecond = result.End.TcpReceived.MbitsPerSecond / 1000000
|
|
result.End.TcpReceived.MBytes = result.End.TcpReceived.MBytes / 1000000
|
|
} else {
|
|
result.End.Udp.MBytes = result.End.Udp.MBytes / 1000000
|
|
result.End.Udp.MbitsPerSecond = result.End.Udp.MbitsPerSecond / 1000000
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
func (s *HstSuite) LogJsonIperfOutput(result IPerfResult) {
|
|
s.Log("\n*******************************************\n"+
|
|
"%s\n"+
|
|
"[%s] %s:%d connected to %s:%d\n"+
|
|
"Started: %s\n",
|
|
result.Start.Version,
|
|
result.Start.Details.Protocol,
|
|
result.Start.Connected[0].LocalHost, result.Start.Connected[0].LocalPort,
|
|
result.Start.Connected[0].RemoteHost, result.Start.Connected[0].RemotePort,
|
|
result.Start.Timestamp.Time)
|
|
|
|
if result.Start.Details.Protocol == "TCP" {
|
|
s.Log("Transfer (sent): %.2f MBytes\n"+
|
|
"Bitrate (sent): %.2f Mbits/sec\n"+
|
|
"Transfer (received): %.2f MBytes\n"+
|
|
"Bitrate (received): %.2f Mbits/sec",
|
|
result.End.TcpSent.MBytes,
|
|
result.End.TcpSent.MbitsPerSecond,
|
|
result.End.TcpReceived.MBytes,
|
|
result.End.TcpReceived.MbitsPerSecond)
|
|
} else {
|
|
s.Log("Transfer: %.2f MBytes\n"+
|
|
"Bitrate: %.2f Mbits/sec\n"+
|
|
"Jitter: %.3f ms\n"+
|
|
"Packets: %d\n"+
|
|
"Packets lost: %d\n"+
|
|
"Percent lost: %.2f%%",
|
|
result.End.Udp.MBytes,
|
|
result.End.Udp.MbitsPerSecond,
|
|
result.End.Udp.JitterMs,
|
|
result.End.Udp.Packets,
|
|
result.End.Udp.LostPackets,
|
|
result.End.Udp.LostPercent)
|
|
}
|
|
s.Log("*******************************************\n")
|
|
}
|