commit 1d1c5a9b16cd03adaa72d3f6506356003ab7f5c6
author Shreshta Manu <shreshtabm@google.com> Mon Apr 29 17:50:23 2024 +0000
committer Shreshta Manu <shreshtabm@google.com> Mon Apr 29 17:54:16 2024 +0000
tree f4c85e4784f57a77d01c8e59102d683ada523891
parent aefedea58801ec7e97d165b1e1099126f9106c19

[uwb] Mock dataTransferPhaseConfig

Do not use requireFlag annotation for the flag, instead mock
the flag to be true.

Bug: 337303634
Test: atest ServiceUwbTests

Change-Id: I105336292a3d5dfc7fa7423117ac4c1e17e8f4ca

service/tests/src/com/android/server/uwb/UwbServiceImplTest.java

diff --git a/service/tests/src/com/android/server/uwb/UwbServiceImplTest.java b/service/tests/src/com/android/server/uwb/UwbServiceImplTest.java
index 5734595..84902d2 100644
--- a/service/tests/src/com/android/server/uwb/UwbServiceImplTest.java
+++ b/service/tests/src/com/android/server/uwb/UwbServiceImplTest.java
@@ -844,9 +844,9 @@
     }
 
     @Test
-    @RequiresFlagsEnabled(Flags.FLAG_DATA_TRANSFER_PHASE_CONFIG)
     public void testSetDataTransferPhaseConfig() throws Exception {
         assumeTrue(SdkLevel.isAtLeastV()); // Test should only run on V+ devices.
+        when(mFeatureFlags.dataTransferPhaseConfig()).thenReturn(true);
         final SessionHandle sessionHandle = mock(SessionHandle.class);
         PersistableBundle bundle = new PersistableBundle();
         mUwbServiceImpl.setDataTransferPhaseConfig(sessionHandle, bundle);