PowerManagerTest: explicit reset mocked listener2

Bug: 147899604
Test: atest PowerManagerTest.java
Change-Id: I7d4510b99fa6696ea1c3e081b44fa9a51e4307dd

1 files changed, 3 insertions, 0 deletions

diff --git a/core/tests/coretests/src/android/os/PowerManagerTest.java b/core/tests/coretests/src/android/os/PowerManagerTest.java
index 5cb7852f7acc..ea0a0fd49f9f 100644
--- a/core/tests/coretests/src/android/os/PowerManagerTest.java
+++ b/core/tests/coretests/src/android/os/PowerManagerTest.java
@@ -214,6 +214,9 @@ public class PowerManagerTest extends AndroidTestCase {
         }
         // Add listener2 on main thread.
         mPm.addThermalStatusListener(mListener2);
+        verify(mListener2, timeout(CALLBACK_TIMEOUT_MILLI_SEC)
+            .times(1)).onThermalStatusChanged(status);
+        reset(mListener2);
         status = PowerManager.THERMAL_STATUS_MODERATE;
         mUiDevice.executeShellCommand("cmd thermalservice override-status "
                 + Integer.toString(status));