summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/wifitests/src/com/android/server/wifi/nan/WifiNanStateManagerTest.java9
1 files changed, 5 insertions, 4 deletions
diff --git a/tests/wifitests/src/com/android/server/wifi/nan/WifiNanStateManagerTest.java b/tests/wifitests/src/com/android/server/wifi/nan/WifiNanStateManagerTest.java
index 9cc09170a..a3dd6dd80 100644
--- a/tests/wifitests/src/com/android/server/wifi/nan/WifiNanStateManagerTest.java
+++ b/tests/wifitests/src/com/android/server/wifi/nan/WifiNanStateManagerTest.java
@@ -421,6 +421,7 @@ public class WifiNanStateManagerTest {
final String peerSsi = "some peer ssi data";
final String peerMatchFilter = "filter binary array represented as string";
final String peerMsg = "some message from peer";
+ final int messageId = 6948;
SubscribeData.Builder dataBuilder = new SubscribeData.Builder();
dataBuilder.setServiceName(serviceName).setServiceSpecificInfo(ssi);
@@ -464,7 +465,7 @@ public class WifiNanStateManagerTest {
inOrder.verify(mockListener).onMessageReceived(requestorId, peerMsg.getBytes(),
peerMsg.length());
- mDut.sendMessage(uid, sessionId, requestorId, ssi.getBytes(), ssi.length());
+ mDut.sendMessage(uid, sessionId, requestorId, ssi.getBytes(), ssi.length(), messageId);
mMockLooper.dispatchAll();
inOrder.verify(mMockNative).sendMessage(transactionId.capture(), eq(subscribeId),
@@ -473,9 +474,9 @@ public class WifiNanStateManagerTest {
mDut.onMessageSendFail(transactionId.getValue(), reasonFail);
mMockLooper.dispatchAll();
- inOrder.verify(mockListener).onMessageSendFail(reasonFail);
+ inOrder.verify(mockListener).onMessageSendFail(messageId, reasonFail);
- mDut.sendMessage(uid, sessionId, requestorId, ssi.getBytes(), ssi.length());
+ mDut.sendMessage(uid, sessionId, requestorId, ssi.getBytes(), ssi.length(), messageId);
mMockLooper.dispatchAll();
inOrder.verify(mMockNative).sendMessage(transactionId.capture(), eq(subscribeId),
@@ -484,7 +485,7 @@ public class WifiNanStateManagerTest {
mDut.onMessageSendSuccess(transactionId.getValue());
mMockLooper.dispatchAll();
- inOrder.verify(mockListener).onMessageSendSuccess();
+ inOrder.verify(mockListener).onMessageSendSuccess(messageId);
verifyNoMoreInteractions(mockListener, mMockNative);
}