v1/src/main/java/com/xtremelabs/robolectric/shadows/ShadowConnectivityManager.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

package com.xtremelabs.robolectric.shadows;

import android.net.ConnectivityManager;
import android.net.NetworkInfo;
import com.xtremelabs.robolectric.internal.Implementation;
import com.xtremelabs.robolectric.internal.Implements;

import java.util.HashMap;
import java.util.Map;

/**
 * Shadow of {@code ConnectivityManager} that provides for the simulation of
 * the active connection status.
 */

@Implements(ConnectivityManager.class)
public class ShadowConnectivityManager {

    private NetworkInfo activeNetwork;
    private boolean backgroundDataSetting;

    private Map<Integer, NetworkInfo> networkTypeToNetworkInfo = new HashMap<Integer, NetworkInfo>();

    public void __constructor__() {
        setActiveNetworkInfo(ShadowNetworkInfo.newInstance());
    }

    @Implementation
    public NetworkInfo getActiveNetworkInfo() {
        return activeNetwork;
    }

    @Implementation
    public NetworkInfo[] getAllNetworkInfo() {
        return networkTypeToNetworkInfo.values().toArray(new NetworkInfo[networkTypeToNetworkInfo.size()]);
    }

    @Implementation
    public NetworkInfo getNetworkInfo(int networkType) {
        return networkTypeToNetworkInfo.get(networkType);
    }

    @Implementation
    public boolean getBackgroundDataSetting() {
        return backgroundDataSetting;
    }

    public void setNetworkInfo(int networkType, NetworkInfo networkInfo) {
        networkTypeToNetworkInfo.put(networkType, networkInfo);
    }

    public void setBackgroundDataSetting(boolean b) {
        backgroundDataSetting = b;
    }

    public void setActiveNetworkInfo(NetworkInfo info) {
        activeNetwork = info;
        if (info != null) {
            networkTypeToNetworkInfo.put(info.getType(), info);
        }  else {
            networkTypeToNetworkInfo.clear();
        }
    }
}