Merge "Visit Person.getUri()." into main

     */

    public void visitUris(@NonNull Consumer<Uri> visitor) {

        visitor.accept(getIconUri());

        if (Flags.visitRiskyUris()) {

            if (mUri != null && !mUri.isEmpty()) {

                visitor.accept(Uri.parse(mUri));

            }

        }

    }

    /** Builder for the immutable {@link Person} class. */

import static com.google.common.base.Preconditions.checkState;

import static com.google.common.truth.Truth.assertThat;

import android.app.Flags;

import android.app.Notification;

import android.app.PendingIntent;

import android.app.Person;

import android.os.Bundle;

import android.os.IBinder;

import android.os.Parcel;

import android.platform.test.flag.junit.SetFlagsRule;

import android.util.Log;

import android.view.LayoutInflater;

import android.view.View;

import androidx.test.InstrumentationRegistry;

import androidx.test.runner.AndroidJUnit4;

import com.android.internal.config.sysui.SystemUiSystemPropertiesFlags;

import com.android.server.UiServiceTestCase;

import com.google.common.base.Strings;

import com.google.common.collect.Multimap;

import com.google.common.truth.Expect;

import org.junit.After;

import org.junit.Before;

import org.junit.Rule;

import org.junit.Test;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Comparator;

import java.util.HashSet;

import java.util.List;

import java.util.Optional;

import java.util.Set;

@RunWith(AndroidJUnit4.class)

public class NotificationVisitUrisTest extends UiServiceTestCase {

    @Rule

    public final SetFlagsRule mSetFlagsRule = new SetFlagsRule();

    private static final String TAG = "VisitUrisTest";

    // Methods that are known to add Uris that are *NOT* verified.

    // This list should be emptied! Items can be removed as bugs are fixed.

    // This list should only be used temporarily if needed, and any element in this list should

    // have a tracking bug associated.

    private static final Multimap<Class<?>, String> KNOWN_BAD =

            ImmutableMultimap.<Class<?>, String>builder()

                    .put(Person.Builder.class, "setUri") // TODO: b/281044385

                    .build();

            ImmutableMultimap.<Class<?>, String>builder().build();

    // Types that we can't really produce. No methods receiving these parameters will be invoked.

    private static final ImmutableSet<Class<?>> UNUSABLE_TYPES =

    @Before

    public void setUp() {

        mContext = InstrumentationRegistry.getInstrumentation().getContext();

        mSetFlagsRule.enableFlags(Flags.FLAG_VISIT_RISKY_URIS);

    }

    @After

    public void tearDown() {

        SystemUiSystemPropertiesFlags.TEST_RESOLVER = null;

    }

    @Test // This is a meta-test, checks that the generators are not broken.

        notification.visitUris(visitor);

        Mockito.verify(visitor, Mockito.atLeastOnce()).accept(visitedUriCaptor.capture());

        List<Uri> visitedUris = new ArrayList<>(visitedUriCaptor.getAllValues());

        Set<Uri> visitedUris = new HashSet<>(visitedUriCaptor.getAllValues());

        visitedUris.remove(null);

        expect.withMessage(notificationTypeMessage)

                .that(visitedUris)

                .containsAtLeastElementsIn(includedUris);

        expect.that(KNOWN_BAD).isNotEmpty(); // Once empty, switch to containsExactlyElementsIn()

                .containsExactlyElementsIn(includedUris);

    }

    private static Generated<Notification> buildNotification(Context context,

    private static class SpecialParameterGenerator {

        private static final ImmutableSet<Class<?>> INTERESTING_CLASSES =

                ImmutableSet.of(

                        Person.class, Uri.class, Icon.class, Intent.class, PendingIntent.class,

                        RemoteViews.class);

                ImmutableSet.of(Person.class, Uri.class, Icon.class, Intent.class,

                        PendingIntent.class, RemoteViews.class);

        private static final ImmutableSet<Class<?>> MOCKED_CLASSES = ImmutableSet.of();

        private static final ImmutableMap<Class<?>, Object> PRIMITIVE_VALUES =

        }

        static boolean canGenerate(Class<?> clazz) {

            return (INTERESTING_CLASSES.contains(clazz) && !clazz.equals(Person.class))

            return INTERESTING_CLASSES.contains(clazz)

                    || MOCKED_CLASSES.contains(clazz)

                    || clazz.equals(Context.class)

                    || clazz.equals(Bundle.class)

                return Icon.createWithContentUri(iconUri);

            }

            if (clazz == Person.class) {

                // TODO(b/310189261): Person.setUri takes a string instead of a URI. We should

                //  find a way to use the SpecialParameterGenerator instead of this custom one.

                Uri personUri = generateUri(

                        where.plus(Person.Builder.class).plus("setUri", String.class));

                Uri iconUri = generateUri(where.plus(Person.Builder.class).plus("setIcon",

                        Icon.class).plus(Icon.class).plus("createWithContentUri", Uri.class));

                return new Person.Builder().setUri(personUri.toString()).setIcon(

                        Icon.createWithContentUri(iconUri)).setName("John Doe").build();

            }

            if (clazz == Intent.class) {

                // TODO(b/281044385): Are Intent Uris (new Intent(String,Uri)) relevant?

                return new Intent("action");

    private static class Location {

        private static class Item {

            @Nullable private final Class<?> mMaybeClass;

            @Nullable private final Executable mMaybeMethod;

            @Nullable private final String mExtra;

            @Nullable

            private final Class<?> mMaybeClass;

            @Nullable

            private final Executable mMaybeMethod;

            @Nullable

            private final String mExtra;

            Item(@NonNull Class<?> clazz) {

                mMaybeClass = checkNotNull(clazz);