Merge pull request #3481 from alaypatel07/dra-gke-plugin-in-kube-system

add resource quota for dra plugin needed for gke test

Author: k8s-ci-robot

clusterloader2/pkg/dependency/dra/dra.go

@@ -33,10 +33,11 @@ import (
 const (
 	draDependencyName = "DRATestDriver"
 	//TODO: this needs to be converted into a parameter. Will will not need this until parititionable devices test
-	draNamespace          = "dra-example-driver"
-	draDaemonsetName      = "dra-example-driver-kubeletplugin"
-	checkDRAReadyInterval = 30 * time.Second
-	defaultDRATimeout     = 10 * time.Minute
+	draNamespace           = "dra-example-driver"
+	defaultWorkerNodeCount = "100"
+	draDaemonsetName       = "dra-example-driver-kubeletplugin"
+	checkDRAReadyInterval  = 30 * time.Second
+	defaultDRATimeout      = 10 * time.Minute
 )
 
 //go:embed manifests/*.yaml
@@ -60,8 +61,19 @@ func (d *draDependency) Setup(config *dependency.Config) error {
 		return fmt.Errorf("namespace %s creation error: %v", draNamespace, err)
 	}
 
+	namespace, ok := config.Params["Namespace"]
+	if !ok {
+		namespace = draNamespace
+	}
+
+	workerCount, ok := config.Params["WorkerNodeCount"]
+	if !ok {
+		workerCount = defaultWorkerNodeCount
+	}
+
 	mapping := map[string]interface{}{
-		"Namespace": draNamespace,
+		"Namespace":       namespace,
+		"WorkerNodeCount": workerCount,
 	}
 	if err := config.ClusterFramework.ApplyTemplatedManifests(
 		manifestsFS,

clusterloader2/pkg/dependency/dra/manifests/resourceQuota.yaml

@@ -0,0 +1,20 @@
+apiVersion: v1
+kind: ResourceQuota
+metadata:
+  name: dra-example-driver-kubeletplugin
+  namespace: {{.Namespace}}
+  labels:
+    helm.sh/chart: dra-example-driver-0.1.3
+    app.kubernetes.io/name: dra-example-driver
+    app.kubernetes.io/instance: dra-example-driver
+    app.kubernetes.io/version: "v0.1.0"
+    app.kubernetes.io/component: kubeletplugin
+spec:
+  hard:
+    pods: {{.WorkerNodeCount}}
+  scopeSelector:
+    matchExpressions:
+      - operator: In
+        scopeName: PriorityClass
+        values:
+          - system-node-critical

clusterloader2/testing/dra/config.yaml

@@ -40,6 +40,8 @@ tuningSets:
 dependencies:
 - name: Install dra-example-driver for test
   Method: DRATestDriver
+  Params:
+    WorkerNodeCount: {{.Nodes}}
   Timeout: 5m
 
 steps: