hs-test: add test suite features

Test suite now supports assertions which on fail stop test case run,
also it allows to create docker containers which are going to be
stopped automatically after the test run is finished.

Type: improvement
Signed-off-by: Maros Ondrejicka <maros.ondrejicka@pantheon.tech>
Change-Id: I2834709b1efd17b8182d36cc0404b986b4ed595d
Signed-off-by: Filip Tehlar <ftehlar@cisco.com>

1 files changed, 124 insertions, 0 deletions

diff --git a/extras/hs-test/vppinstance.go b/extras/hs-test/vppinstance.go
new file mode 100644
index 00000000000..c6d3935cc60
--- /dev/null
+++ b/extras/hs-test/vppinstance.go
@@ -0,0 +1,124 @@
+package main
+
+import (
+	"fmt"
+	"encoding/json"
+	"github.com/edwarnicke/exechelper"
+)
+
+const vppConfigTemplate = `unix {
+  nodaemon
+  log %[1]s/var/log/vpp/vpp.log
+  full-coredump
+  cli-listen %[1]s%[2]s
+  runtime-dir %[1]s/var/run
+  gid vpp
+}
+
+api-trace {
+  on
+}
+
+api-segment {
+  gid vpp
+}
+
+socksvr {
+  socket-name %[1]s/var/run/vpp/api.sock
+}
+
+statseg {
+  socket-name %[1]s/var/run/vpp/stats.sock
+}
+
+plugins {
+  plugin unittest_plugin.so { enable }
+  plugin dpdk_plugin.so { disable }
+  plugin crypto_aesni_plugin.so { enable }
+  plugin quic_plugin.so { enable }
+}
+
+`
+
+type VppInstance struct {
+	container *Container
+	config VppConfig
+	actionFuncName string
+}
+
+type VppConfig struct {
+	Variant string
+	CliSocketFilePath string
+}
+
+func (vc *VppConfig) getTemplate() string {
+	return fmt.Sprintf(vppConfigTemplate, "%[1]s", vc.CliSocketFilePath)
+}
+
+func (vpp *VppInstance) set2VethsServer() {
+	vpp.actionFuncName = "Configure2Veths"
+	vpp.config.Variant = "srv"
+}
+
+func (vpp *VppInstance) set2VethsClient() {
+	vpp.actionFuncName = "Configure2Veths"
+	vpp.config.Variant = "cln"
+}
+
+func (vpp *VppInstance) setCliSocket(filePath string) {
+	vpp.config.CliSocketFilePath = filePath
+}
+
+func (vpp *VppInstance) getCliSocket() string {
+	return fmt.Sprintf("/tmp/%s/%s", vpp.actionFuncName, vpp.config.CliSocketFilePath)
+}
+
+func (vpp *VppInstance) start() error {
+	if vpp.config.Variant == "" {
+		return fmt.Errorf("vpp start failed: variant must not be blank")
+	}
+	if vpp.actionFuncName == "" {
+		return fmt.Errorf("vpp start failed: action function name must not be blank")
+	}
+
+	serializedConfig, err := json.Marshal(vpp.config)
+	if err != nil {
+		return fmt.Errorf("vpp start failed: serializing configuration failed: %s", err)
+	}
+	args := fmt.Sprintf("%s '%s'", vpp.actionFuncName, string(serializedConfig))
+	_, err = hstExec(args, vpp.container.name)
+	if err != nil {
+		return fmt.Errorf("vpp start failed: %s", err)
+	}
+
+	return nil
+}
+
+func (vpp *VppInstance) vppctl(command string) (string, error) {
+	dockerExecCommand := fmt.Sprintf("docker exec --detach=false %[1]s vppctl -s %[2]s %[3]s",
+		vpp.container.name, vpp.getCliSocket(), command)
+	output, err := exechelper.CombinedOutput(dockerExecCommand)
+	if err != nil {
+		return "", fmt.Errorf("vppctl failed: %s", err)
+	}
+
+	return string(output), nil
+}
+
+func NewVppInstance(c *Container) *VppInstance {
+	vpp := new(VppInstance)
+	vpp.container = c
+	return vpp
+}
+
+func DeserializeVppConfig(input string) (VppConfig, error) {
+	var vppConfig VppConfig
+	err := json.Unmarshal([]byte(input), &vppConfig)
+	if err != nil {
+		// Since input is not a  valid JSON it is going be used as variant value
+		// for compatibility reasons
+		vppConfig.Variant = input
+		vppConfig.CliSocketFilePath = "/var/run/vpp/cli.sock"
+	}
+	return vppConfig, nil
+}