hs-test: add nginx perf tests
[vpp.git] / extras / hs-test / http_test.go
old mode 100755 (executable)
new mode 100644 (file)
index 99b509f..28d27bb
 package main
 
 import (
+       "fmt"
+       "os"
+       "os/exec"
        "strings"
-
-       "github.com/edwarnicke/exechelper"
+       "time"
 )
 
 func (s *NsSuite) TestHttpTps() {
-       t := s.T()
        finished := make(chan error, 1)
        server_ip := "10.0.0.2"
        port := "8080"
-       dockerInstance := "http-tps"
 
-       t.Log("starting vpp..")
+       container := s.getContainerByName("vpp")
 
-       err := dockerRun(dockerInstance, "")
-       if err != nil {
-               t.Errorf("%v", err)
-               return
-       }
-       defer func() { exechelper.Run("docker stop " + dockerInstance) }()
+       s.log("starting vpp..")
 
        // start & configure vpp in the container
-       _, err = hstExec("ConfigureHttpTps", dockerInstance)
-       if err != nil {
-               t.Errorf("%v", err)
-               return
-       }
+       _, err := container.execAction("ConfigureHttpTps")
+       s.assertNil(err)
 
-       go startWget(finished, server_ip, port, "client")
+       go startWget(finished, server_ip, port, "test_file_10M", "client")
        // wait for client
        err = <-finished
-       if err != nil {
-               t.Errorf("%v", err)
-       }
+       s.assertNil(err)
 }
 
 func (s *VethsSuite) TestHttpCli() {
-       t := s.T()
-
-       srvInstance := "http-cli-srv"
-       clnInstance := "http-cli-cln"
-       err := dockerRun(srvInstance, "")
-       if err != nil {
-               t.Errorf("%v", err)
-               return
-       }
-       defer func() { exechelper.Run("docker stop " + srvInstance) }()
+       serverContainer := s.getContainerByName("server-vpp")
+       clientContainer := s.getContainerByName("client-vpp")
 
-       err = dockerRun(clnInstance, "")
-       if err != nil {
-               t.Errorf("%v", err)
-               return
-       }
-       defer func() { exechelper.Run("docker stop " + clnInstance) }()
+       _, err := serverContainer.execAction("Configure2Veths srv")
+       s.assertNil(err)
 
-       _, err = hstExec("Configure2Veths srv", srvInstance)
-       if err != nil {
-               t.Errorf("%v", err)
-               return
-       }
+       _, err = clientContainer.execAction("Configure2Veths cln")
+       s.assertNil(err)
 
-       _, err = hstExec("Configure2Veths cln", clnInstance)
-       if err != nil {
-               t.Errorf("%v", err)
-               return
-       }
+       s.log("configured IPs...")
 
-       t.Log("configured IPs...")
+       _, err = serverContainer.execAction("RunHttpCliSrv")
+       s.assertNil(err)
 
-       _, err = hstExec("RunHttpCliSrv", srvInstance)
-       if err != nil {
-               t.Errorf("%v", err)
-               return
-       }
+       s.log("configured http server")
+
+       o, err := clientContainer.execAction("RunHttpCliCln /show/version")
+       s.assertNil(err)
 
-       t.Log("configured http server")
+       s.assertContains(o, "<html>", "<html> not found in the result!")
+}
 
-       o, err := hstExec("RunHttpCliCln /show/version", clnInstance)
-       if err != nil {
-               t.Errorf("%v", err)
-               return
+func waitForApp(vppInst *VppInstance, appName string, timeout int) error {
+       for i := 0; i < timeout; i++ {
+               o, err := vppInst.vppctl("show app")
+               if err != nil {
+                       return fmt.Errorf("Error ocurred during 'show app'")
+               }
+               if strings.Contains(o, appName) {
+                       return nil
+               }
+               time.Sleep(1 * time.Second)
        }
+       return fmt.Errorf("Timeout while waiting for app '%s'", appName)
+}
 
-       if strings.Index(o, "<html>") < 0 {
-               t.Error("<html> not found in the result!")
+func (s *NoTopoSuite) TestNginx() {
+       query := "return_ok"
+       finished := make(chan error, 1)
+       vppCont := s.getContainerByName("vpp")
+       vppInst := NewVppInstance(vppCont)
+       vppInst.actionFuncName = "ConfigureTap"
+       s.assertNil(vppInst.start(), "failed to start vpp")
+
+       nginxCont := s.getContainerByName("nginx")
+       s.assertNil(nginxCont.run())
+
+       err := waitForApp(vppInst, "-app", 5)
+       s.assertNil(err)
+
+       defer func() { os.Remove(query) }()
+       go startWget(finished, "10.10.10.1", "80", query, "")
+       s.assertNil(<-finished)
+}
+
+func runNginxPerf(s *NoTopoSuite, mode, ab_or_wrk string) error {
+       nRequests := 1000000
+       nClients := 2000
+       var args []string
+       var exeName string
+
+       if ab_or_wrk == "ab" {
+               args = []string{"-n", fmt.Sprintf("%d", nRequests), "-c",
+                       fmt.Sprintf("%d", nClients)}
+               if mode == "rps" {
+                       args = append(args, "-k")
+               } else if mode != "cps" {
+                       return fmt.Errorf("invalid mode %s; expected cps/rps", mode)
+               }
+               args = append(args, "http://10.10.10.1:80/64B.json")
+               exeName = "ab"
+       } else {
+               args = []string{"-c", fmt.Sprintf("%d", nClients), "-t", "2", "-d", "30",
+                       "http://10.10.10.1:80"}
+               exeName = "wrk"
        }
+
+       vppCont := s.getContainerByName("vpp")
+       vppInst := NewVppInstance(vppCont)
+       vppInst.actionFuncName = "ConfigureTap"
+       s.assertNil(vppInst.start(), "failed to start vpp")
+
+       nginxCont := s.getContainerByName("nginx")
+       s.assertNil(nginxCont.run())
+       err := waitForApp(vppInst, "-app", 5)
+       s.assertNil(err)
+
+       cmd := exec.Command(exeName, args...)
+       fmt.Println(cmd)
+       o, _ := cmd.CombinedOutput()
+       fmt.Print(string(o))
+       return nil
+}
+
+func (s *NoTopoSuite) TestNginxPerfCps() {
+       s.assertNil(runNginxPerf(s, "cps", "ab"))
+}
+
+func (s *NoTopoSuite) TestNginxPerfRps() {
+       s.assertNil(runNginxPerf(s, "rps", "ab"))
+}
+
+func (s *NoTopoSuite) TestNginxPerfWrk() {
+       s.assertNil(runNginxPerf(s, "", "wrk"))
 }