support custom template functions provided through RunConfig.funcs

bojand · Feb 24, 2021 · ddb571b · ddb571b
1 parent b9aa837
commit ddb571b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/runner/data.go b/runner/data.go
@@ -7,6 +7,7 @@ import (
 	"io"
 	"strings"
 	"sync"
+	"text/template"
 
 	"github.com/golang/protobuf/jsonpb"
 	"github.com/golang/protobuf/proto"
@@ -63,7 +64,8 @@ type mdProvider struct {
 }
 
 func newDataProvider(mtd *desc.MethodDescriptor,
-	binary bool, dataFunc BinaryDataFunc, data []byte) (*dataProvider, error) {
+	binary bool, dataFunc BinaryDataFunc, data []byte,
+	funcs template.FuncMap) (*dataProvider, error) {
 
 	dp := dataProvider{
 		binary:         binary,
@@ -96,7 +98,7 @@ func newDataProvider(mtd *desc.MethodDescriptor,
 	}
 
 	// Test if we can preseed data
-	ctd := newCallData(mtd, nil, "", 0)
+	ctd := newCallData(mtd, funcs, "", 0)
 	ha, err := ctd.hasAction(string(dp.data))
 	if err != nil {
 		return nil, err

diff --git a/runner/requester.go b/runner/requester.go
@@ -125,7 +125,7 @@ func NewRequester(c *RunConfig) (*Requester, error) {
 	if c.dataProviderFunc != nil {
 		reqr.dataProvider = c.dataProviderFunc
 	} else {
-		defaultDataProvider, err := newDataProvider(reqr.mtd, c.binary, c.dataFunc, c.data)
+		defaultDataProvider, err := newDataProvider(reqr.mtd, c.binary, c.dataFunc, c.data, c.funcs)
 		if err != nil {
 			return nil, err
 		}