Merge "services.net: Use the new ModuleNetworkStackClient"
This commit is contained in:
@@ -35,6 +35,7 @@ import android.net.ipmemorystore.IOnStatusListener;
|
|||||||
import android.net.ipmemorystore.NetworkAttributes;
|
import android.net.ipmemorystore.NetworkAttributes;
|
||||||
import android.net.ipmemorystore.NetworkAttributesParcelable;
|
import android.net.ipmemorystore.NetworkAttributesParcelable;
|
||||||
import android.net.ipmemorystore.Status;
|
import android.net.ipmemorystore.Status;
|
||||||
|
import android.net.networkstack.ModuleNetworkStackClient;
|
||||||
import android.os.RemoteException;
|
import android.os.RemoteException;
|
||||||
|
|
||||||
import androidx.test.filters.SmallTest;
|
import androidx.test.filters.SmallTest;
|
||||||
@@ -67,7 +68,7 @@ public class IpMemoryStoreTest {
|
|||||||
@Mock
|
@Mock
|
||||||
Context mMockContext;
|
Context mMockContext;
|
||||||
@Mock
|
@Mock
|
||||||
NetworkStackClient mNetworkStackClient;
|
ModuleNetworkStackClient mModuleNetworkStackClient;
|
||||||
@Mock
|
@Mock
|
||||||
IIpMemoryStore mMockService;
|
IIpMemoryStore mMockService;
|
||||||
@Mock
|
@Mock
|
||||||
@@ -90,14 +91,14 @@ public class IpMemoryStoreTest {
|
|||||||
((IIpMemoryStoreCallbacks) invocation.getArgument(0))
|
((IIpMemoryStoreCallbacks) invocation.getArgument(0))
|
||||||
.onIpMemoryStoreFetched(mMockService);
|
.onIpMemoryStoreFetched(mMockService);
|
||||||
return null;
|
return null;
|
||||||
}).when(mNetworkStackClient).fetchIpMemoryStore(any());
|
}).when(mModuleNetworkStackClient).fetchIpMemoryStore(any());
|
||||||
} else {
|
} else {
|
||||||
doNothing().when(mNetworkStackClient).fetchIpMemoryStore(mCbCaptor.capture());
|
doNothing().when(mModuleNetworkStackClient).fetchIpMemoryStore(mCbCaptor.capture());
|
||||||
}
|
}
|
||||||
mStore = new IpMemoryStore(mMockContext) {
|
mStore = new IpMemoryStore(mMockContext) {
|
||||||
@Override
|
@Override
|
||||||
protected NetworkStackClient getNetworkStackClient() {
|
protected ModuleNetworkStackClient getModuleNetworkStackClient(Context ctx) {
|
||||||
return mNetworkStackClient;
|
return mModuleNetworkStackClient;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user