hs-test: auto register test actions

Type: improvement

Signed-off-by: Filip Tehlar <ftehlar@cisco.com>
Change-Id: Icb5db6f69eda93181aba69b1f8676a73c0a4561b

1 files changed, 12 insertions, 16 deletions

diff --git a/extras/hs-test/main.go b/extras/hs-test/main.go
index f54b6c4d910..9de0d314cb2 100755
--- a/extras/hs-test/main.go
+++ b/extras/hs-test/main.go
@@ -7,15 +7,10 @@ import (
 	"os"
 	"os/exec"
 	"os/signal"
-
-	"git.fd.io/govpp.git/api"
+	"reflect"
 )
 
-type CfgTable map[string]func([]string) *ActionResult
-
-var cfgTable CfgTable
-
-type ConfFn func(context.Context, api.Connection) error
+var actions Actions
 
 func newVppContext() (context.Context, context.CancelFunc) {
 	ctx, cancel := signal.NotifyContext(
@@ -116,15 +111,18 @@ func OkResult() *ActionResult {
 	return NewActionResult(nil)
 }
 
-func reg(key string, fn func([]string) *ActionResult) {
-	cfgTable[key] = fn
-}
-
 func processArgs() *ActionResult {
-	fn := cfgTable[os.Args[1]]
-	if fn == nil {
-		return NewActionResult(fmt.Errorf("internal: no config found for %s", os.Args[1]))
+	nArgs := len(os.Args) - 1 // skip program name
+	if nArgs < 1 {
+		return NewActionResult(fmt.Errorf("internal: no action specified!"))
+	}
+	action := os.Args[1]
+	methodValue := reflect.ValueOf(&actions).MethodByName(action)
+	if !methodValue.IsValid() {
+		return NewActionResult(fmt.Errorf("internal unknown action %s!", action))
 	}
+	methodIface := methodValue.Interface()
+	fn := methodIface.(func([]string) *ActionResult)
 	return fn(os.Args)
 }
 
@@ -144,8 +142,6 @@ func main() {
 		os.Exit(0)
 	}
 
-	RegisterActions()
-
 	var err error
 	res := processArgs()
 	err = writeSyncFile(res)