tests/src/com/android/ims/rcs/uce/presence/publish/DeviceCapabilityInfoTest.java

/*
 * 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, this method will still return a SIP URI, since there are no TEL
        // URIs found in the list.
        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);

        numbers = outUri.getSchemeSpecificPart();
        numberParts = numbers.split("[@;:]");
        number = numberParts[0];

        assertEquals(number, telNumber);

    }

    private DeviceCapabilityInfo createDeviceCapabilityInfo() {
        DeviceCapabilityInfo deviceCapInfo = new DeviceCapabilityInfo(mSubId, null);
        return deviceCapInfo;
    }

}