run only caching test

Author: awesomenix

.pipelines/scripts/e2e_run.sh

@@ -94,7 +94,7 @@ rm -f "$temp_file"
 # gotestsum configure to only show logs for failed tests, json file for detailed logs
 # Run the tests! Yey!
 test_exit_code=0
-./bin/gotestsum --format testdox --junitfile "${BUILD_SRC_DIR}/e2e/report.xml" --jsonfile "${BUILD_SRC_DIR}/e2e/test-log.json" -- -parallel 150 -timeout 90m || test_exit_code=$?
+./bin/gotestsum --format standard-verbose --junitfile "${BUILD_SRC_DIR}/e2e/report.xml" --jsonfile "${BUILD_SRC_DIR}/e2e/test-log.json" -- -parallel 150 -timeout 90m -run ^Test_Ubuntu2404_VHDCaching$ || test_exit_code=$?
 
 # Upload test results as Azure DevOps artifacts
 echo "##vso[artifact.upload containerfolder=test-results;artifactname=e2e-test-log]${BUILD_SRC_DIR}/e2e/test-log.json"

e2e/test_helpers.go

@@ -120,6 +120,8 @@ func runScenarioWithPreProvision(t *testing.T, original *Scenario) {
 		customVHD = CreateImage(ctx, stage1)
 		customVHDJSON, _ := json.MarshalIndent(customVHD, "", "  ")
 		t.Logf("Created custom VHD image: %s", string(customVHDJSON))
+		cleanupBastionTunnel(firstStage.Runtime.VM.SSHClient)
+		firstStage.Runtime.VM.SSHClient = nil
 	}
 	firstStage.Config.VMConfigMutator = func(vmss *armcompute.VirtualMachineScaleSet) {
 		if original.VMConfigMutator != nil {