src/net/android/javatests/src/org/chromium/net/NetworkChangeNotifierTest.java

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 package org.chromium.net;
   6
   7 import android.content.Context;
   8 import android.content.Intent;
   9 import android.net.ConnectivityManager;
  10 import android.telephony.TelephonyManager;
  11 import android.test.InstrumentationTestCase;
  12 import android.test.UiThreadTest;
  13 import android.test.suitebuilder.annotation.MediumTest;
  14
  15 import org.chromium.base.ActivityStatus;
  16 import org.chromium.base.test.util.Feature;
  17
  18 public class NetworkChangeNotifierTest extends InstrumentationTestCase {
  19     /**
  20      * Listens for alerts fired by the NetworkChangeNotifier when network status changes.
  21      */
  22     private static class NetworkChangeNotifierTestObserver
  23             implements NetworkChangeNotifier.ConnectionTypeObserver {
  24         private boolean mReceivedNotification = false;
  25
  26         @Override
  27         public void onConnectionTypeChanged(int connectionType) {
  28             mReceivedNotification = true;
  29         }
  30
  31         public boolean hasReceivedNotification() {
  32             return mReceivedNotification;
  33         }
  34
  35         public void resetHasReceivedNotification() {
  36             mReceivedNotification = false;
  37         }
  38     }
  39
  40     /**
  41      * Mocks out calls to the ConnectivityManager.
  42      */
  43     class MockConnectivityManagerDelegate
  44             extends NetworkChangeNotifierAutoDetect.ConnectivityManagerDelegate {
  45         private boolean mActiveNetworkExists;
  46         private int mNetworkType;
  47         private int mNetworkSubtype;
  48
  49         @Override
  50         boolean activeNetworkExists() {
  51             return mActiveNetworkExists;
  52         }
  53
  54         @Override
  55         boolean isConnected() {
  56             return getNetworkType() != NetworkChangeNotifier.CONNECTION_NONE;
  57         }
  58
  59         void setActiveNetworkExists(boolean networkExists) {
  60             mActiveNetworkExists = networkExists;
  61         }
  62
  63         @Override
  64         int getNetworkType() {
  65             return mNetworkType;
  66         }
  67
  68         void setNetworkType(int networkType) {
  69             mNetworkType = networkType;
  70         }
  71
  72         @Override
  73         int getNetworkSubtype() {
  74             return mNetworkSubtype;
  75         }
  76
  77         void setNetworkSubtype(int networkSubtype) {
  78             mNetworkSubtype = networkSubtype;
  79         }
  80     }
  81
  82     /**
  83      * Tests that when Chrome gets an intent indicating a change in network connectivity, it sends a
  84      * notification to Java observers.
  85      */
  86     @UiThreadTest
  87     @MediumTest
  88     @Feature({"Android-AppBase"})
  89     public void testNetworkChangeNotifierJavaObservers() throws InterruptedException {
  90         // Create a new notifier that doesn't have a native-side counterpart.
  91         Context context = getInstrumentation().getTargetContext();
  92         NetworkChangeNotifier.resetInstanceForTests(context);
  93
  94         NetworkChangeNotifier.setAutoDetectConnectivityState(true);
  95         NetworkChangeNotifierAutoDetect receiver = NetworkChangeNotifier.getAutoDetectorForTest();
  96         assertTrue(receiver != null);
  97
  98         MockConnectivityManagerDelegate connectivityDelegate =
  99                 new MockConnectivityManagerDelegate();
 100         connectivityDelegate.setActiveNetworkExists(true);
 101         connectivityDelegate.setNetworkType(NetworkChangeNotifier.CONNECTION_UNKNOWN);
 102         connectivityDelegate.setNetworkSubtype(TelephonyManager.NETWORK_TYPE_UNKNOWN);
 103         receiver.setConnectivityManagerDelegateForTests(connectivityDelegate);
 104
 105         // Initialize the NetworkChangeNotifier with a connection.
 106         Intent connectivityIntent = new Intent(ConnectivityManager.CONNECTIVITY_ACTION);
 107         receiver.onReceive(getInstrumentation().getTargetContext(), connectivityIntent);
 108
 109         // We shouldn't be re-notified if the connection hasn't actually changed.
 110         NetworkChangeNotifierTestObserver observer = new NetworkChangeNotifierTestObserver();
 111         NetworkChangeNotifier.addConnectionTypeObserver(observer);
 112         receiver.onReceive(getInstrumentation().getTargetContext(), connectivityIntent);
 113         assertFalse(observer.hasReceivedNotification());
 114
 115         // Mimic that connectivity has been lost and ensure that Chrome notifies our observer.
 116         connectivityDelegate.setActiveNetworkExists(false);
 117         connectivityDelegate.setNetworkType(NetworkChangeNotifier.CONNECTION_NONE);
 118         Intent noConnectivityIntent = new Intent(ConnectivityManager.CONNECTIVITY_ACTION);
 119         receiver.onReceive(getInstrumentation().getTargetContext(), noConnectivityIntent);
 120         assertTrue(observer.hasReceivedNotification());
 121
 122         observer.resetHasReceivedNotification();
 123         // Pretend we got moved to the background.
 124         receiver.onActivityStateChange(ActivityStatus.PAUSED);
 125         // Change the state.
 126         connectivityDelegate.setActiveNetworkExists(true);
 127         connectivityDelegate.setNetworkType(NetworkChangeNotifier.CONNECTION_WIFI);
 128         // The NetworkChangeNotifierAutoDetect doesn't receive any notification while we are in the
 129         // background, but when we get back to the foreground the state changed should be detected
 130         // and a notification sent.
 131         receiver.onActivityStateChange(ActivityStatus.RESUMED);
 132         assertTrue(observer.hasReceivedNotification());
 133     }
 134 }