atrost | c3c2e4d | 2019-11-29 11:27:27 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2019 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 com.android.server.pm; |
| 18 | |
| 19 | import static org.hamcrest.MatcherAssert.assertThat; |
| 20 | import static org.hamcrest.core.Is.is; |
Winson | 655a5b9 | 2019-10-23 10:49:32 -0700 | [diff] [blame] | 21 | import static org.mockito.ArgumentMatchers.argThat; |
| 22 | import static org.mockito.ArgumentMatchers.eq; |
atrost | c3c2e4d | 2019-11-29 11:27:27 +0000 | [diff] [blame] | 23 | import static org.mockito.Mockito.when; |
| 24 | |
Winson | 655a5b9 | 2019-10-23 10:49:32 -0700 | [diff] [blame] | 25 | import android.os.Build; |
atrost | c3c2e4d | 2019-11-29 11:27:27 +0000 | [diff] [blame] | 26 | import android.platform.test.annotations.Presubmit; |
| 27 | |
| 28 | import com.android.server.compat.PlatformCompat; |
Winson | 33eacc6 | 2020-01-24 12:02:58 -0800 | [diff] [blame] | 29 | import com.android.server.pm.parsing.pkg.AndroidPackage; |
| 30 | import com.android.server.pm.parsing.pkg.PackageImpl; |
| 31 | import com.android.server.pm.parsing.pkg.ParsedPackage; |
atrost | c3c2e4d | 2019-11-29 11:27:27 +0000 | [diff] [blame] | 32 | |
atrost | c3c2e4d | 2019-11-29 11:27:27 +0000 | [diff] [blame] | 33 | import org.junit.Test; |
| 34 | import org.junit.runner.RunWith; |
| 35 | import org.mockito.Mock; |
| 36 | import org.mockito.junit.MockitoJUnitRunner; |
| 37 | |
| 38 | |
| 39 | /** |
| 40 | * {@link SELinuxMMAC} tests. |
| 41 | */ |
| 42 | @RunWith(MockitoJUnitRunner.class) |
| 43 | @Presubmit |
| 44 | public class SELinuxMMACTest { |
| 45 | |
| 46 | private static final String PACKAGE_NAME = "my.package"; |
Winson | 655a5b9 | 2019-10-23 10:49:32 -0700 | [diff] [blame] | 47 | private static final int OPT_IN_VERSION = Build.VERSION_CODES.R; |
atrost | c3c2e4d | 2019-11-29 11:27:27 +0000 | [diff] [blame] | 48 | |
| 49 | @Mock |
| 50 | PlatformCompat mMockCompatibility; |
| 51 | |
atrost | c3c2e4d | 2019-11-29 11:27:27 +0000 | [diff] [blame] | 52 | @Test |
| 53 | public void getSeInfoOptInToLatest() { |
Winson | 655a5b9 | 2019-10-23 10:49:32 -0700 | [diff] [blame] | 54 | AndroidPackage pkg = makePackage(Build.VERSION_CODES.P); |
Alex Buynytskyy | 6e3c563 | 2020-06-08 15:12:25 -0700 | [diff] [blame] | 55 | when(mMockCompatibility.isChangeEnabledInternal(eq(SELinuxMMAC.SELINUX_LATEST_CHANGES), |
Winson | 655a5b9 | 2019-10-23 10:49:32 -0700 | [diff] [blame] | 56 | argThat(argument -> argument.packageName.equals(pkg.getPackageName())))) |
| 57 | .thenReturn(true); |
| 58 | assertThat(SELinuxMMAC.getSeInfo(pkg, null, mMockCompatibility), |
atrost | c3c2e4d | 2019-11-29 11:27:27 +0000 | [diff] [blame] | 59 | is("default:targetSdkVersion=" + OPT_IN_VERSION)); |
| 60 | } |
| 61 | |
| 62 | @Test |
| 63 | public void getSeInfoNoOptIn() { |
Winson | 655a5b9 | 2019-10-23 10:49:32 -0700 | [diff] [blame] | 64 | AndroidPackage pkg = makePackage(Build.VERSION_CODES.P); |
Alex Buynytskyy | 6e3c563 | 2020-06-08 15:12:25 -0700 | [diff] [blame] | 65 | when(mMockCompatibility.isChangeEnabledInternal(eq(SELinuxMMAC.SELINUX_LATEST_CHANGES), |
Winson | 655a5b9 | 2019-10-23 10:49:32 -0700 | [diff] [blame] | 66 | argThat(argument -> argument.packageName.equals(pkg.getPackageName())))) |
| 67 | .thenReturn(false); |
| 68 | assertThat(SELinuxMMAC.getSeInfo(pkg, null, mMockCompatibility), |
atrost | c3c2e4d | 2019-11-29 11:27:27 +0000 | [diff] [blame] | 69 | is("default:targetSdkVersion=28")); |
| 70 | } |
| 71 | |
| 72 | @Test |
| 73 | public void getSeInfoNoOptInButAlreadyR() { |
Winson | 655a5b9 | 2019-10-23 10:49:32 -0700 | [diff] [blame] | 74 | AndroidPackage pkg = makePackage(OPT_IN_VERSION); |
Alex Buynytskyy | 6e3c563 | 2020-06-08 15:12:25 -0700 | [diff] [blame] | 75 | when(mMockCompatibility.isChangeEnabledInternal(eq(SELinuxMMAC.SELINUX_LATEST_CHANGES), |
Winson | 655a5b9 | 2019-10-23 10:49:32 -0700 | [diff] [blame] | 76 | argThat(argument -> argument.packageName.equals(pkg.getPackageName())))) |
| 77 | .thenReturn(false); |
| 78 | assertThat(SELinuxMMAC.getSeInfo(pkg, null, mMockCompatibility), |
atrost | c3c2e4d | 2019-11-29 11:27:27 +0000 | [diff] [blame] | 79 | is("default:targetSdkVersion=" + OPT_IN_VERSION)); |
| 80 | } |
Winson | 655a5b9 | 2019-10-23 10:49:32 -0700 | [diff] [blame] | 81 | |
| 82 | private AndroidPackage makePackage(int targetSdkVersion) { |
Winson | 33eacc6 | 2020-01-24 12:02:58 -0800 | [diff] [blame] | 83 | return ((ParsedPackage) PackageImpl.forTesting(PACKAGE_NAME) |
Winson | 655a5b9 | 2019-10-23 10:49:32 -0700 | [diff] [blame] | 84 | .setTargetSdkVersion(targetSdkVersion) |
Winson | 33eacc6 | 2020-01-24 12:02:58 -0800 | [diff] [blame] | 85 | .hideAsParsed()) |
Winson | 655a5b9 | 2019-10-23 10:49:32 -0700 | [diff] [blame] | 86 | .hideAsFinal(); |
| 87 | } |
atrost | c3c2e4d | 2019-11-29 11:27:27 +0000 | [diff] [blame] | 88 | } |