hs-test: auto register test actions

[vpp.git] / extras / hs-test / main.go

diff --git a/extras/hs-test/main.go b/extras/hs-test/main.go
index f54b6c4..9de0d31 100755 (executable)
--- 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)