Fix failure due to ScaleWorkload replicas type (#236)

kanisterio · Aug 26, 2019 · 088ab18 · 088ab18
1 parent e36c022
commit 088ab18
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/pkg/function/scale_test.go b/pkg/function/scale_test.go
@@ -269,7 +269,7 @@ func (s *ScaleSuite) TestGetArgs(c *C) {
 				},
 			},
 			args: map[string]interface{}{
-				ScaleWorkloadReplicas: "2",
+				ScaleWorkloadReplicas: int64(2),
 			},
 			wantKind:      param.DeploymentKind,
 			wantName:      "app",
@@ -285,7 +285,7 @@ func (s *ScaleSuite) TestGetArgs(c *C) {
 				},
 			},
 			args: map[string]interface{}{
-				ScaleWorkloadReplicas:     2,
+				ScaleWorkloadReplicas:     int32(2),
 				ScaleWorkloadNamespaceArg: "notfoo",
 				ScaleWorkloadNameArg:      "notapp",
 				ScaleWorkloadKindArg:      param.DeploymentKind,

diff --git a/pkg/function/scale_workload.go b/pkg/function/scale_workload.go
@@ -70,6 +70,10 @@ func getArgs(tp param.TemplateParams, args map[string]interface{}) (namespace, k
 	switch rep.(type) {
 	case int:
 		val = rep.(int)
+	case int32:
+		val = int(rep.(int32))
+	case int64:
+		val = int(rep.(int64))
 	case string:
 		if val, err = strconv.Atoi(rep.(string)); err != nil {
 			err = errors.Wrapf(err, "Cannot convert %s to int ", rep.(string))