Adding ability to clear list of sent VirtualSensorEvents from ShadowV…

…irtualSensor

Within a single test suite, if you send VirtualSensorEvents to a ShadowVirtualSensor in testA then call ShadowVirtualSensor.getSentEvents() in testB, then you will see stale data leak from the last test. This new method will help clear stale stale between tests.

PiperOrigin-RevId: 621990558

robolectric/src/test/java/org/robolectric/shadows/ShadowVirtualSensorTest.java

@@ -0,0 +1,59 @@
+package org.robolectric.shadows;
+
+import static android.os.Build.VERSION_CODES.UPSIDE_DOWN_CAKE;
+import static com.google.common.truth.Truth.assertThat;
+
+import android.companion.virtual.sensor.VirtualSensor;
+import android.companion.virtual.sensor.VirtualSensorEvent;
+import android.hardware.Sensor;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.robolectric.RobolectricTestRunner;
+import org.robolectric.annotation.Config;
+import org.robolectric.shadow.api.Shadow;
+
+/** Unit test for ShadowVirtualSensor. */
+@Config(minSdk = UPSIDE_DOWN_CAKE)
+@RunWith(RobolectricTestRunner.class)
+public class ShadowVirtualSensorTest {
+
+  @Test
+  public void virtualSensor_getDeviceId() {
+    VirtualSensor sensor = (VirtualSensor) createVirtualSensor();
+    ((ShadowVirtualSensor) Shadow.extract(sensor)).setDeviceId(121);
+    assertThat(sensor.getDeviceId()).isEqualTo(121);
+  }
+
+  @Test
+  public void virtualSensor_getEvents() {
+    VirtualSensor sensor = (VirtualSensor) createVirtualSensor();
+    VirtualSensorEvent event =
+        new VirtualSensorEvent.Builder(new float[] {12f})
+            .setTimestampNanos(System.nanoTime())
+            .build();
+
+    sensor.sendEvent(event);
+
+    assertThat(((ShadowVirtualSensor) Shadow.extract(sensor)).getSentEvents())
+        .containsExactly(event);
+  }
+
+  @Test
+  public void virtualSensor_clearEvents() {
+    VirtualSensor sensor = (VirtualSensor) createVirtualSensor();
+    ShadowVirtualSensor virtualSensor = (ShadowVirtualSensor) Shadow.extract(sensor);
+    VirtualSensorEvent event =
+        new VirtualSensorEvent.Builder(new float[] {12f})
+            .setTimestampNanos(System.nanoTime())
+            .build();
+
+    sensor.sendEvent(event);
+    assertThat(virtualSensor.getSentEvents()).containsExactly(event);
+    virtualSensor.clearSentEvents();
+    assertThat(virtualSensor.getSentEvents()).isEmpty();
+  }
+
+  private /* VirtualSensor */ Object createVirtualSensor() {
+    return new VirtualSensor(1, Sensor.TYPE_ACCELEROMETER, "some name", null, null);
+  }
+}

shadows/framework/src/main/java/org/robolectric/shadows/ShadowVirtualSensor.java

@@ -2,6 +2,7 @@
 
 import android.companion.virtual.sensor.VirtualSensor;
 import android.companion.virtual.sensor.VirtualSensorEvent;
+import com.google.common.collect.ImmutableList;
 import java.util.ArrayList;
 import java.util.List;
 import org.robolectric.annotation.Implementation;
@@ -30,7 +31,11 @@ protected void sendEvent(VirtualSensorEvent event) {
   }
 
   public List<VirtualSensorEvent> getSentEvents() {
-    return sentEvents;
+    return ImmutableList.copyOf(sentEvents);
+  }
+
+  public void clearSentEvents() {
+    sentEvents.clear();
   }
 
   void setDeviceId(int deviceId) {