29 lines
656 B
Go
29 lines
656 B
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/testground/sdk-go/run"
|
|
"github.com/testground/sdk-go/runtime"
|
|
)
|
|
|
|
var testplans = map[string]interface{}{
|
|
"lotus-baseline": doRun(baselineRoles),
|
|
"lotus-deal-stress-test": doRun(dealStressTestRoles),
|
|
}
|
|
|
|
func main() {
|
|
run.InvokeMap(testplans)
|
|
}
|
|
|
|
func doRun(roles map[string]func(*TestEnvironment) error) run.InitializedTestCaseFn {
|
|
return func(runenv *runtime.RunEnv, initCtx *run.InitContext) error {
|
|
role := runenv.StringParam("role")
|
|
proc, ok := roles[role]
|
|
if ok {
|
|
return proc(&TestEnvironment{RunEnv: runenv, InitContext: initCtx})
|
|
}
|
|
return fmt.Errorf("Unknown role: %s", role)
|
|
}
|
|
}
|