add more tests

bojand · Sep 28, 2021 · f3b6b46 · f3b6b46
1 parent bae6051
commit f3b6b46
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 3 deletions.
diff --git a/runner/calldata.go b/runner/calldata.go
@@ -58,7 +58,7 @@ func newCallData(
 		fns[k] = v
 	}
 
-	for k, v := range sprig.TxtFuncMap() {
+	for k, v := range sprig.FuncMap() {
 		fns[k] = v
 	}
 

diff --git a/runner/calldata_test.go b/runner/calldata_test.go
@@ -317,4 +317,18 @@ func TestCallTemplateData_ExecuteFuncs(t *testing.T) {
 		assert.Equal(t, "custom-uuid", rm["trace_id"])
 		assert.Equal(t, "custom-sku", rm["sku"])
 	})
+
+	t.Run("sprig functions", func(t *testing.T) {
+
+		ctd := newCallData(md, nil, "worker_id_123", 200)
+		assert.NotNil(t, ctd)
+
+		r, err := ctd.ExecuteData(`{"trace_id":"{{add 1 2}}"}`)
+		assert.NoError(t, err)
+		assert.Equal(t, `{"trace_id":"3"}`, string(r))
+
+		r, err = ctd.ExecuteData(`{"trace_id":"{{"abc" | upper | repeat 3 }}"}`)
+		assert.NoError(t, err)
+		assert.Equal(t, `{"trace_id":"ABCABCABC"}`, string(r))
+	})
 }
diff --git a/runner/run_test.go b/runner/run_test.go
@@ -116,7 +116,7 @@ func TestRunUnary(t *testing.T) {
 		gs.ResetCounters()
 
 		data := make(map[string]interface{})
-		data["name"] = "{{ add 1 2 }}"
+		data["name"] = "{{ `test` | upper | repeat 5 }}"
 
 		report, err := Run(
 			"helloworld.Greeter.SayHello",
@@ -140,7 +140,7 @@ func TestRunUnary(t *testing.T) {
 		assert.Len(t, calls, 1)
 
 		msg := calls[0][0]
-		assert.Equal(t, msg.GetName(), "3")
+		assert.Equal(t, msg.GetName(), "TESTTESTTESTTESTTEST")
 	})
 
 	t.Run("test custom template functions", func(t *testing.T) {