tests/tests/telephony/sdk28/src/android/telephony/sdk28/cts/CellInfoTest.java

/*
 * Copyright (C) 2019 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 android.telephony.sdk28.cts;

import static androidx.test.InstrumentationRegistry.getContext;

import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;

import android.content.Context;
import android.content.pm.PackageManager;
import android.telephony.AccessNetworkConstants;
import android.telephony.CellInfo;
import android.telephony.NetworkRegistrationInfo;
import android.telephony.ServiceState;
import android.telephony.TelephonyManager;
import android.util.Log;

import com.android.compatibility.common.util.ShellIdentityUtils;

import org.junit.Before;
import org.junit.Test;

import java.util.List;

public class CellInfoTest {
    private static final String TAG = "CellInfoTest";

    private static final int MAX_WAIT_SECONDS = 15;
    private static final int POLL_INTERVAL_MILLIS = 1000;

    private static final String[] sPermissions = new String[] {
            android.Manifest.permission.READ_PHONE_STATE,
            android.Manifest.permission.ACCESS_COARSE_LOCATION};

    private TelephonyManager mTm;
    private PackageManager mPm;

    private boolean isCamped() {
        ServiceState ss = ShellIdentityUtils.invokeMethodWithShellPermissions(
                mTm, TelephonyManager::getServiceState);

        if (ss == null) return false;
        if (ss.getState() == ServiceState.STATE_EMERGENCY_ONLY) return true;
        List<NetworkRegistrationInfo> nris = ss.getNetworkRegistrationInfoList();
        for (NetworkRegistrationInfo nri : nris) {
            if (nri.getTransportType() != AccessNetworkConstants.TRANSPORT_TYPE_WWAN) continue;
            if (nri.isRegistered()) return true;
        }
        return false;
    }

    @Before
    public void setUp() throws Exception {
        mTm = (TelephonyManager) getContext().getSystemService(Context.TELEPHONY_SERVICE);
        mPm = getContext().getPackageManager();

        for (String permission : sPermissions) {
            assertTrue("Something (not this test) has denied needed permission=" + permission,
                    getContext().checkSelfPermission(permission)
                            == android.content.pm.PackageManager.PERMISSION_GRANTED);
        }
    }

    @Test
    public void testCellInfoSdk28() {
        if (!mPm.hasSystemFeature(PackageManager.FEATURE_TELEPHONY)) {
            Log.d(TAG, "Skipping test that requires FEATURE_TELEPHONY");
            return;
        }

        if (!isCamped()) fail("Device is not camped to a cell");

        List<CellInfo> cellInfo = ShellIdentityUtils.invokeMethodWithShellPermissions(
                mTm, TelephonyManager::getAllCellInfo);

        // getAllCellInfo should never return null, and there should be at least one entry.
        assertNotNull("TelephonyManager.getAllCellInfo() returned NULL CellInfo", cellInfo);
        assertFalse("TelephonyManager.getAllCellInfo() returned an empty list", cellInfo.isEmpty());

        final long initialTime = cellInfo.get(0).getTimeStamp();

        for(int i = 0; i < MAX_WAIT_SECONDS; i++) {
            try {
                Thread.sleep(POLL_INTERVAL_MILLIS); // 1 second
            } catch (InterruptedException ie) {
                fail("Thread was interrupted");
            }
            List<CellInfo> newCellInfo = ShellIdentityUtils.invokeMethodWithShellPermissions(
                    mTm, TelephonyManager::getAllCellInfo);
            assertNotNull("TelephonyManager.getAllCellInfo() returned NULL CellInfo", newCellInfo);
            assertFalse("TelephonyManager.getAllCellInfo() returned an empty list",
                    newCellInfo.isEmpty());
            // Test that new CellInfo has been retrieved from the modem
            if (newCellInfo.get(0).getTimeStamp() != initialTime) return;
        }
        fail("CellInfo failed to update after " + MAX_WAIT_SECONDS + " seconds.");
    }
}