Merge "[CS] Unregister callback as part of onUnavailable dispatch" into qt-dev

This commit is contained in:
Lorenzo Colitti
2019-05-09 09:42:38 +00:00
committed by Android (Google) Code Review
2 changed files with 11 additions and 2 deletions

View File

@@ -3449,6 +3449,10 @@ public class ConnectivityManager {
final NetworkCallback callback; final NetworkCallback callback;
synchronized (sCallbacks) { synchronized (sCallbacks) {
callback = sCallbacks.get(request); callback = sCallbacks.get(request);
if (message.what == CALLBACK_UNAVAIL) {
sCallbacks.remove(request);
callback.networkRequest = ALREADY_UNREGISTERED;
}
} }
if (DBG) { if (DBG) {
Log.d(TAG, getCallbackName(message.what) + " for network " + network); Log.d(TAG, getCallbackName(message.what) + " for network " + network);
@@ -3995,8 +3999,10 @@ public class ConnectivityManager {
synchronized (sCallbacks) { synchronized (sCallbacks) {
Preconditions.checkArgument(networkCallback.networkRequest != null, Preconditions.checkArgument(networkCallback.networkRequest != null,
"NetworkCallback was not registered"); "NetworkCallback was not registered");
Preconditions.checkArgument(networkCallback.networkRequest != ALREADY_UNREGISTERED, if (networkCallback.networkRequest == ALREADY_UNREGISTERED) {
"NetworkCallback was already unregistered"); Log.d(TAG, "NetworkCallback was already unregistered");
return;
}
for (Map.Entry<NetworkRequest, NetworkCallback> e : sCallbacks.entrySet()) { for (Map.Entry<NetworkRequest, NetworkCallback> e : sCallbacks.entrySet()) {
if (e.getValue() == networkCallback) { if (e.getValue() == networkCallback) {
reqs.add(e.getKey()); reqs.add(e.getKey());

View File

@@ -3854,6 +3854,9 @@ public class ConnectivityServiceTest {
networkCallback.expectCallback(CallbackState.UNAVAILABLE, null); networkCallback.expectCallback(CallbackState.UNAVAILABLE, null);
testFactory.waitForRequests(); testFactory.waitForRequests();
// unregister network callback - a no-op, but should not fail
mCm.unregisterNetworkCallback(networkCallback);
testFactory.unregister(); testFactory.unregister();
handlerThread.quit(); handlerThread.quit();
} }