Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 441fbfe9 authored by William Escande's avatar William Escande
Browse files

SystemServer: Parametrized Flag testing 

Run all tests with all the possible SystemServer flag value

Bug: 310000873
Test: atest BluetoothManagerServiceTest
Change-Id: Ifbc07ee48cb18215e7195aebc580465ede3b829b
parent 2a7fd906

service/tests/src/com/android/server/bluetooth/BluetoothManagerServiceTest.java

+69 −5
Original line number Original line Diff line number Diff line
@@ -30,6 +30,8 @@ import static com.android.server.bluetooth.BluetoothManagerService.MESSAGE_TIMEO 




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

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





import static org.junit.runners.Parameterized.Parameter;

import static org.junit.runners.Parameterized.Parameters;

import static org.mockito.ArgumentMatchers.anyBoolean;

import static org.mockito.ArgumentMatchers.anyBoolean;

import static org.mockito.ArgumentMatchers.eq;

import static org.mockito.ArgumentMatchers.eq;

import static org.mockito.Mockito.any;

import static org.mockito.Mockito.any;
@@ -58,23 +60,30 @@ import android.os.test.TestLooper; 
import android.provider.Settings;

import android.provider.Settings;





import androidx.test.platform.app.InstrumentationRegistry;

import androidx.test.platform.app.InstrumentationRegistry;

import androidx.test.runner.AndroidJUnit4;





import com.android.bluetooth.flags.FakeFeatureFlagsImpl;

import com.android.bluetooth.flags.FakeFeatureFlagsImpl;

import com.android.bluetooth.flags.Flags;

import com.android.bluetooth.flags.Flags;





import com.google.common.collect.Lists;



import org.junit.After;

import org.junit.After;

import org.junit.Before;

import org.junit.Before;

import org.junit.Test;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.junit.runner.RunWith;

import org.junit.runners.Parameterized;

import org.mockito.ArgumentCaptor;

import org.mockito.ArgumentCaptor;

import org.mockito.Mock;

import org.mockito.Mock;

import org.mockito.MockitoAnnotations;

import org.mockito.MockitoAnnotations;

import org.mockito.Spy;

import org.mockito.Spy;





import java.util.Arrays;

import java.util.List;

import java.util.Map;

import java.util.stream.Collectors;

import java.util.stream.IntStream;

import java.util.stream.IntStream;

import java.util.stream.Stream;





@RunWith(AndroidJUnit4.class)

@RunWith(Parameterized.class)

public class BluetoothManagerServiceTest {

public class BluetoothManagerServiceTest {

    private static final String TAG = BluetoothManagerServiceTest.class.getSimpleName();

    private static final String TAG = BluetoothManagerServiceTest.class.getSimpleName();

    private static final int STATE_BLE_TURNING_ON = 14; // can't find the symbol because hidden api

    private static final int STATE_BLE_TURNING_ON = 14; // can't find the symbol because hidden api
@@ -98,6 +107,63 @@ public class BluetoothManagerServiceTest { 
    @Mock AdapterBinder mAdapterBinder;

    @Mock AdapterBinder mAdapterBinder;

    private FakeFeatureFlagsImpl mFakeFlagsImpl;

    private FakeFeatureFlagsImpl mFakeFlagsImpl;





    @Parameter public FlagsValue mFlagsValue;



    static class FlagsValue {

        final Map<String, Boolean> mFlagsValue;



        FlagsValue(Map<String, Boolean> flagsValue) {

            mFlagsValue = flagsValue;

        }



        private static String formatFlag(String key) {

            return key.substring(key.lastIndexOf(".") + 1);

        }



        @Override

        public String toString() {

            return mFlagsValue.entrySet().stream()

                    .filter(Map.Entry::getValue)

                    .map(Map.Entry::getKey)

                    .map(FlagsValue::formatFlag)

                    .sorted()

                    .collect(Collectors.joining(", "));

        }

    }



    private static boolean filterFlags(Map<String, Boolean> map) {

        if (map.get(Flags.FLAG_USE_NEW_AIRPLANE_MODE)) {

            return !map.get(Flags.FLAG_AIRPLANE_RESSOURCES_IN_APP);

        }

        return true;

    }



    /** Generate the Map of flag for this test Suite */

    @Parameters(name = "{0}")

    public static Iterable<? extends Object> generateParameterizedFlagsValue() {

        final String[] flags = {

            Flags.FLAG_AIRPLANE_RESSOURCES_IN_APP,

            Flags.FLAG_USE_NEW_AIRPLANE_MODE,

            Flags.FLAG_USE_NEW_SATELLITE_MODE,

        };

        final Boolean[] values = {true, false};



        List<List<Map.Entry<String, Boolean>>> flagValues =

                Arrays.stream(flags)

                        .map(flag -> Arrays.stream(values).map(val -> Map.entry(flag, val)))

                        .map(Stream::toList)

                        .toList();



        return Lists.cartesianProduct(flagValues).stream()

                .map(list -> list.toArray(new Map.Entry[0]))

                .map(Map::ofEntries)

                .filter(BluetoothManagerServiceTest::filterFlags)

                .map(FlagsValue::new)

                .map(List::of)

                .map(List::toArray)

                .toList();

    }



    TestLooper mLooper;

    TestLooper mLooper;





    boolean mHasException = false;

    boolean mHasException = false;
@@ -148,9 +214,7 @@ public class BluetoothManagerServiceTest { 
        mLooper = new TestLooper();

        mLooper = new TestLooper();





        mFakeFlagsImpl = new FakeFeatureFlagsImpl();

        mFakeFlagsImpl = new FakeFeatureFlagsImpl();

        mFakeFlagsImpl.setFlag(Flags.FLAG_AIRPLANE_RESSOURCES_IN_APP, false);

        mFlagsValue.mFlagsValue.forEach(mFakeFlagsImpl::setFlag);

        mFakeFlagsImpl.setFlag(Flags.FLAG_USE_NEW_AIRPLANE_MODE, false);

        mFakeFlagsImpl.setFlag(Flags.FLAG_USE_NEW_SATELLITE_MODE, false);





        mManagerService =

        mManagerService =

                new BluetoothManagerService(mContext, mLooper.getLooper(), mFakeFlagsImpl);

                new BluetoothManagerService(mContext, mLooper.getLooper(), mFakeFlagsImpl);