Add e2e restart test

Author: SuperStrongDinosaur

e2etests/orchestration/restart_test/BUILD.bazel

@@ -0,0 +1,28 @@
+# Copyright (C) 2026 The Android Open Source Project
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+load("@rules_go//go:def.bzl", "go_test")
+
+go_test(
+    name = "restart_test_test",
+    srcs = ["main_test.go"],
+    data = [
+        "//orchestration/artifacts:cvd_host_package",
+        "//orchestration/artifacts:images_zip",
+    ],
+    deps = [
+        "//orchestration/common",
+        "@com_github_google_android_cuttlefish_frontend_src_libhoclient//:libhoclient",
+    ],
+)

e2etests/orchestration/restart_test/main_test.go

@@ -0,0 +1,83 @@
+// Copyright (C) 2026 The Android Open Source Project
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package main
+
+import (
+	"log"
+	"testing"
+
+	"github.com/google/android-cuttlefish/e2etests/orchestration/common"
+
+	hoclient "github.com/google/android-cuttlefish/frontend/src/libhoclient"
+)
+
+const baseURL = "http://0.0.0.0:2080"
+
+func TestRestart(t *testing.T) {
+	adbH := common.NewAdbHelper()
+	if err := adbH.StartServer(); err != nil {
+		t.Fatal(err)
+	}
+	srv := hoclient.NewHostOrchestratorClient(baseURL)
+	t.Cleanup(func() {
+		if err := common.CollectHOLogs(baseURL); err != nil {
+			log.Printf("failed to collect HO logs: %s", err)
+		}
+	})
+	imageDir, err := common.PrepareArtifact(srv, "../artifacts/images.zip")
+	if err != nil {
+		t.Fatal(err)
+	}
+	hostPkgDir, err := common.PrepareArtifact(srv, "../artifacts/cvd-host_package.tar.gz")
+	if err != nil {
+		t.Fatal(err)
+	}
+	cvd, err := common.CreateCVDFromImageDirs(srv, hostPkgDir, imageDir)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if err := adbH.Connect(cvd.ADBSerial); err != nil {
+		t.Fatal(err)
+	}
+	if err := adbH.WaitForDevice(cvd.ADBSerial); err != nil {
+		t.Fatal(err)
+	}
+
+	// Get boot time before restart
+	bootTimeBefore, err := adbH.ExecShellCommand(cvd.ADBSerial, []string{"uptime", "-s"})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// Restart the device
+	if err := srv.Restart(cvd.Group, cvd.Name); err != nil {
+		t.Fatal(err)
+	}
+
+	// Verify the device comes back online
+	if err := adbH.WaitForDevice(cvd.ADBSerial); err != nil {
+		t.Fatal(err)
+	}
+
+	// Get boot time after restart
+	bootTimeAfter, err := adbH.ExecShellCommand(cvd.ADBSerial, []string{"uptime", "-s"})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if bootTimeBefore == bootTimeAfter {
+		t.Fatalf("Device does not seem to have restarted. Boot time remained: %s", bootTimeBefore)
+	}
+}