Skip to content

Commit

Permalink
Propagate getMinReplicaCount error to the handler
Browse files Browse the repository at this point in the history
**What**
- Return parsing errors from getMinReplicaCount
- Add tests for the expected errors

Signed-off-by: Lucas Roesler <[email protected]>
  • Loading branch information
LucasRoesler committed Jun 23, 2018
1 parent c64dec6 commit dca3875
Show file tree
Hide file tree
Showing 4 changed files with 73 additions and 11 deletions.
5 changes: 4 additions & 1 deletion handlers/deploy.go
Original file line number Diff line number Diff line change
Expand Up @@ -131,7 +131,10 @@ func makeDeploymentSpec(request requests.CreateFunctionRequest, existingSecrets
probe = nil
}

initialReplicas := getMinReplicaCount(request.Labels)
initialReplicas, replicaErr := getMinReplicaCount(request.Labels)
if replicaErr != nil {
return nil, replicaErr
}
labels := buildLabels(request.Service, request.Labels)
nodeSelector := createSelector(request.Constraints)
resources, resourceErr := createResources(request)
Expand Down
19 changes: 12 additions & 7 deletions handlers/labels.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,10 @@ package handlers

import (
"fmt"
"log"
"strconv"
"time"

"github.com/pkg/errors"
)

const (
Expand Down Expand Up @@ -42,20 +43,24 @@ func buildLabels(functionName string, requestLables *map[string]string) map[stri

// getMinReplicaCount extracts the functions minimum replica count from the user's
// request labels. If the value is not found, this will return the default value, 1.
func getMinReplicaCount(labels *map[string]string) *int32 {
func getMinReplicaCount(labels *map[string]string) (*int32, error) {
if labels == nil {
return int32p(initialReplicasCount)
return int32p(initialReplicasCount), nil
}

l := *labels
if value, exists := l[FunctionMinReplicaCount]; exists {
minReplicas, err := strconv.Atoi(value)
if err == nil && minReplicas > 0 {
return int32p(int32(minReplicas))
if err != nil {
return nil, errors.Wrap(err, "could not parse the minimum replica value")
}

if minReplicas > 0 {
return int32p(int32(minReplicas)), nil
}

log.Println(err)
return nil, errors.New("replica count must be a positive integer")
}

return int32p(initialReplicasCount)
return int32p(initialReplicasCount), nil
}
51 changes: 49 additions & 2 deletions handlers/lables_test.go
Original file line number Diff line number Diff line change
@@ -1,6 +1,9 @@
package handlers

import "testing"
import (
"strings"
"testing"
)

func Test_getMinReplicaCount(t *testing.T) {
scenarios := []struct {
Expand All @@ -27,7 +30,10 @@ func Test_getMinReplicaCount(t *testing.T) {

for _, s := range scenarios {
t.Run(s.name, func(t *testing.T) {
output := getMinReplicaCount(s.labels)
output, err := getMinReplicaCount(s.labels)
if err != nil {
t.Errorf("getMinReplicaCount should not error on an empty or valid label")
}
if output == nil {
t.Errorf("getMinReplicaCount should not return nil pointer")
}
Expand All @@ -40,6 +46,47 @@ func Test_getMinReplicaCount(t *testing.T) {
}
}

func Test_getMinReplicaCount_Error(t *testing.T) {
scenarios := []struct {
name string
labels *map[string]string
msg string
}{
{
name: "negative values should return an error",
labels: &map[string]string{FunctionMinReplicaCount: "-2"},
msg: "replica count must be a positive integer",
},
{
name: "zero values should return an error",
labels: &map[string]string{FunctionMinReplicaCount: "0"},
msg: "replica count must be a positive integer",
},
{
name: "decimal values should return an error",
labels: &map[string]string{FunctionMinReplicaCount: "10.5"},
msg: "could not parse the minimum replica value",
},
{
name: "non-integer values should return an error",
labels: &map[string]string{FunctionMinReplicaCount: "test"},
msg: "could not parse the minimum replica value",
},
}

for _, s := range scenarios {
t.Run(s.name, func(t *testing.T) {
output, err := getMinReplicaCount(s.labels)
if output != nil {
t.Errorf("getMinReplicaCount should return nil value on invalid input")
}
if !strings.Contains(err.Error(), s.msg) {
t.Errorf("unexpected error: expected '%s', got '%s'", s.msg, err.Error())
}
})
}
}

func Test_parseLabels(t *testing.T) {
scenarios := []struct {
name string
Expand Down
9 changes: 8 additions & 1 deletion handlers/update.go
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,14 @@ func MakeUpdateHandler(functionNamespace string, clientset *kubernetes.Clientset
labels := buildLabels(request.Service, request.Labels)
deployment.Labels = labels
deployment.Spec.Template.ObjectMeta.Labels = labels
deployment.Spec.Replicas = getMinReplicaCount(request.Labels)

replicaCount, replicaErr := getMinReplicaCount(request.Labels)
if replicaErr != nil {
w.WriteHeader(http.StatusBadRequest)
w.Write([]byte(replicaErr.Error()))
return
}
deployment.Spec.Replicas = replicaCount

resources, resourceErr := createResources(request)
if resourceErr != nil {
Expand Down

0 comments on commit dca3875

Please sign in to comment.