Include Test case for Limits

Signed-off-by: Nitishkumar Singh <[email protected]>

Makefile

@@ -15,7 +15,8 @@ TEST_FUNCTIONS = \
 	test-scaling-disabled \
 	test-scaling-to-zero \
 	test-logger \
-	redirector-test
+	redirector-test \
+	memory-limit
 
 TEST_SECRETS = \
 	secret-name

tests/deploy_test.go

@@ -92,6 +92,20 @@ func Test_Deploy_MetaData(t *testing.T) {
 				Namespace: config.DefaultNamespace,
 			},
 		},
+		{
+			name: "Deploy with memory limit",
+			function: types.FunctionDeployment{
+				Image:       imagePath,
+				Service:     "memory-limit",
+				EnvProcess:  "env",
+				Annotations: &map[string]string{},
+				Labels:      &map[string]string{},
+				Namespace:   config.DefaultNamespace,
+				Limits: &types.FunctionResources{
+					Memory: "10M",
+				},
+			},
+		},
 	}
 
 	// Add Test case, if CERTIFIER_NAMESPACES defined
@@ -224,6 +238,17 @@ func compareDeployAndStatus(deploy types.FunctionDeployment, status types.Functi
 		return fmt.Errorf("got %v, expected Requests %v", status.Requests, deploy.Requests)
 	}
 
+	if deploy.Limits != nil {
+		if status.Limits == nil {
+			return fmt.Errorf("got nil Requested Limit, expected %v", *deploy.Limits)
+		}
+		deployLimits := *deploy.Limits
+		statusLimits := *status.Limits
+		if deployLimits.Memory != statusLimits.Memory {
+			return fmt.Errorf("got %s, expected Requested Limit %s", statusLimits.Memory, deployLimits.Memory)
+		}
+	}
+
 	if config.ProviderName != faasdProviderName {
 		// we expect all systems to add the `faas_function` label?
 		expectedLabels := copyStrMap(deploy.Labels)

tests/function_helpers_test.go

@@ -7,6 +7,7 @@ import (
 	"testing"
 
 	sdk "github.com/openfaas/faas-cli/proxy"
+	"github.com/openfaas/faas-cli/stack"
 	"github.com/openfaas/faas-provider/types"
 )
 
@@ -99,5 +100,14 @@ func createDeploymentSpec(test FunctionTestCase) *sdk.DeployFunctionSpec {
 		functionRequest.Labels = *test.function.Labels
 	}
 
+	if test.function.Limits != nil {
+		limits := *test.function.Limits
+		functionRequest.FunctionResourceRequest = sdk.FunctionResourceRequest{
+			Limits: &stack.FunctionResources{
+				Memory: limits.Memory,
+			},
+		}
+	}
+
 	return functionRequest
 }