summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--core/tests/BroadcastRadioTests/src/android/hardware/radio/ProgramListTest.java10
-rw-r--r--core/tests/BroadcastRadioTests/src/android/hardware/radio/RadioManagerTest.java15
-rw-r--r--core/tests/BroadcastRadioTests/src/android/hardware/radio/RadioMetadataTest.java28
-rw-r--r--core/tests/BroadcastRadioTests/src/android/hardware/radio/TunerAdapterTest.java19
-rw-r--r--core/tests/BroadcastRadioTests/src/com/android/server/broadcastradio/aidl/ConversionUtilsTest.java30
-rw-r--r--core/tests/BroadcastRadioTests/src/com/android/server/broadcastradio/aidl/TunerSessionTest.java11
6 files changed, 53 insertions, 60 deletions
diff --git a/core/tests/BroadcastRadioTests/src/android/hardware/radio/ProgramListTest.java b/core/tests/BroadcastRadioTests/src/android/hardware/radio/ProgramListTest.java
index d4a88c49b38f..a3a1d3a5e7a2 100644
--- a/core/tests/BroadcastRadioTests/src/android/hardware/radio/ProgramListTest.java
+++ b/core/tests/BroadcastRadioTests/src/android/hardware/radio/ProgramListTest.java
@@ -34,9 +34,7 @@ import android.content.pm.ApplicationInfo;
import android.os.Build;
import android.os.Parcel;
import android.os.RemoteException;
-import android.platform.test.annotations.RequiresFlagsEnabled;
-import android.platform.test.flag.junit.CheckFlagsRule;
-import android.platform.test.flag.junit.DeviceFlagsValueProvider;
+import android.platform.test.flag.junit.SetFlagsRule;
import android.util.ArraySet;
import com.google.common.truth.Expect;
@@ -150,7 +148,7 @@ public final class ProgramListTest {
@Rule
public final Expect mExpect = Expect.create();
@Rule
- public final CheckFlagsRule mCheckFlagsRule = DeviceFlagsValueProvider.createCheckFlagsRule();
+ public final SetFlagsRule mSetFlagsRule = new SetFlagsRule();
@Test
public void getIdentifierTypes_forFilter() {
@@ -631,8 +629,8 @@ public final class ProgramListTest {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void getProgramInfos() throws Exception {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
createRadioTuner();
mProgramList = mRadioTuner.getDynamicProgramList(TEST_FILTER);
registerListCallbacks(/* numCallbacks= */ 1);
@@ -648,8 +646,8 @@ public final class ProgramListTest {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void getProgramInfos_withIdNotFound() throws Exception {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
createRadioTuner();
mProgramList = mRadioTuner.getDynamicProgramList(TEST_FILTER);
registerListCallbacks(/* numCallbacks= */ 1);
diff --git a/core/tests/BroadcastRadioTests/src/android/hardware/radio/RadioManagerTest.java b/core/tests/BroadcastRadioTests/src/android/hardware/radio/RadioManagerTest.java
index 03de1430fec8..89464d14d1c7 100644
--- a/core/tests/BroadcastRadioTests/src/android/hardware/radio/RadioManagerTest.java
+++ b/core/tests/BroadcastRadioTests/src/android/hardware/radio/RadioManagerTest.java
@@ -32,9 +32,7 @@ import android.content.Context;
import android.content.pm.ApplicationInfo;
import android.os.Parcel;
import android.os.RemoteException;
-import android.platform.test.annotations.RequiresFlagsEnabled;
-import android.platform.test.flag.junit.CheckFlagsRule;
-import android.platform.test.flag.junit.DeviceFlagsValueProvider;
+import android.platform.test.flag.junit.SetFlagsRule;
import android.util.ArrayMap;
import org.junit.Rule;
@@ -168,7 +166,7 @@ public final class RadioManagerTest {
private ICloseHandle mCloseHandleMock;
@Rule
- public final CheckFlagsRule mCheckFlagsRule = DeviceFlagsValueProvider.createCheckFlagsRule();
+ public final SetFlagsRule mSetFlagsRule = new SetFlagsRule();
@Test
public void getType_forBandDescriptor() {
@@ -962,22 +960,25 @@ public final class RadioManagerTest {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void isSignalAcquired_forProgramInfo() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
+
assertWithMessage("Signal acquisition status for HD program info")
.that(HD_PROGRAM_INFO.isSignalAcquired()).isTrue();
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void isHdSisAvailable_forProgramInfo() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
+
assertWithMessage("SIS information acquisition status for HD program")
.that(HD_PROGRAM_INFO.isHdSisAvailable()).isTrue();
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void isHdAudioAvailable_forProgramInfo() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
+
assertWithMessage("Audio acquisition status for HD program")
.that(HD_PROGRAM_INFO.isHdAudioAvailable()).isFalse();
}
diff --git a/core/tests/BroadcastRadioTests/src/android/hardware/radio/RadioMetadataTest.java b/core/tests/BroadcastRadioTests/src/android/hardware/radio/RadioMetadataTest.java
index 3891accbba44..7b9121eb6085 100644
--- a/core/tests/BroadcastRadioTests/src/android/hardware/radio/RadioMetadataTest.java
+++ b/core/tests/BroadcastRadioTests/src/android/hardware/radio/RadioMetadataTest.java
@@ -22,10 +22,7 @@ import static org.junit.Assert.assertThrows;
import android.graphics.Bitmap;
import android.os.Parcel;
-import android.platform.test.annotations.RequiresFlagsDisabled;
-import android.platform.test.annotations.RequiresFlagsEnabled;
-import android.platform.test.flag.junit.CheckFlagsRule;
-import android.platform.test.flag.junit.DeviceFlagsValueProvider;
+import android.platform.test.flag.junit.SetFlagsRule;
import org.junit.Rule;
import org.junit.Test;
@@ -52,7 +49,7 @@ public final class RadioMetadataTest {
private Bitmap mBitmapValue;
@Rule
- public final CheckFlagsRule mCheckFlagsRule = DeviceFlagsValueProvider.createCheckFlagsRule();
+ public final SetFlagsRule mSetFlagsRule = new SetFlagsRule();
@Test
public void describeContents_forClock() {
@@ -128,8 +125,8 @@ public final class RadioMetadataTest {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void putStringArray_withIllegalKey_throwsException() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
String invalidStringArrayKey = RadioMetadata.METADATA_KEY_HD_STATION_NAME_LONG;
IllegalArgumentException thrown = assertThrows(IllegalArgumentException.class, () -> {
@@ -142,8 +139,9 @@ public final class RadioMetadataTest {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void putStringArray_withNullKey_throwsException() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
+
NullPointerException thrown = assertThrows(NullPointerException.class, () -> {
mBuilder.putStringArray(/* key= */ null, UFIDS_VALUE);
});
@@ -153,8 +151,9 @@ public final class RadioMetadataTest {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void putStringArray_withNullString_throwsException() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
+
NullPointerException thrown = assertThrows(NullPointerException.class, () -> {
mBuilder.putStringArray(RadioMetadata.METADATA_KEY_UFIDS, /* value= */ null);
});
@@ -281,8 +280,8 @@ public final class RadioMetadataTest {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void getStringArray_withKeyInMetadata() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
String key = RadioMetadata.METADATA_KEY_UFIDS;
RadioMetadata metadata = mBuilder.putStringArray(key, UFIDS_VALUE).build();
@@ -291,8 +290,8 @@ public final class RadioMetadataTest {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void getStringArray_withKeyNotInMetadata() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
String key = RadioMetadata.METADATA_KEY_UFIDS;
RadioMetadata metadata = mBuilder.build();
@@ -305,8 +304,8 @@ public final class RadioMetadataTest {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void getStringArray_withNullKey() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
RadioMetadata metadata = mBuilder.build();
NullPointerException thrown = assertThrows(NullPointerException.class, () -> {
@@ -318,8 +317,8 @@ public final class RadioMetadataTest {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void getStringArray_withInvalidKey() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
String invalidClockKey = RadioMetadata.METADATA_KEY_HD_STATION_NAME_LONG;
RadioMetadata metadata = mBuilder.build();
@@ -413,7 +412,6 @@ public final class RadioMetadataTest {
}
@Test
- @RequiresFlagsDisabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void writeToParcel_forRadioMetadata() {
RadioMetadata metadataExpected = mBuilder
.putInt(RadioMetadata.METADATA_KEY_RDS_PI, INT_KEY_VALUE)
@@ -430,8 +428,8 @@ public final class RadioMetadataTest {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void writeToParcel_forRadioMetadata_withStringArrayTypeMetadata() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
RadioMetadata metadataExpected = mBuilder
.putInt(RadioMetadata.METADATA_KEY_RDS_PI, INT_KEY_VALUE)
.putString(RadioMetadata.METADATA_KEY_ARTIST, ARTIST_KEY_VALUE)
@@ -443,7 +441,7 @@ public final class RadioMetadataTest {
parcel.setDataPosition(0);
RadioMetadata metadataFromParcel = RadioMetadata.CREATOR.createFromParcel(parcel);
- assertWithMessage("Radio metadata created from parcel")
+ assertWithMessage("Radio metadata created from parcel with string array type metadata")
.that(metadataFromParcel).isEqualTo(metadataExpected);
}
}
diff --git a/core/tests/BroadcastRadioTests/src/android/hardware/radio/TunerAdapterTest.java b/core/tests/BroadcastRadioTests/src/android/hardware/radio/TunerAdapterTest.java
index 7ca806b49b68..4841711f712d 100644
--- a/core/tests/BroadcastRadioTests/src/android/hardware/radio/TunerAdapterTest.java
+++ b/core/tests/BroadcastRadioTests/src/android/hardware/radio/TunerAdapterTest.java
@@ -36,10 +36,7 @@ import android.content.pm.ApplicationInfo;
import android.graphics.Bitmap;
import android.os.Build;
import android.os.RemoteException;
-import android.platform.test.annotations.RequiresFlagsDisabled;
-import android.platform.test.annotations.RequiresFlagsEnabled;
-import android.platform.test.flag.junit.CheckFlagsRule;
-import android.platform.test.flag.junit.DeviceFlagsValueProvider;
+import android.platform.test.flag.junit.SetFlagsRule;
import org.junit.After;
import org.junit.Before;
@@ -84,7 +81,7 @@ public final class TunerAdapterTest {
private RadioTuner.Callback mCallbackMock;
@Rule
- public final CheckFlagsRule mCheckFlagsRule = DeviceFlagsValueProvider.createCheckFlagsRule();
+ public final SetFlagsRule mSetFlagsRule = new SetFlagsRule();
@Before
public void setUp() throws Exception {
@@ -613,9 +610,9 @@ public final class TunerAdapterTest {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void isConfigFlagSet_withForceAnalogWhenFmForceAnalogSupported()
throws Exception {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
when(mTunerMock.isConfigFlagSupported(anyInt())).thenReturn(true);
when(mTunerMock.isConfigFlagSet(RadioManager.CONFIG_FORCE_ANALOG_FM))
.thenReturn(true);
@@ -626,9 +623,9 @@ public final class TunerAdapterTest {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void isConfigFlagSet_withForceAnalogWhenFmForceAnalogNotSupported()
throws Exception {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
when(mTunerMock.isConfigFlagSupported(RadioManager.CONFIG_FORCE_ANALOG_FM))
.thenReturn(false);
when(mTunerMock.isConfigFlagSupported(RadioManager.CONFIG_FORCE_ANALOG)).thenReturn(true);
@@ -640,9 +637,9 @@ public final class TunerAdapterTest {
}
@Test
- @RequiresFlagsDisabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void isConfigFlagSet_withForceAnalogWhenHdRadioImprovedFeatureNotEnabled()
throws Exception {
+ mSetFlagsRule.disableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
when(mTunerMock.isConfigFlagSupported(anyInt())).thenReturn(true);
when(mTunerMock.isConfigFlagSet(RadioManager.CONFIG_FORCE_ANALOG)).thenReturn(false);
@@ -683,8 +680,8 @@ public final class TunerAdapterTest {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void setConfigFlag_withForceAnalogWhenFmForceAnalogSupported() throws Exception {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
when(mTunerMock.isConfigFlagSupported(anyInt())).thenReturn(true);
mRadioTuner.setConfigFlag(RadioManager.CONFIG_FORCE_ANALOG, /* value= */ false);
@@ -695,8 +692,8 @@ public final class TunerAdapterTest {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void setConfigFlag_withForceAnalogWhenFmForceAnalogNotSupported() throws Exception {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
when(mTunerMock.isConfigFlagSupported(anyInt())).thenReturn(true);
when(mTunerMock.isConfigFlagSupported(RadioManager.CONFIG_FORCE_ANALOG_FM))
.thenReturn(false);
@@ -709,9 +706,9 @@ public final class TunerAdapterTest {
}
@Test
- @RequiresFlagsDisabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void setConfigFlag_withForceAnalogWhenHdRadioImprovedFeatureNotEnabled()
throws Exception {
+ mSetFlagsRule.disableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
when(mTunerMock.isConfigFlagSupported(anyInt())).thenReturn(true);
mRadioTuner.setConfigFlag(RadioManager.CONFIG_FORCE_ANALOG, /* value= */ false);
diff --git a/core/tests/BroadcastRadioTests/src/com/android/server/broadcastradio/aidl/ConversionUtilsTest.java b/core/tests/BroadcastRadioTests/src/com/android/server/broadcastradio/aidl/ConversionUtilsTest.java
index 49a7ba855908..15bb66b3e303 100644
--- a/core/tests/BroadcastRadioTests/src/com/android/server/broadcastradio/aidl/ConversionUtilsTest.java
+++ b/core/tests/BroadcastRadioTests/src/com/android/server/broadcastradio/aidl/ConversionUtilsTest.java
@@ -40,10 +40,7 @@ import android.hardware.radio.RadioManager;
import android.hardware.radio.RadioMetadata;
import android.hardware.radio.UniqueProgramIdentifier;
import android.os.ServiceSpecificException;
-import android.platform.test.annotations.RequiresFlagsDisabled;
-import android.platform.test.annotations.RequiresFlagsEnabled;
-import android.platform.test.flag.junit.CheckFlagsRule;
-import android.platform.test.flag.junit.DeviceFlagsValueProvider;
+import android.platform.test.flag.junit.SetFlagsRule;
import com.android.dx.mockito.inline.extended.StaticMockitoSessionBuilder;
import com.android.server.broadcastradio.ExtendedRadioMockitoTestCase;
@@ -159,7 +156,7 @@ public final class ConversionUtilsTest extends ExtendedRadioMockitoTestCase {
@Rule
public final Expect expect = Expect.create();
@Rule
- public final CheckFlagsRule mCheckFlagsRule = DeviceFlagsValueProvider.createCheckFlagsRule();
+ public final SetFlagsRule mSetFlagsRule = new SetFlagsRule();
@Override
protected void initializeSession(StaticMockitoSessionBuilder builder) {
@@ -317,9 +314,10 @@ public final class ConversionUtilsTest extends ExtendedRadioMockitoTestCase {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void identifierToHalProgramIdentifier_withFlagEnabled() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
ProgramSelector.Identifier hdLocationId = createHdStationLocationIdWithFlagEnabled();
+
ProgramIdentifier halHdLocationId =
ConversionUtils.identifierToHalProgramIdentifier(hdLocationId);
@@ -336,10 +334,11 @@ public final class ConversionUtilsTest extends ExtendedRadioMockitoTestCase {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void identifierFromHalProgramIdentifier_withFlagEnabled() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
ProgramSelector.Identifier hdLocationIdExpected =
createHdStationLocationIdWithFlagEnabled();
+
ProgramSelector.Identifier hdLocationId =
ConversionUtils.identifierFromHalProgramIdentifier(TEST_HAL_HD_STATION_LOCATION_ID);
@@ -348,8 +347,9 @@ public final class ConversionUtilsTest extends ExtendedRadioMockitoTestCase {
}
@Test
- @RequiresFlagsDisabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void identifierFromHalProgramIdentifier_withFlagDisabled_returnsNull() {
+ mSetFlagsRule.disableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
+
ProgramSelector.Identifier hdLocationId =
ConversionUtils.identifierFromHalProgramIdentifier(TEST_HAL_HD_STATION_LOCATION_ID);
@@ -432,8 +432,8 @@ public final class ConversionUtilsTest extends ExtendedRadioMockitoTestCase {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void programSelectorMeetsSdkVersionRequirement_withLowerVersionSecondaryId() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
ProgramSelector hdSelector = createHdSelectorWithFlagEnabled();
expect.withMessage("Selector %s with secondary id requiring higher-version SDK version",
@@ -449,8 +449,8 @@ public final class ConversionUtilsTest extends ExtendedRadioMockitoTestCase {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void programSelectorMeetsSdkVersionRequirement_withRequiredVersionAndFlagEnabled() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
ProgramSelector hdSelector = createHdSelectorWithFlagEnabled();
expect.withMessage("Selector %s with required SDK version and feature flag enabled",
@@ -548,8 +548,8 @@ public final class ConversionUtilsTest extends ExtendedRadioMockitoTestCase {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void configFlagMeetsSdkVersionRequirement_withRequiredSdkVersionAndFlagEnabled() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
int halForceAmAnalogFlag = ConfigFlag.FORCE_ANALOG_FM;
expect.withMessage("Force Analog FM flag with required SDK version and feature flag"
@@ -558,8 +558,8 @@ public final class ConversionUtilsTest extends ExtendedRadioMockitoTestCase {
}
@Test
- @RequiresFlagsDisabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void configFlagMeetsSdkVersionRequirement_withRequiredSdkVersionAndFlagDisabled() {
+ mSetFlagsRule.disableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
int halForceAmAnalogFlag = ConfigFlag.FORCE_ANALOG_FM;
expect.withMessage("Force Analog FM with required SDK version and with feature flag"
@@ -586,8 +586,9 @@ public final class ConversionUtilsTest extends ExtendedRadioMockitoTestCase {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void radioMetadataFromHalMetadata_withFlagEnabled() {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
+
RadioMetadata convertedMetadata = ConversionUtils.radioMetadataFromHalMetadata(
new Metadata[]{TEST_HAL_SONG_TITLE, TEST_HAL_HD_SUBCHANNELS, TEST_HAL_ALBUM_ART});
@@ -605,8 +606,9 @@ public final class ConversionUtilsTest extends ExtendedRadioMockitoTestCase {
}
@Test
- @RequiresFlagsDisabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void radioMetadataFromHalMetadata_withFlagDisabled() {
+ mSetFlagsRule.disableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
+
RadioMetadata convertedMetadata = ConversionUtils.radioMetadataFromHalMetadata(
new Metadata[]{TEST_HAL_SONG_TITLE, TEST_HAL_HD_SUBCHANNELS, TEST_HAL_ALBUM_ART});
diff --git a/core/tests/BroadcastRadioTests/src/com/android/server/broadcastradio/aidl/TunerSessionTest.java b/core/tests/BroadcastRadioTests/src/com/android/server/broadcastradio/aidl/TunerSessionTest.java
index 7bef5abc3d63..296c45136292 100644
--- a/core/tests/BroadcastRadioTests/src/com/android/server/broadcastradio/aidl/TunerSessionTest.java
+++ b/core/tests/BroadcastRadioTests/src/com/android/server/broadcastradio/aidl/TunerSessionTest.java
@@ -53,10 +53,7 @@ import android.os.ParcelableException;
import android.os.RemoteException;
import android.os.ServiceSpecificException;
import android.os.UserHandle;
-import android.platform.test.annotations.RequiresFlagsDisabled;
-import android.platform.test.annotations.RequiresFlagsEnabled;
-import android.platform.test.flag.junit.CheckFlagsRule;
-import android.platform.test.flag.junit.DeviceFlagsValueProvider;
+import android.platform.test.flag.junit.SetFlagsRule;
import android.util.ArrayMap;
import android.util.ArraySet;
@@ -164,7 +161,7 @@ public final class TunerSessionTest extends ExtendedRadioMockitoTestCase {
@Rule
public final Expect expect = Expect.create();
@Rule
- public final CheckFlagsRule mCheckFlagsRule = DeviceFlagsValueProvider.createCheckFlagsRule();
+ public final SetFlagsRule mSetFlagsRule = new SetFlagsRule();
@Override
protected void initializeSession(StaticMockitoSessionBuilder builder) {
@@ -1231,8 +1228,8 @@ public final class TunerSessionTest extends ExtendedRadioMockitoTestCase {
}
@Test
- @RequiresFlagsEnabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void onConfigFlagUpdated_withRequiredFlagEnabled_invokesCallbacks() throws Exception {
+ mSetFlagsRule.enableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
openAidlClients(/* numClients= */ 1);
mHalTunerCallback.onConfigFlagUpdated(ConfigFlag.FORCE_ANALOG_FM, true);
@@ -1242,9 +1239,9 @@ public final class TunerSessionTest extends ExtendedRadioMockitoTestCase {
}
@Test
- @RequiresFlagsDisabled(Flags.FLAG_HD_RADIO_IMPROVED)
public void onConfigFlagUpdated_withRequiredFlagDisabled_doesNotInvokeCallbacks()
throws Exception {
+ mSetFlagsRule.disableFlags(Flags.FLAG_HD_RADIO_IMPROVED);
openAidlClients(/* numClients= */ 1);
mHalTunerCallback.onConfigFlagUpdated(ConfigFlag.FORCE_ANALOG_FM, true);