Internal change

PiperOrigin-RevId: 753832573

Author: DeviceInfra

src/java/com/google/devtools/deviceinfra/ext/devicemanagement/device/platform/android/realdevice/AndroidRealDeviceDelegate.java

@@ -48,6 +48,7 @@
 import com.google.devtools.mobileharness.api.testrunner.device.cache.DeviceCache;
 import com.google.devtools.mobileharness.infra.container.sandbox.device.DeviceSandboxController;
 import com.google.devtools.mobileharness.infra.controller.device.config.ApiConfig;
+import com.google.devtools.mobileharness.platform.aluminium.device.AluminiumDeviceHelper;
 import com.google.devtools.mobileharness.platform.android.app.devicedaemon.DeviceDaemonApkInfoProvider;
 import com.google.devtools.mobileharness.platform.android.app.devicedaemon.DeviceDaemonHelper;
 import com.google.devtools.mobileharness.platform.android.app.mtaastools.MtaasToolsInstantiator;
@@ -167,6 +168,7 @@ public abstract class AndroidRealDeviceDelegate {
   private final MtaasToolsInstantiator mtaasToolsInstantiator;
 
   private final DeviceAdminUtil deviceAdminUtil;
+  private final AluminiumDeviceHelper aluminiumDeviceHelper;
 
   protected AndroidRealDeviceDelegate(
       AndroidDevice device,
@@ -192,7 +194,8 @@ protected AndroidRealDeviceDelegate(
       Fastboot fastboot,
       LocalFileUtil fileUtil,
       DeviceAdminUtil deviceAdminUtil,
-      MtaasToolsInstantiator mtaasToolsInstantiator) {
+      MtaasToolsInstantiator mtaasToolsInstantiator,
+      AluminiumDeviceHelper aluminiumDeviceHelper) {
     this.device = device;
     this.androidDeviceDelegate = androidDeviceDelegate;
     this.deviceStat = deviceStat;
@@ -217,6 +220,7 @@ protected AndroidRealDeviceDelegate(
     this.fileUtil = fileUtil;
     this.deviceAdminUtil = deviceAdminUtil;
     this.mtaasToolsInstantiator = mtaasToolsInstantiator;
+    this.aluminiumDeviceHelper = new AluminiumDeviceHelper();
 
     this.deviceId = device.getDeviceId();
     device.setProperty(
@@ -248,6 +252,20 @@ public void setUp() throws MobileHarnessException, InterruptedException {
             "Device is undetectable. Please replug the usb cable or reboot the device.");
       }
     }
+    if (aluminiumDeviceHelper != null
+        && aluminiumDeviceHelper.isAlAteHost()
+        && aluminiumDeviceHelper.isAluminiumDevice(device)) {
+      aluminiumDeviceHelper.updateSchedulingDimensions(device);
+    } else {
+      logger.atInfo().log("Not an AL ATE host or not an Aluminium device");
+      if (aluminiumDeviceHelper != null) {
+        logger.atInfo().log(
+            "isALAteHost: %s, isAluminiumDevice: %s",
+            aluminiumDeviceHelper.isAlAteHost(), aluminiumDeviceHelper.isAluminiumDevice(device));
+      } else {
+        logger.atInfo().log("Aluminium device helper is null");
+      }
+    }
 
     device.updateDimension(Dimension.Name.SUPPORTS_GMSCORE, Dimension.Value.TRUE);
     extrasInSetUp();
@@ -1019,6 +1037,20 @@ boolean checkOnlineModeDevice() throws MobileHarnessException, InterruptedExcept
     enforceSafeDischargeLevelIfNeeded();
 
     isDimensionChanged |= updateCheckinGroupStatus();
+    if (aluminiumDeviceHelper != null
+        && aluminiumDeviceHelper.isAlAteHost()
+        && aluminiumDeviceHelper.isAluminiumDevice(device)) {
+      return isDimensionChanged |= aluminiumDeviceHelper.updateSchedulingDimensions(device);
+    } else {
+      logger.atInfo().log("Not an AL ATE host or not an Aluminium device");
+      if (aluminiumDeviceHelper != null) {
+        logger.atInfo().log(
+            "isALAteHost: %s, isAluminiumDevice: %s",
+            aluminiumDeviceHelper.isAlAteHost(), aluminiumDeviceHelper.isAluminiumDevice(device));
+      } else {
+        logger.atInfo().log("Aluminium device helper is null");
+      }
+    }
 
     return isDimensionChanged;
   }

src/java/com/google/devtools/deviceinfra/ext/devicemanagement/device/platform/android/realdevice/BUILD

@@ -27,6 +27,7 @@ java_library(
     ],
     deps = [
         ":android_real_device_constants",
+        "//java/com/google/devtools/mobileharness/platform/aluminium/device:aluminium_device_helper",
         "//src/devtools/common/metrics/stability/model/proto:error_java_proto",
         "//src/devtools/mobileharness/api/deviceconfig/proto:basic_java_proto",
         "//src/devtools/mobileharness/api/model/proto:device_java_proto",