Modify test to accomodate unchangable flag

Bug: 301669353
Change-Id: I6d5857d2e414a0c87e1a6212bf36da1bcb5fcbdf
This commit is contained in:
Kyle Zhang
2024-03-19 20:50:39 +00:00
parent c35fc38be9
commit e99e5f39c4

View File

@@ -68,7 +68,7 @@ public class ForceSwSecureCryptoFallbackPreferenceControllerTest {
public void updateState_flagEnabled_checkPreference() { public void updateState_flagEnabled_checkPreference() {
mSetFlagsRule.enableFlags(Flags.FLAG_FORCE_L3_ENABLED); mSetFlagsRule.enableFlags(Flags.FLAG_FORCE_L3_ENABLED);
mController.updateState(mPreference); mController.updateState(mPreference);
assertThat(mPreference.isEnabled()).isTrue(); assumeTrue(mPreference.isEnabled());
assertThat(mPreference.isChecked()).isFalse(); assertThat(mPreference.isChecked()).isFalse();
assertThat(WidevineProperties.forcel3_enabled().orElse(false)).isFalse(); assertThat(WidevineProperties.forcel3_enabled().orElse(false)).isFalse();
@@ -107,12 +107,11 @@ public class ForceSwSecureCryptoFallbackPreferenceControllerTest {
@Test @Test
public void updateState_checkWidevine() throws Exception { public void updateState_checkWidevine() throws Exception {
MediaDrm drm; try (MediaDrm drm = new MediaDrm(WIDEVINE_UUID)) {
try {
drm = new MediaDrm(WIDEVINE_UUID);
assumeTrue(drm.getPropertyString("securityLevel").equals("L1")); assumeTrue(drm.getPropertyString("securityLevel").equals("L1"));
mSetFlagsRule.enableFlags(Flags.FLAG_FORCE_L3_ENABLED); mSetFlagsRule.enableFlags(Flags.FLAG_FORCE_L3_ENABLED);
drm.close(); mController.updateState(mPreference);
assumeTrue(mPreference.isEnabled());
} catch (UnsupportedSchemeException ex) { } catch (UnsupportedSchemeException ex) {
assumeNoException(ex); assumeNoException(ex);
} }
@@ -120,17 +119,22 @@ public class ForceSwSecureCryptoFallbackPreferenceControllerTest {
// L3 enforced // L3 enforced
mController.setChecked(true); mController.setChecked(true);
mController.updateState(mPreference); mController.updateState(mPreference);
assertThat(WidevineProperties.forcel3_enabled().orElse(false)).isTrue();
assertThat(mPreference.isEnabled()).isTrue(); assertThat(mPreference.isEnabled()).isTrue();
assertThat(mPreference.isChecked()).isTrue(); assertThat(mPreference.isChecked()).isTrue();
drm = new MediaDrm(WIDEVINE_UUID); try (MediaDrm drm = new MediaDrm(WIDEVINE_UUID)) {
assertThat(drm.getPropertyString("securityLevel")).isEqualTo("L3"); assertThat(drm.getPropertyString("securityLevel")).isEqualTo("L3");
} catch (UnsupportedSchemeException ex) {
assumeNoException(ex);
}
// Switch back to L1 // Switch back to L1
mController.setChecked(false); mController.setChecked(false);
mController.updateState(mPreference); mController.updateState(mPreference);
drm.close();
drm = new MediaDrm(WIDEVINE_UUID); try (MediaDrm drm = new MediaDrm(WIDEVINE_UUID)) {
assertThat(drm.getPropertyString("securityLevel")).isEqualTo("L1"); assertThat(drm.getPropertyString("securityLevel")).isEqualTo("L1");
} catch (UnsupportedSchemeException ex) {
assumeNoException(ex);
}
} }
} }