org.robolectric.shadows.ShadowWifiScanner Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of shadows-framework Show documentation
An alternative Android testing framework.
The newest version!
package org.robolectric.shadows;

import static com.google.common.util.concurrent.MoreExecutors.directExecutor;
import static org.robolectric.util.reflector.Reflector.reflector;

import android.net.wifi.IWifiScannerListener;
import android.net.wifi.ScanResult;
import android.net.wifi.WifiScanner;
import android.net.wifi.WifiScanner.ScanData;
import android.net.wifi.WifiScanner.ScanListener;
import android.os.Build.VERSION;
import android.os.Build.VERSION_CODES;
import android.os.RemoteException;
import android.util.SparseArray;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.Executor;
import org.robolectric.annotation.Implementation;
import org.robolectric.annotation.Implements;
import org.robolectric.annotation.RealObject;
import org.robolectric.util.reflector.Accessor;
import org.robolectric.util.reflector.ForType;

/** Shadow for {@link android.net.wifi.WifiScanner}. */
@Implements(value = WifiScanner.class, isInAndroidSdk = false)
public class ShadowWifiScanner {
  @RealObject protected WifiScanner realWifiScanner;

  private List scanResults = new ArrayList<>();

  /**
   * This method sets the scanResults that are immediately passed to the listeners, or returned by
   * getSingleScanResults().
   *
   * All registered listeners are notified immediately after this is called.
   */
  public void setScanResults(List scanResults) {
    this.scanResults = scanResults;
    if (VERSION.SDK_INT >= VERSION_CODES.UPSIDE_DOWN_CAKE) {
      notifyScanListenersU();
    } else {
      notifyScanListeners();
    }
  }

  @Implementation(minSdk = VERSION_CODES.R)
  protected List getSingleScanResults() {
    return scanResults;
  }

  private void notifyScanListenersU() {
    Object listenerMapLock =
        reflector(WifiScannerReflector.class, realWifiScanner).getListenerMapLock();

    Map listenerMap =
        reflector(WifiScannerReflector.class, realWifiScanner).getListenerMap();

    List listeners = null;
    synchronized (listenerMapLock) {
      listeners = new ArrayList<>(listenerMap.values());
    }

    for (IWifiScannerListener.Stub listener : listeners) {
      try {
        listener.onResults(buildScanData());
      } catch (RemoteException e) {
        throw new RuntimeException("Failed to notify scan listeners", e);
      }
    }
  }

  private void notifyScanListeners() {
    Object listenerMapLock =
        reflector(WifiScannerReflector.class, realWifiScanner).getListenerMapLock();
    SparseArray