1 files changed, 143 insertions, 0 deletions
diff --git a/tests/src/com/android/ims/rcs/uce/presence/publish/DeviceCapabilityInfoTest.java b/tests/src/com/android/ims/rcs/uce/presence/publish/DeviceCapabilityInfoTest.java
new file mode 100644
index 00000000..72cd26f4
--- /dev/null
+++ b/tests/src/com/android/ims/rcs/uce/presence/publish/DeviceCapabilityInfoTest.java
@@ -0,0 +1,143 @@
+/*
+ * Copyright (C) 2022 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.android.ims.rcs.uce.presence.publish;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNull;
+
+import android.net.Uri;
+import android.telecom.PhoneAccount;
+
+import androidx.test.ext.junit.runners.AndroidJUnit4;
+import androidx.test.filters.SmallTest;
+
+import com.android.ims.ImsTestBase;
+
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+
+@RunWith(AndroidJUnit4.class)
+public class DeviceCapabilityInfoTest extends ImsTestBase {
+
+    int mSubId = 1;
+
+    String sipNumber = "123456789";
+    String telNumber = "987654321";
+
+    @Before
+    public void setUp() throws Exception {
+        super.setUp();
+    }
+
+    @After
+    public void tearDown() throws Exception {
+        super.tearDown();
+    }
+
+    @Test
+    @SmallTest
+    public void testGetImsAssociatedUriWithoutPreferTelUri() throws Exception {
+        DeviceCapabilityInfo deviceCapInfo = createDeviceCapabilityInfo();
+
+        Uri[] uris = new Uri[2];
+        uris[0] = Uri.fromParts(PhoneAccount.SCHEME_SIP, sipNumber, null);
+        uris[1] = Uri.fromParts(PhoneAccount.SCHEME_TEL, telNumber, null);
+
+        // When stored in the order of SIP, TEL URI, check whether the SIP URI saved at
+        // the beginning is retrieved.
+        deviceCapInfo.updateRcsAssociatedUri(uris);
+        Uri outUri = deviceCapInfo.getImsAssociatedUri(false);
+
+        String numbers = outUri.getSchemeSpecificPart();
+        String[] numberParts = numbers.split("[@;:]");
+        String number = numberParts[0];
+
+        assertEquals(number, sipNumber);
+
+        // When stored in the order of TEL, SIP URI, check whether the TEL URI saved at
+        // the beginning is retrieved.
+        deviceCapInfo = createDeviceCapabilityInfo();
+
+        uris[0] = Uri.fromParts(PhoneAccount.SCHEME_TEL, telNumber, null);
+        uris[1] = Uri.fromParts(PhoneAccount.SCHEME_SIP, sipNumber, null);
+
+        deviceCapInfo.updateRcsAssociatedUri(uris);
+        outUri = deviceCapInfo.getImsAssociatedUri(false);
+
+        numbers = outUri.getSchemeSpecificPart();
+        numberParts = numbers.split("[@;:]");
+        number = numberParts[0];
+
+        assertEquals(number, telNumber);
+    }
+
+    @Test
+    @SmallTest
+    public void testGetImsAssociatedUriWithPreferTelUri() throws Exception {
+        DeviceCapabilityInfo deviceCapInfo = createDeviceCapabilityInfo();
+
+        Uri[] uris = new Uri[2];
+        uris[0] = Uri.fromParts(PhoneAccount.SCHEME_SIP, sipNumber, null);
+        uris[1] = Uri.fromParts(PhoneAccount.SCHEME_TEL, telNumber, null);
+
+        // Check whether TEL URI is returned when preferTelUri is true even if SIP and TEL URI
+        // are in the order.
+        deviceCapInfo.updateRcsAssociatedUri(uris);
+        Uri outUri = deviceCapInfo.getImsAssociatedUri(true);
+
+        String numbers = outUri.getSchemeSpecificPart();
+        String[] numberParts = numbers.split("[@;:]");
+        String number = numberParts[0];
+
+        assertEquals(number, telNumber);
+
+        // If preferTelUri is true, check if a TEL URI is returned.
+        deviceCapInfo = createDeviceCapabilityInfo();
+
+        uris[0] = Uri.fromParts(PhoneAccount.SCHEME_TEL, telNumber, null);
+        uris[1] = Uri.fromParts(PhoneAccount.SCHEME_SIP, sipNumber, null);
+
+        deviceCapInfo.updateRcsAssociatedUri(uris);
+        outUri = deviceCapInfo.getImsAssociatedUri(true);
+
+        numbers = outUri.getSchemeSpecificPart();
+        numberParts = numbers.split("[@;:]");
+        number = numberParts[0];
+
+        assertEquals(number, telNumber);
+
+        //If there is only SIP URI, check the return uri is null if preferTelUir is true.
+        deviceCapInfo = createDeviceCapabilityInfo();
+
+        uris[0] = Uri.fromParts(PhoneAccount.SCHEME_SIP, telNumber, null);
+        uris[1] = Uri.fromParts(PhoneAccount.SCHEME_SIP, sipNumber, null);
+
+        deviceCapInfo.updateRcsAssociatedUri(uris);
+        outUri = deviceCapInfo.getImsAssociatedUri(true);
+
+        assertNull(outUri);
+    }
+
+    private DeviceCapabilityInfo createDeviceCapabilityInfo() {
+        DeviceCapabilityInfo deviceCapInfo = new DeviceCapabilityInfo(mSubId, null);
+        return deviceCapInfo;
+    }
+
+}
+\ No newline at end of file