hs-test: add nginx perf tests

[vpp.git] / extras / hs-test / http_test.go
diff --git a/extras/hs-test/http_test.go b/extras/hs-test/http_test.go

old mode 100755 (executable)

new mode 100644 (file)

index ae1c8b8..28d27bb
--- a/extras/hs-test/http_test.go
+++ b/extras/hs-test/http_test.go
@@ -1,5 +1,13 @@
  package main
  
+import (
+       "fmt"
+       "os"
+       "os/exec"
+       "strings"
+       "time"
+)
+
  func (s *NsSuite) TestHttpTps() {
         finished := make(chan error, 1)
         server_ip := "10.0.0.2"
@@ -13,7 +21,7 @@ func (s *NsSuite) TestHttpTps() {
         _, 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
         s.assertNil(err)
@@ -41,3 +49,87 @@ func (s *VethsSuite) TestHttpCli() {
  
         s.assertContains(o, "<html>", "<html> not found in the result!")
  }
+
+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)
+}
+
+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"))
+}