Add "local build" mode to running the e2e over kind (#219)

* Allow the e2e to use the current code as the base for version to be tested by the e2e

Author: davidLif

hack/run-e2e-kind.sh

@@ -10,11 +10,33 @@ KIND_CONFIG=${REPO_ROOT}/hack/e2e-kind-config.yaml
 GOPATH=${HOME}/go
 GOBIN=${GOPATH}/bin
 
-TEST_THIRD_PARTY_INTEGRATIONS=${1}
-if [ -z "$TEST_THIRD_PARTY_INTEGRATIONS" ]; then
-    echo "TEST_THIRD_PARTY_INTEGRATIONS argument isn't provided, defaulting to false"
-    TEST_THIRD_PARTY_INTEGRATIONS="false"
-fi
+# Parse named parameters
+TEST_THIRD_PARTY_INTEGRATIONS="false"
+LOCAL_IMAGES_BUILD="false"
+
+while [[ $# -gt 0 ]]; do
+  case $1 in
+    --test-third-party-integrations)
+      TEST_THIRD_PARTY_INTEGRATIONS="true"
+      shift
+      ;;
+    --local-images-build)
+      LOCAL_IMAGES_BUILD="true"
+      shift
+      ;;
+    -h|--help)
+      echo "Usage: $0 [--test-third-party-integrations] [--local-images-build]"
+      echo "  --test-third-party-integrations: Install third party operators for compatibility testing"
+      echo "  --local-images-build: Build and use local images instead of pulling from registry"
+      exit 0
+      ;;
+    *)
+      echo "Unknown option $1"
+      echo "Use --help for usage information"
+      exit 1
+      ;;
+  esac
+done
 
 kind create cluster --config ${KIND_CONFIG} --name $CLUSTER_NAME
 
@@ -28,8 +50,21 @@ if [ "$TEST_THIRD_PARTY_INTEGRATIONS" = "true" ]; then
     ${REPO_ROOT}/hack/third_party_integrations/deploy_knative.sh
 fi
 
-PACKAGE_VERSION=0.0.0-$(git rev-parse --short origin/main)
-helm upgrade -i kai-scheduler oci://ghcr.io/nvidia/kai-scheduler/kai-scheduler -n kai-scheduler --create-namespace --set "global.gpuSharing=true" --version "$PACKAGE_VERSION"
+if [ "$LOCAL_IMAGES_BUILD" = "true" ]; then
+    cd ${REPO_ROOT}
+    PACKAGE_VERSION=0.0.0-$(git rev-parse --short HEAD)
+    make build VERSION=$PACKAGE_VERSION
+    for image in $(docker images --format '{{.Repository}}:{{.Tag}}' | grep $PACKAGE_VERSION); do
+        kind load docker-image $image --name $CLUSTER_NAME
+    done
+    helm package ./deployments/kai-scheduler -d ./charts --app-version $PACKAGE_VERSION --version $PACKAGE_VERSION
+    helm upgrade -i kai-scheduler ./charts/kai-scheduler-$PACKAGE_VERSION.tgz  -n kai-scheduler --create-namespace --set "global.gpuSharing=true" --wait
+    rm -rf ./charts/kai-scheduler-$PACKAGE_VERSION.tgz 
+    cd ${REPO_ROOT}/hack
+else
+    PACKAGE_VERSION=0.0.0-$(git rev-parse --short origin/main)
+    helm upgrade -i kai-scheduler oci://ghcr.io/nvidia/kai-scheduler/kai-scheduler -n kai-scheduler --create-namespace --set "global.gpuSharing=true" --version "$PACKAGE_VERSION"
+fi
 
 # Allow all the pods in the fake-gpu-operator and kai-scheduler to start
 sleep 30