Make device mode enum value private

PiperOrigin-RevId: 547323151

Author: Paige McAuliffe

espresso/device/java/androidx/test/espresso/device/action/BaseSingleFoldDeviceAction.kt

@@ -52,7 +52,7 @@ internal open class BaseSingleFoldDeviceAction(
   @OptIn(androidx.window.core.ExperimentalWindowApi::class)
   override fun perform(context: ActionContext, deviceController: DeviceController) {
     if (isRobolectricTest()) {
-      deviceController.setDeviceMode(deviceMode.mode)
+      deviceController.setDeviceMode(deviceMode.getMode())
       return
     }
     val activity =
@@ -97,7 +97,7 @@ internal open class BaseSingleFoldDeviceAction(
       }
     }
 
-    deviceController.setDeviceMode(deviceMode.mode)
+    deviceController.setDeviceMode(deviceMode.getMode())
     latch.await(5, TimeUnit.SECONDS)
 
     if (latch.getCount() != 0L) {

espresso/device/java/androidx/test/espresso/device/controller/DeviceMode.kt

@@ -16,6 +16,8 @@
 
 package androidx.test.espresso.device.controller
 
+import androidx.annotation.RestrictTo
+import androidx.annotation.RestrictTo.Scope
 import androidx.test.annotation.ExperimentalTestApi
 
 /**
@@ -25,13 +27,20 @@ import androidx.test.annotation.ExperimentalTestApi
  * https://developer.android.com/guide/topics/large-screens/learn-about-foldables
  */
 @ExperimentalTestApi
-enum class DeviceMode(val mode: Int) {
+enum class DeviceMode(private val mode: Int) {
   // the device screen is completely flat, like a tablet
   FLAT(0),
   // the device is folded with the hinge in a horizontal position
   TABLETOP(1),
   // the device is folded with the hinge in a vertical position
   BOOK(2),
   // the device screen is closed
-  CLOSED(3)
+  CLOSED(3);
+
+  /**
+   * Returns the mode.
+   *
+   * @hide
+   */
+  @RestrictTo(Scope.LIBRARY) fun getMode(): Int = mode
 }

espresso/device/java/androidx/test/espresso/device/controller/emulator/EmulatorController.kt

@@ -48,20 +48,20 @@ constructor(
 
   override fun setDeviceMode(deviceMode: Int) {
     if (
-      !(deviceMode == DeviceMode.FLAT.mode ||
-        deviceMode == DeviceMode.TABLETOP.mode ||
-        deviceMode == DeviceMode.BOOK.mode ||
-        deviceMode == DeviceMode.CLOSED.mode)
+      !(deviceMode == DeviceMode.FLAT.getMode() ||
+        deviceMode == DeviceMode.TABLETOP.getMode() ||
+        deviceMode == DeviceMode.BOOK.getMode() ||
+        deviceMode == DeviceMode.CLOSED.getMode())
     ) {
       throw UnsupportedDeviceOperationException(
         "The provided device mode is not supported on this device."
       )
     }
 
     val postureValue: PostureValue =
-      if (deviceMode == DeviceMode.FLAT.mode) {
+      if (deviceMode == DeviceMode.FLAT.getMode()) {
         PostureValue.POSTURE_OPENED
-      } else if (deviceMode == DeviceMode.CLOSED.mode) {
+      } else if (deviceMode == DeviceMode.CLOSED.getMode()) {
         PostureValue.POSTURE_CLOSED
       } else {
         PostureValue.POSTURE_HALF_OPENED