ci: add debug code for test vm lockup on windows

Signed-off-by: Austin Vazquez <[email protected]>

e2e/vm/vm_darwin_test.go

@@ -96,3 +96,8 @@ var resetDisks = func(_ *option.Option, installed bool) {
 	}
 	gomega.Expect(os.RemoveAll(dataDiskDir)).ShouldNot(gomega.HaveOccurred())
 }
+
+var shutdownWSL = func() error {
+	// no-op on darwin
+	return nil
+}

e2e/vm/vm_test.go

@@ -4,7 +4,6 @@
 package vm
 
 import (
-	"os/exec"
 	"runtime"
 	"time"
 
@@ -29,7 +28,7 @@ var resetVM = func(o *option.Option) {
 		// clean up iptables
 		//nolint:lll // link to explanation
 		// https://docs.rancherdesktop.io/troubleshooting-tips/#q-how-do-i-fix-fata0005-subnet-1040024-overlaps-with-other-one-on-this-address-space-when-running-a-container-using-nerdctl-run
-		gomega.Expect(exec.Command("wsl", "--shutdown").Run()).Should(gomega.BeNil())
+		gomega.Expect(shutdownWSL()).Should(gomega.BeNil())
 	}
 
 	ginkgo.DeferCleanup(func() {
@@ -38,7 +37,7 @@ var resetVM = func(o *option.Option) {
 		time.Sleep(1 * time.Second)
 		command.New(o, virtualMachineRootCmd, "remove", "-f").WithoutCheckingExitCode().WithTimeoutInSeconds(10).Run()
 		if runtime.GOOS == "windows" {
-			gomega.Expect(exec.Command("wsl", "--shutdown").Run()).Should(gomega.BeNil())
+			gomega.Expect(shutdownWSL()).Should(gomega.BeNil())
 		}
 		time.Sleep(1 * time.Second)
 		command.New(o, virtualMachineRootCmd, "init").WithoutCheckingExitCode().WithTimeoutInSeconds(160).Run()

e2e/vm/vm_windows_test.go

@@ -7,9 +7,13 @@
 package vm
 
 import (
+	"context"
+	"fmt"
 	"os"
+	"os/exec"
 	"path/filepath"
 	"testing"
+	"time"
 
 	"github.com/onsi/ginkgo/v2"
 	"github.com/onsi/gomega"
@@ -60,3 +64,34 @@ var resetDisks = func(_ *option.Option, _ bool) {
 	dataDiskDir := filepath.Join(finchRootDir, ".finch", ".disks")
 	gomega.Expect(os.RemoveAll(dataDiskDir)).ShouldNot(gomega.HaveOccurred())
 }
+
+// shutdownWSL is a wrapper function for "wsl --shutdown" to only block for 15 seconds
+// before failing.
+var shutdownWSL = func() error {
+	ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
+	defer cancel()
+
+	shutdownCommand := exec.CommandContext(ctx, "wsl", "--user", "root", "--shutdown")
+	shutdownCommand.Stdout = ginkgo.GinkgoWriter
+	shutdownCommand.Stderr = ginkgo.GinkgoWriter
+
+	if err := shutdownCommand.Run(); err != nil {
+		fmt.Fprintf(ginkgo.GinkgoWriter, "err(wsl --user root --shutdown): %v", err)
+
+		// Failed due to shutdown command not returning in time. Gather extra information.
+		additionalCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+		defer cancel()
+
+		lsCommand := exec.CommandContext(additionalCtx, "wsl", "--user", "root", "--list", "--verbose")
+		lsCommand.Stdout = ginkgo.GinkgoWriter
+		lsCommand.Stderr = ginkgo.GinkgoWriter
+
+		if innerErr := lsCommand.Run(); innerErr != nil {
+			fmt.Fprintf(ginkgo.GinkgoWriter, "err(wsl --user root --list --verbose): %v", innerErr)
+		}
+
+		return err
+	}
+
+	return nil
+}