NetworkConnectivityShould.java

NetworkConnectivityShould.java

public class NetworkConnectivityShould {

  @Rule public ImmediateSchedulersTestRule rule = new ImmediateSchedulersTestRule();

  @Mock private ConnectivityManager mockConnectivityManager;
  @Mock private Context mockContext;

  private NetworkConnectivity networkConnectivity;

  @Before public void setUp() {
    MockitoAnnotations.initMocks(this);

    networkConnectivity = new NetworkConnectivity();

    when(mockContext.getSystemService(NetworkConnectivity.SYSTEM_SERVICE)).thenReturn(
        mockConnectivityManager);
  }

  @Test public void register_to_connectivity_network_changes() {
    networkConnectivity.addConnectivityCallback(mockContext,
        spy(NetworkConnectivity.ConnectivityCallback.class));

    verify(mockConnectivityManager).registerNetworkCallback(any(NetworkRequest.class),
        any(ConnectivityManager.NetworkCallback.class));
  }

  @Test public void unregister_to_connectivity_network_callback_on_complete() {
    NetworkConnectivity.ConnectivityCallback spyConnectivityCallback =
        spy(NetworkConnectivity.ConnectivityCallback.class);
    Network fakeNetwork = mock(Network.class);
    doAnswer(invocation -> {
      ((ConnectivityManager.NetworkCallback) invocation.getArguments()[1]).onAvailable(fakeNetwork);
      return null;
    }).when(mockConnectivityManager)
        .registerNetworkCallback(any(NetworkRequest.class),
            any(ConnectivityManager.NetworkCallback.class));

    networkConnectivity.addConnectivityCallback(mockContext, spyConnectivityCallback);

    verify(mockConnectivityManager).unregisterNetworkCallback(
        any(ConnectivityManager.NetworkCallback.class));
  }

  @Test public void unregister_to_connectivity_network_on_methodcall() {
    NetworkConnectivity.ConnectivityCallback spyConnectivityCallback =
        spy(NetworkConnectivity.ConnectivityCallback.class);
    doAnswer(invocation -> {
      return null;
    }).when(mockConnectivityManager)
        .registerNetworkCallback(any(NetworkRequest.class),
            any(ConnectivityManager.NetworkCallback.class));

    networkConnectivity.addConnectivityCallback(mockContext, spyConnectivityCallback);

    networkConnectivity.removeConnectivityCallback();

    verify(mockConnectivityManager).unregisterNetworkCallback(
        any(ConnectivityManager.NetworkCallback.class));
  }

  @Test public void call_callback_on_network_connection() {
    NetworkConnectivity.ConnectivityCallback spyConnectivityCallback =
        spy(NetworkConnectivity.ConnectivityCallback.class);
    Network fakeNetwork = mock(Network.class);
    doAnswer(invocation -> {
      ((ConnectivityManager.NetworkCallback) invocation.getArguments()[1]).onAvailable(fakeNetwork);
      return null;
    }).when(mockConnectivityManager)
        .registerNetworkCallback(any(NetworkRequest.class),
            any(ConnectivityManager.NetworkCallback.class));

    networkConnectivity.addConnectivityCallback(mockContext, spyConnectivityCallback);

    verify(spyConnectivityCallback).onConnected();
  }

  @Test public void unregister_callback_on_error() {
    NetworkConnectivity.ConnectivityCallback spyConnectivityCallback =
        spy(NetworkConnectivity.ConnectivityCallback.class);
    networkConnectivity.addConnectivityCallback(mockContext, spyConnectivityCallback);

    spyConnectivityCallback.onError(any(Exception.class));

    verify(mockConnectivityManager).unregisterNetworkCallback(
        any(ConnectivityManager.NetworkCallback.class));
  }
}