Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2014 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License |
| 15 | */ |
| 16 | |
| 17 | package android.net; |
| 18 | |
| 19 | import android.Manifest.permission; |
| 20 | import android.content.Context; |
| 21 | import android.content.Intent; |
| 22 | import android.content.pm.ActivityInfo; |
Jeff Davidson | c741553 | 2014-06-23 18:15:34 -0700 | [diff] [blame] | 23 | import android.content.pm.ApplicationInfo; |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 24 | import android.content.pm.PackageManager; |
| 25 | import android.content.pm.ResolveInfo; |
Jeremy Joslin | dd251ef | 2016-03-14 11:17:41 -0700 | [diff] [blame] | 26 | import android.content.pm.ServiceInfo; |
Jeff Davidson | c741553 | 2014-06-23 18:15:34 -0700 | [diff] [blame] | 27 | import android.net.NetworkScorerAppManager.NetworkScorerAppData; |
Jeremy Joslin | fe505af | 2016-02-10 07:34:46 -0800 | [diff] [blame] | 28 | import android.os.UserHandle; |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 29 | import android.test.InstrumentationTestCase; |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 30 | |
| 31 | import org.mockito.ArgumentMatcher; |
| 32 | import org.mockito.Mock; |
| 33 | import org.mockito.Mockito; |
| 34 | import org.mockito.MockitoAnnotations; |
| 35 | |
Jeff Davidson | b6646a8 | 2014-06-27 16:24:42 -0700 | [diff] [blame] | 36 | import java.util.ArrayList; |
| 37 | import java.util.Collections; |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 38 | import java.util.Iterator; |
Jeff Davidson | b6646a8 | 2014-06-27 16:24:42 -0700 | [diff] [blame] | 39 | import java.util.List; |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 40 | |
Jeff Davidson | 6a4b220 | 2014-04-16 17:29:40 -0700 | [diff] [blame] | 41 | public class NetworkScorerAppManagerTest extends InstrumentationTestCase { |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 42 | @Mock private Context mMockContext; |
| 43 | @Mock private PackageManager mMockPm; |
| 44 | |
Amin Shaikh | aa09aa0 | 2016-11-21 17:27:53 -0800 | [diff] [blame^] | 45 | private NetworkScorerAppManager mNetworkScorerAppManager; |
| 46 | |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 47 | @Override |
| 48 | public void setUp() throws Exception { |
| 49 | super.setUp(); |
| 50 | |
| 51 | // Configuration needed to make mockito/dexcache work. |
| 52 | System.setProperty("dexmaker.dexcache", |
| 53 | getInstrumentation().getTargetContext().getCacheDir().getPath()); |
| 54 | ClassLoader newClassLoader = getInstrumentation().getClass().getClassLoader(); |
| 55 | Thread.currentThread().setContextClassLoader(newClassLoader); |
| 56 | |
| 57 | MockitoAnnotations.initMocks(this); |
| 58 | Mockito.when(mMockContext.getPackageManager()).thenReturn(mMockPm); |
Amin Shaikh | aa09aa0 | 2016-11-21 17:27:53 -0800 | [diff] [blame^] | 59 | mNetworkScorerAppManager = new NetworkScorerAppManager(mMockContext); |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 60 | } |
| 61 | |
| 62 | public void testGetAllValidScorers() throws Exception { |
| 63 | // Package 1 - Valid scorer. |
Jeremy Joslin | dd251ef | 2016-03-14 11:17:41 -0700 | [diff] [blame] | 64 | ResolveInfoHolder package1 = buildResolveInfo("package1", 1, true, true, false, false); |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 65 | |
Jeff Davidson | 1619779 | 2014-11-03 17:39:54 -0800 | [diff] [blame] | 66 | // Package 2 - Receiver does not have BROADCAST_NETWORK_PRIVILEGED permission. |
Jeremy Joslin | dd251ef | 2016-03-14 11:17:41 -0700 | [diff] [blame] | 67 | ResolveInfoHolder package2 = buildResolveInfo("package2", 2, false, true, false, false); |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 68 | |
| 69 | // Package 3 - App does not have SCORE_NETWORKS permission. |
Jeremy Joslin | dd251ef | 2016-03-14 11:17:41 -0700 | [diff] [blame] | 70 | ResolveInfoHolder package3 = buildResolveInfo("package3", 3, true, false, false, false); |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 71 | |
Jeff Davidson | b6646a8 | 2014-06-27 16:24:42 -0700 | [diff] [blame] | 72 | // Package 4 - Valid scorer w/ optional config activity. |
Jeremy Joslin | dd251ef | 2016-03-14 11:17:41 -0700 | [diff] [blame] | 73 | ResolveInfoHolder package4 = buildResolveInfo("package4", 4, true, true, true, false); |
Jeff Davidson | b6646a8 | 2014-06-27 16:24:42 -0700 | [diff] [blame] | 74 | |
Jeremy Joslin | dd251ef | 2016-03-14 11:17:41 -0700 | [diff] [blame] | 75 | // Package 5 - Valid scorer w/ optional service to bind to. |
| 76 | ResolveInfoHolder package5 = buildResolveInfo("package5", 5, true, true, false, true); |
| 77 | |
| 78 | List<ResolveInfoHolder> scorers = new ArrayList<>(); |
Jeff Davidson | b6646a8 | 2014-06-27 16:24:42 -0700 | [diff] [blame] | 79 | scorers.add(package1); |
| 80 | scorers.add(package2); |
| 81 | scorers.add(package3); |
| 82 | scorers.add(package4); |
Jeremy Joslin | dd251ef | 2016-03-14 11:17:41 -0700 | [diff] [blame] | 83 | scorers.add(package5); |
Jeff Davidson | b6646a8 | 2014-06-27 16:24:42 -0700 | [diff] [blame] | 84 | setScorers(scorers); |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 85 | |
Jeff Davidson | c741553 | 2014-06-23 18:15:34 -0700 | [diff] [blame] | 86 | Iterator<NetworkScorerAppData> result = |
Amin Shaikh | aa09aa0 | 2016-11-21 17:27:53 -0800 | [diff] [blame^] | 87 | mNetworkScorerAppManager.getAllValidScorers().iterator(); |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 88 | |
| 89 | assertTrue(result.hasNext()); |
Jeff Davidson | b6646a8 | 2014-06-27 16:24:42 -0700 | [diff] [blame] | 90 | NetworkScorerAppData next = result.next(); |
| 91 | assertEquals("package1", next.mPackageName); |
Shirish Kalele | 4cab12d | 2015-08-06 12:34:22 -0700 | [diff] [blame] | 92 | assertEquals(1, next.mPackageUid); |
Jeff Davidson | b6646a8 | 2014-06-27 16:24:42 -0700 | [diff] [blame] | 93 | assertNull(next.mConfigurationActivityClassName); |
| 94 | |
| 95 | assertTrue(result.hasNext()); |
| 96 | next = result.next(); |
| 97 | assertEquals("package4", next.mPackageName); |
Shirish Kalele | 4cab12d | 2015-08-06 12:34:22 -0700 | [diff] [blame] | 98 | assertEquals(4, next.mPackageUid); |
Jeff Davidson | b6646a8 | 2014-06-27 16:24:42 -0700 | [diff] [blame] | 99 | assertEquals(".ConfigActivity", next.mConfigurationActivityClassName); |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 100 | |
Jeremy Joslin | dd251ef | 2016-03-14 11:17:41 -0700 | [diff] [blame] | 101 | assertTrue(result.hasNext()); |
| 102 | next = result.next(); |
| 103 | assertEquals("package5", next.mPackageName); |
| 104 | assertEquals(5, next.mPackageUid); |
| 105 | assertEquals(".ScoringService", next.mScoringServiceClassName); |
| 106 | |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 107 | assertFalse(result.hasNext()); |
| 108 | } |
| 109 | |
Jeremy Joslin | dd251ef | 2016-03-14 11:17:41 -0700 | [diff] [blame] | 110 | private void setScorers(List<ResolveInfoHolder> scorers) { |
Jeff Davidson | b6646a8 | 2014-06-27 16:24:42 -0700 | [diff] [blame] | 111 | List<ResolveInfo> receivers = new ArrayList<>(); |
Jeremy Joslin | dd251ef | 2016-03-14 11:17:41 -0700 | [diff] [blame] | 112 | for (final ResolveInfoHolder scorer : scorers) { |
| 113 | receivers.add(scorer.scorerResolveInfo); |
| 114 | if (scorer.configActivityResolveInfo != null) { |
Jeff Davidson | b6646a8 | 2014-06-27 16:24:42 -0700 | [diff] [blame] | 115 | // This scorer has a config activity. |
| 116 | Mockito.when(mMockPm.queryIntentActivities( |
| 117 | Mockito.argThat(new ArgumentMatcher<Intent>() { |
| 118 | @Override |
| 119 | public boolean matches(Object object) { |
| 120 | Intent intent = (Intent) object; |
| 121 | return NetworkScoreManager.ACTION_CUSTOM_ENABLE.equals( |
| 122 | intent.getAction()) |
Jeremy Joslin | dd251ef | 2016-03-14 11:17:41 -0700 | [diff] [blame] | 123 | && scorer.scorerResolveInfo.activityInfo.packageName.equals( |
Jeff Davidson | b6646a8 | 2014-06-27 16:24:42 -0700 | [diff] [blame] | 124 | intent.getPackage()); |
| 125 | } |
Jeremy Joslin | dd251ef | 2016-03-14 11:17:41 -0700 | [diff] [blame] | 126 | }), Mockito.eq(0))).thenReturn( |
| 127 | Collections.singletonList(scorer.configActivityResolveInfo)); |
| 128 | } |
| 129 | |
| 130 | if (scorer.serviceResolveInfo != null) { |
| 131 | // This scorer has a service to bind to |
| 132 | Mockito.when(mMockPm.resolveService( |
| 133 | Mockito.argThat(new ArgumentMatcher<Intent>() { |
| 134 | @Override |
| 135 | public boolean matches(Object object) { |
| 136 | Intent intent = (Intent) object; |
| 137 | return NetworkScoreManager.ACTION_SCORE_NETWORKS.equals( |
| 138 | intent.getAction()) |
| 139 | && scorer.scorerResolveInfo.activityInfo.packageName.equals( |
| 140 | intent.getPackage()); |
| 141 | } |
| 142 | }), Mockito.eq(0))).thenReturn(scorer.serviceResolveInfo); |
Jeff Davidson | b6646a8 | 2014-06-27 16:24:42 -0700 | [diff] [blame] | 143 | } |
| 144 | } |
| 145 | |
Jeremy Joslin | fe505af | 2016-02-10 07:34:46 -0800 | [diff] [blame] | 146 | Mockito.when(mMockPm.queryBroadcastReceiversAsUser( |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 147 | Mockito.argThat(new ArgumentMatcher<Intent>() { |
| 148 | @Override |
| 149 | public boolean matches(Object object) { |
| 150 | Intent intent = (Intent) object; |
| 151 | return NetworkScoreManager.ACTION_SCORE_NETWORKS.equals(intent.getAction()); |
| 152 | } |
Jeremy Joslin | fe505af | 2016-02-10 07:34:46 -0800 | [diff] [blame] | 153 | }), Mockito.eq(0), Mockito.eq(UserHandle.USER_SYSTEM))) |
Jeff Davidson | b6646a8 | 2014-06-27 16:24:42 -0700 | [diff] [blame] | 154 | .thenReturn(receivers); |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 155 | } |
| 156 | |
Jeremy Joslin | dd251ef | 2016-03-14 11:17:41 -0700 | [diff] [blame] | 157 | private ResolveInfoHolder buildResolveInfo(String packageName, int packageUid, |
| 158 | boolean hasReceiverPermission, boolean hasScorePermission, boolean hasConfigActivity, |
| 159 | boolean hasServiceInfo) throws Exception { |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 160 | Mockito.when(mMockPm.checkPermission(permission.SCORE_NETWORKS, packageName)) |
| 161 | .thenReturn(hasScorePermission ? |
| 162 | PackageManager.PERMISSION_GRANTED : PackageManager.PERMISSION_DENIED); |
| 163 | |
| 164 | ResolveInfo resolveInfo = new ResolveInfo(); |
| 165 | resolveInfo.activityInfo = new ActivityInfo(); |
| 166 | resolveInfo.activityInfo.packageName = packageName; |
Jeff Davidson | c741553 | 2014-06-23 18:15:34 -0700 | [diff] [blame] | 167 | resolveInfo.activityInfo.applicationInfo = new ApplicationInfo(); |
Shirish Kalele | 4cab12d | 2015-08-06 12:34:22 -0700 | [diff] [blame] | 168 | resolveInfo.activityInfo.applicationInfo.uid = packageUid; |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 169 | if (hasReceiverPermission) { |
Jeff Davidson | 1619779 | 2014-11-03 17:39:54 -0800 | [diff] [blame] | 170 | resolveInfo.activityInfo.permission = permission.BROADCAST_NETWORK_PRIVILEGED; |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 171 | } |
Jeff Davidson | b6646a8 | 2014-06-27 16:24:42 -0700 | [diff] [blame] | 172 | |
| 173 | ResolveInfo configActivityInfo = null; |
| 174 | if (hasConfigActivity) { |
| 175 | configActivityInfo = new ResolveInfo(); |
| 176 | configActivityInfo.activityInfo = new ActivityInfo(); |
| 177 | configActivityInfo.activityInfo.name = ".ConfigActivity"; |
| 178 | } |
Jeremy Joslin | dd251ef | 2016-03-14 11:17:41 -0700 | [diff] [blame] | 179 | |
| 180 | ResolveInfo serviceInfo = null; |
| 181 | if (hasServiceInfo) { |
| 182 | serviceInfo = new ResolveInfo(); |
| 183 | serviceInfo.serviceInfo = new ServiceInfo(); |
| 184 | serviceInfo.serviceInfo.name = ".ScoringService"; |
| 185 | } |
| 186 | |
| 187 | return new ResolveInfoHolder(resolveInfo, configActivityInfo, serviceInfo); |
| 188 | } |
| 189 | |
| 190 | private static class ResolveInfoHolder { |
| 191 | final ResolveInfo scorerResolveInfo; |
| 192 | final ResolveInfo configActivityResolveInfo; |
| 193 | final ResolveInfo serviceResolveInfo; |
| 194 | |
| 195 | public ResolveInfoHolder(ResolveInfo scorerResolveInfo, |
| 196 | ResolveInfo configActivityResolveInfo, ResolveInfo serviceResolveInfo) { |
| 197 | this.scorerResolveInfo = scorerResolveInfo; |
| 198 | this.configActivityResolveInfo = configActivityResolveInfo; |
| 199 | this.serviceResolveInfo = serviceResolveInfo; |
| 200 | } |
Jeff Davidson | dd6fd1e | 2014-04-14 15:14:30 -0700 | [diff] [blame] | 201 | } |
| 202 | } |