Merge "Add Tests for PackagePolicy"

This commit is contained in:
Pete Varvarezis
2023-02-06 17:12:24 +00:00
committed by Android (Google) Code Review

View File

@@ -0,0 +1,181 @@
/*
* Copyright (C) 2023 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package android.app.admin;
import static android.app.admin.PackagePolicy.PACKAGE_POLICY_ALLOWLIST;
import static android.app.admin.PackagePolicy.PACKAGE_POLICY_ALLOWLIST_AND_SYSTEM;
import static android.app.admin.PackagePolicy.PACKAGE_POLICY_BLOCKLIST;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import android.os.Parcel;
import android.platform.test.annotations.Presubmit;
import android.util.ArraySet;
import androidx.test.filters.SmallTest;
import androidx.test.runner.AndroidJUnit4;
import org.junit.Test;
import org.junit.runner.RunWith;
import java.util.Collections;
import java.util.Set;
@RunWith(AndroidJUnit4.class)
@SmallTest
@Presubmit
public class PackagePolicyTest {
private static final String TEST_PACKAGE_NAME = "com.example";
private static final String TEST_PACKAGE_NAME_2 = "com.example.2";
private static final String TEST_SYSTEM_PACKAGE_NAME = "com.system";
@Test
public void testParceling() {
final int policyType = PACKAGE_POLICY_BLOCKLIST;
final Set<String> packageNames = new ArraySet<>();
packageNames.add(TEST_PACKAGE_NAME);
final Parcel parcel = Parcel.obtain();
PackagePolicy packagePolicy = new PackagePolicy(policyType, packageNames);
try {
packagePolicy.writeToParcel(parcel, 0);
parcel.setDataPosition(0);
packagePolicy = PackagePolicy.CREATOR.createFromParcel(parcel);
} finally {
parcel.recycle();
}
assertEquals(policyType, packagePolicy.getPolicyType());
assertNotNull(packagePolicy.getPackageNames());
assertEquals(1, packagePolicy.getPackageNames().size());
assertTrue(packagePolicy.getPackageNames().contains(TEST_PACKAGE_NAME));
}
@Test
public void testEmptyBlocklistCreation() {
PackagePolicy policy = new PackagePolicy(PACKAGE_POLICY_BLOCKLIST);
assertEquals(PACKAGE_POLICY_BLOCKLIST, policy.getPolicyType());
assertNotNull(policy.getPackageNames());
assertTrue(policy.getPackageNames().isEmpty());
}
@Test
public void testEmptyAllowlistCreation() {
PackagePolicy policy = new PackagePolicy(PACKAGE_POLICY_ALLOWLIST);
assertEquals(PACKAGE_POLICY_ALLOWLIST, policy.getPolicyType());
assertNotNull(policy.getPackageNames());
assertTrue(policy.getPackageNames().isEmpty());
}
@Test
public void testEmptyAllowlistAndSystemCreation() {
PackagePolicy policy = new PackagePolicy(PACKAGE_POLICY_ALLOWLIST_AND_SYSTEM);
assertEquals(PACKAGE_POLICY_ALLOWLIST_AND_SYSTEM, policy.getPolicyType());
assertNotNull(policy.getPackageNames());
assertTrue(policy.getPackageNames().isEmpty());
}
@Test
public void testSuppliedBlocklistCreation() {
final Set<String> packageNames = new ArraySet<>();
packageNames.add(TEST_PACKAGE_NAME);
PackagePolicy policy = new PackagePolicy(PACKAGE_POLICY_BLOCKLIST, packageNames);
assertEquals(PACKAGE_POLICY_BLOCKLIST, policy.getPolicyType());
assertNotNull(policy.getPackageNames());
assertEquals(1, policy.getPackageNames().size());
assertTrue(policy.getPackageNames().contains(TEST_PACKAGE_NAME));
}
@Test
public void testSuppliedAllowlistCreation() {
final Set<String> packageNames = new ArraySet<>();
packageNames.add(TEST_PACKAGE_NAME);
PackagePolicy policy = new PackagePolicy(PACKAGE_POLICY_ALLOWLIST, packageNames);
assertEquals(PACKAGE_POLICY_ALLOWLIST, policy.getPolicyType());
assertNotNull(policy.getPackageNames());
assertEquals(1, policy.getPackageNames().size());
assertTrue(policy.getPackageNames().contains(TEST_PACKAGE_NAME));
}
@Test
public void testSuppliedAllowlistAndSystemCreation() {
final Set<String> packageNames = new ArraySet<>();
packageNames.add(TEST_PACKAGE_NAME);
PackagePolicy policy = new PackagePolicy(PACKAGE_POLICY_ALLOWLIST_AND_SYSTEM, packageNames);
assertEquals(PACKAGE_POLICY_ALLOWLIST_AND_SYSTEM, policy.getPolicyType());
assertNotNull(policy.getPackageNames());
assertEquals(1, policy.getPackageNames().size());
assertTrue(policy.getPackageNames().contains(TEST_PACKAGE_NAME));
}
@Test
public void testBlocklist_isPackageAllowed() {
final Set<String> packageNames = new ArraySet<>();
packageNames.add(TEST_PACKAGE_NAME);
final Set<String> systemPackages = new ArraySet<>();
systemPackages.add(TEST_SYSTEM_PACKAGE_NAME);
PackagePolicy policy = new PackagePolicy(PACKAGE_POLICY_BLOCKLIST, packageNames);
assertFalse(policy.isPackageAllowed(TEST_PACKAGE_NAME, Collections.emptySet()));
assertFalse(policy.isPackageAllowed(TEST_PACKAGE_NAME, systemPackages));
assertTrue(policy.isPackageAllowed(TEST_PACKAGE_NAME_2, Collections.emptySet()));
assertTrue(policy.isPackageAllowed(TEST_PACKAGE_NAME_2, systemPackages));
assertTrue(policy.isPackageAllowed(TEST_SYSTEM_PACKAGE_NAME, Collections.emptySet()));
assertTrue(policy.isPackageAllowed(TEST_SYSTEM_PACKAGE_NAME, systemPackages));
}
@Test
public void testAllowlist_isPackageAllowed() {
final Set<String> packageNames = new ArraySet<>();
packageNames.add(TEST_PACKAGE_NAME);
final Set<String> systemPackages = new ArraySet<>();
systemPackages.add(TEST_SYSTEM_PACKAGE_NAME);
PackagePolicy policy = new PackagePolicy(PACKAGE_POLICY_ALLOWLIST, packageNames);
assertTrue(policy.isPackageAllowed(TEST_PACKAGE_NAME, Collections.emptySet()));
assertTrue(policy.isPackageAllowed(TEST_PACKAGE_NAME, systemPackages));
assertFalse(policy.isPackageAllowed(TEST_PACKAGE_NAME_2, Collections.emptySet()));
assertFalse(policy.isPackageAllowed(TEST_PACKAGE_NAME_2, systemPackages));
assertFalse(policy.isPackageAllowed(TEST_SYSTEM_PACKAGE_NAME, Collections.emptySet()));
assertFalse(policy.isPackageAllowed(TEST_SYSTEM_PACKAGE_NAME, systemPackages));
}
@Test
public void testAllowlistAndSystem_isPackageAllowed() {
final Set<String> packageNames = new ArraySet<>();
packageNames.add(TEST_PACKAGE_NAME);
final Set<String> systemPackages = new ArraySet<>();
systemPackages.add(TEST_SYSTEM_PACKAGE_NAME);
PackagePolicy policy = new PackagePolicy(PACKAGE_POLICY_ALLOWLIST_AND_SYSTEM, packageNames);
assertTrue(policy.isPackageAllowed(TEST_PACKAGE_NAME, Collections.emptySet()));
assertTrue(policy.isPackageAllowed(TEST_PACKAGE_NAME, systemPackages));
assertFalse(policy.isPackageAllowed(TEST_PACKAGE_NAME_2, Collections.emptySet()));
assertFalse(policy.isPackageAllowed(TEST_PACKAGE_NAME_2, systemPackages));
assertFalse(policy.isPackageAllowed(TEST_SYSTEM_PACKAGE_NAME, Collections.emptySet()));
assertTrue(policy.isPackageAllowed(TEST_SYSTEM_PACKAGE_NAME, systemPackages));
}
}