/* * Copyright (C) 2024 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 com.android.ondevicepersonalization.services.serviceflow; import static org.junit.Assert.assertArrayEquals; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertNull; import static org.junit.Assert.assertFalse; import static org.junit.Assert.assertTrue; import static org.mockito.Mockito.doReturn; import static org.mockito.Mockito.spy; import static org.mockito.Mockito.when; import android.adservices.ondevicepersonalization.CalleeMetadata; import android.adservices.ondevicepersonalization.Constants; import android.adservices.ondevicepersonalization.aidl.IExecuteCallback; import android.content.ComponentName; import android.content.ContentValues; import android.content.Context; import android.os.Bundle; import android.os.PersistableBundle; import androidx.test.core.app.ApplicationProvider; import com.android.compatibility.common.util.ShellUtils; import com.android.dx.mockito.inline.extended.ExtendedMockito; import com.android.modules.utils.testing.ExtendedMockitoRule; import com.android.odp.module.common.PackageUtils; import com.android.ondevicepersonalization.internal.util.ByteArrayParceledSlice; import com.android.ondevicepersonalization.internal.util.PersistableBundleUtils; import com.android.ondevicepersonalization.services.Flags; import com.android.ondevicepersonalization.services.FlagsFactory; import com.android.ondevicepersonalization.services.PhFlagsTestUtil; import com.android.ondevicepersonalization.services.data.DbUtils; import com.android.ondevicepersonalization.services.data.OnDevicePersonalizationDbHelper; import com.android.ondevicepersonalization.services.data.events.EventsContract; import com.android.ondevicepersonalization.services.data.events.EventsDao; import com.android.ondevicepersonalization.services.data.events.QueriesContract; import com.android.ondevicepersonalization.services.data.events.Query; import com.android.ondevicepersonalization.services.data.user.UserPrivacyStatus; import com.android.ondevicepersonalization.services.data.vendor.OnDevicePersonalizationVendorDataDao; import com.android.ondevicepersonalization.services.data.vendor.VendorData; import com.android.ondevicepersonalization.services.util.OnDevicePersonalizationFlatbufferUtils; import org.junit.After; import org.junit.Before; import org.junit.Rule; import org.junit.Test; import org.junit.runner.RunWith; import org.junit.runners.JUnit4; import org.mockito.Mock; import org.mockito.Spy; import org.mockito.quality.Strictness; import java.util.ArrayList; import java.util.List; import java.util.concurrent.CountDownLatch; @RunWith(JUnit4.class) public class AppRequestFlowTest { private final Context mContext = spy(ApplicationProvider.getApplicationContext()); private final CountDownLatch mLatch = new CountDownLatch(1); private final OnDevicePersonalizationDbHelper mDbHelper = OnDevicePersonalizationDbHelper.getInstanceForTest(mContext); private boolean mCallbackSuccess; private boolean mCallbackError; private int mCallbackErrorCode; private int mIsolatedServiceErrorCode; private String mErrorMessage; private Bundle mExecuteCallback; private ServiceFlowOrchestrator mSfo; @Mock UserPrivacyStatus mUserPrivacyStatus; @Spy private Flags mSpyFlags = spy(FlagsFactory.getFlags()); @Rule public final ExtendedMockitoRule mExtendedMockitoRule = new ExtendedMockitoRule.Builder(this) .mockStatic(FlagsFactory.class) .spyStatic(UserPrivacyStatus.class) .setStrictness(Strictness.LENIENT) .build(); @Before public void setup() throws Exception { PhFlagsTestUtil.setUpDeviceConfigPermissions(); ExtendedMockito.doReturn(mSpyFlags).when(FlagsFactory::getFlags); when(mSpyFlags.getGlobalKillSwitch()).thenReturn(false); ShellUtils.runShellCommand("settings put global hidden_api_policy 1"); ExtendedMockito.doReturn(mUserPrivacyStatus).when(UserPrivacyStatus::getInstance); doReturn(true).when(mUserPrivacyStatus).isMeasurementEnabled(); doReturn(true).when(mUserPrivacyStatus).isProtectedAudienceEnabled(); setUpTestData(); mSfo = new ServiceFlowOrchestrator(); } @After public void cleanup() { mDbHelper.getWritableDatabase().close(); mDbHelper.getReadableDatabase().close(); mDbHelper.close(); } @Test public void testAppRequestFlow_MeasurementControlRevoked() throws InterruptedException { int originalQueriesCount = getDbTableSize(QueriesContract.QueriesEntry.TABLE_NAME); int originalEventsCount = getDbTableSize(EventsContract.EventsEntry.TABLE_NAME); doReturn(false).when(mUserPrivacyStatus).isMeasurementEnabled(); mSfo.schedule(ServiceFlowType.APP_REQUEST_FLOW, mContext.getPackageName(), new ComponentName(mContext.getPackageName(), "com.test.TestPersonalizationService"), createWrappedAppParams(), new TestExecuteCallback(), mContext, 100L, 110L); mLatch.await(); assertTrue(mCallbackSuccess); assertFalse(mExecuteCallback.isEmpty()); // make sure no request or event records are written to the database assertEquals(originalQueriesCount, getDbTableSize(QueriesContract.QueriesEntry.TABLE_NAME)); assertEquals(originalEventsCount, getDbTableSize(EventsContract.EventsEntry.TABLE_NAME)); } @Test public void testAppRequestFlow_TargetingControlRevoked() throws InterruptedException { doReturn(false).when(mUserPrivacyStatus).isProtectedAudienceEnabled(); mSfo.schedule(ServiceFlowType.APP_REQUEST_FLOW, mContext.getPackageName(), new ComponentName(mContext.getPackageName(), "com.test.TestPersonalizationService"), createWrappedAppParams(), new TestExecuteCallback(), mContext, 100L, 110L); mLatch.await(); assertTrue(mCallbackSuccess); assertTrue(mExecuteCallback.isEmpty()); } @Test public void testAppRequestFlow_OutputDataBlocked() throws InterruptedException { when(mSpyFlags.getOutputDataAllowList()).thenReturn(""); mSfo.schedule(ServiceFlowType.APP_REQUEST_FLOW, mContext.getPackageName(), new ComponentName(mContext.getPackageName(), "com.test.TestPersonalizationService"), createWrappedAppParams(), new TestExecuteCallback(), mContext, 100L, 110L); mLatch.await(); assertTrue(mCallbackSuccess); assertNull(mExecuteCallback.getByteArray(Constants.EXTRA_OUTPUT_DATA)); assertEquals(2, getDbTableSize(QueriesContract.QueriesEntry.TABLE_NAME)); assertEquals(1, getDbTableSize(EventsContract.EventsEntry.TABLE_NAME)); } @Test public void testAppRequestFlow_OutputDataAllowed() throws InterruptedException { String contextPackageName = mContext.getPackageName(); when(mSpyFlags.getOutputDataAllowList()).thenReturn( contextPackageName + ";" + contextPackageName); mSfo.schedule(ServiceFlowType.APP_REQUEST_FLOW, mContext.getPackageName(), new ComponentName(mContext.getPackageName(), "com.test.TestPersonalizationService"), createWrappedAppParams(), new TestExecuteCallback(), mContext, 100L, 110L); mLatch.await(); assertTrue(mCallbackSuccess); assertArrayEquals( mExecuteCallback.getByteArray(Constants.EXTRA_OUTPUT_DATA), new byte[] {1, 2, 3}); assertEquals(2, getDbTableSize(QueriesContract.QueriesEntry.TABLE_NAME)); assertEquals(1, getDbTableSize(EventsContract.EventsEntry.TABLE_NAME)); } private int getDbTableSize(String tableName) { return mDbHelper.getReadableDatabase().query(tableName, null, null, null, null, null, null).getCount(); } private void setUpTestData() throws Exception { ArrayList rows = new ArrayList<>(); ContentValues row1 = new ContentValues(); row1.put("a", 1); rows.add(row1); ContentValues row2 = new ContentValues(); row2.put("b", 2); rows.add(row2); ComponentName service = new ComponentName( mContext.getPackageName(), "com.test.TestPersonalizationService"); byte[] queryDataBytes = OnDevicePersonalizationFlatbufferUtils.createQueryData( DbUtils.toTableValue(service), "AABBCCDD", rows); EventsDao.getInstanceForTest(mContext).insertQuery( new Query.Builder( System.currentTimeMillis(), mContext.getPackageName(), service, "AABBCCDD", queryDataBytes) .build()); EventsDao.getInstanceForTest(mContext); OnDevicePersonalizationVendorDataDao testVendorDao = OnDevicePersonalizationVendorDataDao .getInstanceForTest(mContext, new ComponentName(mContext.getPackageName(), "com.test.TestPersonalizationService"), PackageUtils.getCertDigest(mContext, mContext.getPackageName())); VendorData vendorData = new VendorData.Builder().setData(new byte[5]).setKey( "bid1").build(); testVendorDao.batchUpdateOrInsertVendorDataTransaction( List.of(vendorData), List.of("bid1"), 0 ); } private Bundle createWrappedAppParams() { try { Bundle wrappedParams = new Bundle(); ByteArrayParceledSlice buffer = new ByteArrayParceledSlice( PersistableBundleUtils.toByteArray(PersistableBundle.EMPTY)); wrappedParams.putParcelable(Constants.EXTRA_APP_PARAMS_SERIALIZED, buffer); return wrappedParams; } catch (Exception e) { throw new RuntimeException(e); } } class TestExecuteCallback extends IExecuteCallback.Stub { @Override public void onSuccess(Bundle bundle, CalleeMetadata calleeMetadata) { mCallbackSuccess = true; mExecuteCallback = bundle; mLatch.countDown(); } @Override public void onError(int errorCode, int isolatedServiceErrorCode, String message, CalleeMetadata calleeMetadata) { mCallbackError = true; mCallbackErrorCode = errorCode; mIsolatedServiceErrorCode = isolatedServiceErrorCode; mErrorMessage = message; mLatch.countDown(); } } }