From 3fbea91a035c97fafaf9ee17366d47fa3a9f39d2 Mon Sep 17 00:00:00 2001
From: Wei Wang <wvw@google.com>
Date: Fri, 31 Jan 2020 12:40:56 -0800
Subject: PowerManagerTest: explicit reset mocked listener2

Bug: 147899604
Test: atest PowerManagerTest.java
Change-Id: I7d4510b99fa6696ea1c3e081b44fa9a51e4307dd
---
 core/tests/coretests/src/android/os/PowerManagerTest.java | 3 +++
 1 file changed, 3 insertions(+)

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));
-- 
cgit v1.2.3-59-g8ed1b