SupplicantP2pIfaceHal: Port over link to death handling

Port over the link to death handling from StaIface.

Bug: 33383725
Test: Compiles
Change-Id: I6a7c60ea759d638fe315b04ba9831b9bf90bd0a6

1 files changed, 4 insertions, 2 deletions

diff --git a/tests/wifitests/src/com/android/server/wifi/SupplicantP2pIfaceHalTest.java b/tests/wifitests/src/com/android/server/wifi/SupplicantP2pIfaceHalTest.java
index b08ebdf97..f19fb11de 100644
--- a/tests/wifitests/src/com/android/server/wifi/SupplicantP2pIfaceHalTest.java
+++ b/tests/wifitests/src/com/android/server/wifi/SupplicantP2pIfaceHalTest.java
@@ -37,10 +37,8 @@ import android.net.wifi.p2p.WifiP2pDevice;
 import android.net.wifi.p2p.WifiP2pGroup;
 import android.net.wifi.p2p.WifiP2pManager;
 import android.net.wifi.p2p.nsd.WifiP2pServiceInfo;
-import android.os.HandlerThread;
 import android.os.IHwBinder;
 import android.os.RemoteException;
-import android.os.test.TestLooper;
 
 import com.android.server.wifi.util.NativeUtil;
 
@@ -166,6 +164,10 @@ public class SupplicantP2pIfaceHalTest {
                 anyLong())).thenReturn(true);
         when(mServiceManagerMock.registerForNotifications(anyString(), anyString(),
                 any(IServiceNotification.Stub.class))).thenReturn(true);
+        when(mISupplicantMock.linkToDeath(any(IHwBinder.DeathRecipient.class),
+                anyLong())).thenReturn(true);
+        when(mISupplicantP2pIfaceMock.linkToDeath(any(IHwBinder.DeathRecipient.class),
+                anyLong())).thenReturn(true);
         mDut = new SupplicantP2pIfaceHalSpy();
     }