extras/hs-test/vcl_test.go

   1 package main
   2
   3 import (
   4         "fmt"
   5         "time"
   6 )
   7
   8 func getVclConfig(c *Container, ns_id_optional ...string) string {
   9         var s Stanza
  10         ns_id := "default"
  11         if len(ns_id_optional) > 0 {
  12                 ns_id = ns_id_optional[0]
  13         }
  14         s.newStanza("vcl").
  15                 append(fmt.Sprintf("app-socket-api %[1]s/var/run/app_ns_sockets/%[2]s", c.getContainerWorkDir(), ns_id)).
  16                 append("app-scope-global").
  17                 append("app-scope-local").
  18                 append("use-mq-eventfd")
  19         if len(ns_id_optional) > 0 {
  20                 s.append(fmt.Sprintf("namespace-id %[1]s", ns_id)).
  21                         append(fmt.Sprintf("namespace-secret %[1]s", ns_id))
  22         }
  23         return s.close().toString()
  24 }
  25
  26 func (s *VethsSuite) TestXEchoVclClientUdp() {
  27         s.testXEchoVclClient("udp")
  28 }
  29
  30 func (s *VethsSuite) TestXEchoVclClientTcp() {
  31         s.testXEchoVclClient("tcp")
  32 }
  33
  34 func (s *VethsSuite) testXEchoVclClient(proto string) {
  35         port := "12345"
  36         serverVpp := s.getContainerByName("server-vpp").vppInstance
  37
  38         serverVeth := s.netInterfaces[serverInterfaceName]
  39         serverVpp.vppctl("test echo server uri %s://%s/%s fifo-size 64k", proto, serverVeth.ip4AddressString(), port)
  40
  41         echoClnContainer := s.getTransientContainerByName("client-app")
  42         echoClnContainer.createFile("/vcl.conf", getVclConfig(echoClnContainer))
  43
  44         testClientCommand := "vcl_test_client -N 100 -p " + proto + " " + serverVeth.ip4AddressString() + " " + port
  45         s.log(testClientCommand)
  46         echoClnContainer.addEnvVar("VCL_CONFIG", "/vcl.conf")
  47         o := echoClnContainer.exec(testClientCommand)
  48         s.log(o)
  49         s.assertContains(o, "CLIENT RESULTS")
  50 }
  51
  52 func (s *VethsSuite) TestXEchoVclServerUdp() {
  53         s.testXEchoVclServer("udp")
  54 }
  55
  56 func (s *VethsSuite) TestXEchoVclServerTcp() {
  57         s.testXEchoVclServer("tcp")
  58 }
  59
  60 func (s *VethsSuite) testXEchoVclServer(proto string) {
  61         port := "12345"
  62         srvVppCont := s.getContainerByName("server-vpp")
  63         srvAppCont := s.getContainerByName("server-app")
  64
  65         srvAppCont.createFile("/vcl.conf", getVclConfig(srvVppCont))
  66         srvAppCont.addEnvVar("VCL_CONFIG", "/vcl.conf")
  67         vclSrvCmd := fmt.Sprintf("vcl_test_server -p %s %s", proto, port)
  68         srvAppCont.execServer(vclSrvCmd)
  69
  70         serverVeth := s.netInterfaces[serverInterfaceName]
  71         serverVethAddress := serverVeth.ip4AddressString()
  72
  73         clientVpp := s.getContainerByName("client-vpp").vppInstance
  74         o := clientVpp.vppctl("test echo client uri %s://%s/%s fifo-size 64k verbose mbytes 2", proto, serverVethAddress, port)
  75         s.log(o)
  76         s.assertContains(o, "Test finished at")
  77 }
  78
  79 func (s *VethsSuite) testVclEcho(proto string) {
  80         port := "12345"
  81         srvVppCont := s.getContainerByName("server-vpp")
  82         srvAppCont := s.getContainerByName("server-app")
  83
  84         srvAppCont.createFile("/vcl.conf", getVclConfig(srvVppCont))
  85         srvAppCont.addEnvVar("VCL_CONFIG", "/vcl.conf")
  86         srvAppCont.execServer("vcl_test_server " + port)
  87
  88         serverVeth := s.netInterfaces[serverInterfaceName]
  89         serverVethAddress := serverVeth.ip4AddressString()
  90
  91         echoClnContainer := s.getTransientContainerByName("client-app")
  92         echoClnContainer.createFile("/vcl.conf", getVclConfig(echoClnContainer))
  93
  94         testClientCommand := "vcl_test_client -p " + proto + " " + serverVethAddress + " " + port
  95         echoClnContainer.addEnvVar("VCL_CONFIG", "/vcl.conf")
  96         o := echoClnContainer.exec(testClientCommand)
  97         s.log(o)
  98 }
  99
 100 func (s *VethsSuite) TestVclEchoTcp() {
 101         s.testVclEcho("tcp")
 102 }
 103
 104 func (s *VethsSuite) TestVclEchoUdp() {
 105         s.testVclEcho("udp")
 106 }
 107
 108 func (s *VethsSuite) TestVclRetryAttach() {
 109         s.skip("this test takes too long, for now it's being skipped")
 110         s.testRetryAttach("tcp")
 111 }
 112
 113 func (s *VethsSuite) testRetryAttach(proto string) {
 114         srvVppContainer := s.getTransientContainerByName("server-vpp")
 115
 116         echoSrvContainer := s.getContainerByName("server-app")
 117
 118         echoSrvContainer.createFile("/vcl.conf", getVclConfig(echoSrvContainer))
 119
 120         echoSrvContainer.addEnvVar("VCL_CONFIG", "/vcl.conf")
 121         echoSrvContainer.execServer("vcl_test_server -p " + proto + " 12346")
 122
 123         s.log("This whole test case can take around 3 minutes to run. Please be patient.")
 124         s.log("... Running first echo client test, before disconnect.")
 125
 126         serverVeth := s.netInterfaces[serverInterfaceName]
 127         serverVethAddress := serverVeth.ip4AddressString()
 128
 129         echoClnContainer := s.getTransientContainerByName("client-app")
 130         echoClnContainer.createFile("/vcl.conf", getVclConfig(echoClnContainer))
 131
 132         testClientCommand := "vcl_test_client -U -p " + proto + " " + serverVethAddress + " 12346"
 133         echoClnContainer.addEnvVar("VCL_CONFIG", "/vcl.conf")
 134         o := echoClnContainer.exec(testClientCommand)
 135         s.log(o)
 136         s.log("... First test ended. Stopping VPP server now.")
 137
 138         // Stop server-vpp-instance, start it again and then run vcl-test-client once more
 139         srvVppContainer.vppInstance.disconnect()
 140         stopVppCommand := "/bin/bash -c 'ps -C vpp_main -o pid= | xargs kill -9'"
 141         srvVppContainer.exec(stopVppCommand)
 142
 143         s.setupServerVpp()
 144
 145         s.log("... VPP server is starting again, so waiting for a bit.")
 146         time.Sleep(30 * time.Second) // Wait a moment for the re-attachment to happen
 147
 148         s.log("... Running second echo client test, after disconnect and re-attachment.")
 149         o = echoClnContainer.exec(testClientCommand)
 150         s.log(o)
 151         s.log("Done.")
 152 }