| /* |
| * Copyright (C) 2023 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.providers.telephony; |
| |
| import static org.mockito.ArgumentMatchers.anyInt; |
| import static org.mockito.ArgumentMatchers.anyString; |
| import static org.mockito.ArgumentMatchers.eq; |
| import static org.mockito.Mockito.mock; |
| import static org.mockito.Mockito.when; |
| |
| import android.app.AppOpsManager; |
| import android.content.ContentValues; |
| import android.content.Context; |
| import android.content.pm.PackageManager; |
| import android.content.pm.ProviderInfo; |
| import android.database.ContentObserver; |
| import android.database.Cursor; |
| import android.net.Uri; |
| import android.os.UserManager; |
| import android.provider.Telephony; |
| import android.telephony.TelephonyManager; |
| import android.test.mock.MockContentResolver; |
| import android.util.Log; |
| |
| import junit.framework.TestCase; |
| |
| import org.junit.Test; |
| import org.mockito.Mock; |
| import org.mockito.MockitoAnnotations; |
| |
| public class MmsProviderTest extends TestCase { |
| private static final String TAG = "MmsProviderTest"; |
| |
| @Mock private Context mContext; |
| private MockContentResolver mContentResolver; |
| private MmsProviderTestable mMmsProviderTestable; |
| @Mock private PackageManager mPackageManager; |
| |
| private int notifyChangeCount; |
| private UserManager mUserManager; |
| |
| @Override |
| protected void setUp() throws Exception { |
| super.setUp(); |
| MockitoAnnotations.initMocks(this); |
| mMmsProviderTestable = new MmsProviderTestable(); |
| mUserManager = mock(UserManager.class); |
| |
| // setup mocks |
| when(mContext.getSystemService(eq(Context.APP_OPS_SERVICE))) |
| .thenReturn(mock(AppOpsManager.class)); |
| when(mContext.getSystemService(eq(Context.TELEPHONY_SERVICE))) |
| .thenReturn(mock(TelephonyManager.class)); |
| when(mContext.getSystemService(eq(Context.USER_SERVICE))) |
| .thenReturn(mUserManager); |
| |
| when(mContext.checkCallingOrSelfPermission(anyString())) |
| .thenReturn(PackageManager.PERMISSION_GRANTED); |
| when(mContext.getUserId()).thenReturn(0); |
| when(mContext.getPackageManager()).thenReturn(mPackageManager); |
| |
| /** |
| * This is used to give the MmsProviderTest a mocked context which takes a |
| * SmsProvider and attaches it to the ContentResolver with telephony authority. |
| * The mocked context also gives WRITE_APN_SETTINGS permissions |
| */ |
| mContentResolver = new MockContentResolver() { |
| @Override |
| public void notifyChange(Uri uri, ContentObserver observer, boolean syncToNetwork, |
| int userHandle) { |
| notifyChangeCount++; |
| } |
| }; |
| when(mContext.getContentResolver()).thenReturn(mContentResolver); |
| |
| // Add authority="mms" to given mmsProvider |
| ProviderInfo providerInfo = new ProviderInfo(); |
| providerInfo.authority = "mms"; |
| |
| // Add context to given mmsProvider |
| mMmsProviderTestable.attachInfoForTesting(mContext, providerInfo); |
| Log.d(TAG, "MockContextWithProvider: mmsProvider.getContext(): " |
| + mMmsProviderTestable.getContext()); |
| |
| // Add given MmsProvider to mResolver with authority="mms" so that |
| // mResolver can send queries to mMmsProvider |
| mContentResolver.addProvider("mms", mMmsProviderTestable); |
| Log.d(TAG, "MockContextWithProvider: Add MmsProvider to mResolver"); |
| notifyChangeCount = 0; |
| } |
| |
| @Override |
| protected void tearDown() throws Exception { |
| super.tearDown(); |
| mMmsProviderTestable.closeDatabase(); |
| } |
| |
| @Test |
| public void testInsertMms() { |
| final ContentValues values = new ContentValues(); |
| values.put(Telephony.Mms.READ, 1); |
| values.put(Telephony.Mms.SEEN, 1); |
| values.put(Telephony.Mms.SUBSCRIPTION_ID, 1); |
| values.put(Telephony.Mms.MESSAGE_BOX, Telephony.Mms.MESSAGE_BOX_ALL); |
| values.put(Telephony.Mms.TEXT_ONLY, 1); |
| values.put(Telephony.Mms.THREAD_ID, 1); |
| |
| Uri expected = Uri.parse("content://mms/1"); |
| Uri actual = mContentResolver.insert(Telephony.Mms.CONTENT_URI, values); |
| |
| assertEquals(expected, actual); |
| assertEquals(1, notifyChangeCount); |
| } |
| |
| @Test |
| public void testInsertUsingManagedProfile() { |
| when(mUserManager.isManagedProfile(anyInt())).thenReturn(true); |
| |
| try { |
| assertNull(mContentResolver.insert(Telephony.Mms.CONTENT_URI, null)); |
| } catch (Exception e) { |
| Log.d(TAG, "Error inserting mms: " + e); |
| } |
| } |
| |
| @Test |
| public void testQueryUsingManagedProfile() { |
| when(mUserManager.isManagedProfile(anyInt())).thenReturn(true); |
| |
| try (Cursor cursor = mContentResolver.query(Telephony.Mms.CONTENT_URI, |
| null, null, null, null)) { |
| assertEquals(0, cursor.getCount()); |
| } catch (Exception e) { |
| Log.d(TAG, "Exception in getting count: " + e); |
| } |
| } |
| |
| @Test |
| public void testUpdateUsingManagedProfile() { |
| when(mUserManager.isManagedProfile(anyInt())).thenReturn(true); |
| |
| try { |
| assertEquals(0, mContentResolver.update(Telephony.Mms.CONTENT_URI, null, null, null)); |
| } catch (Exception e) { |
| Log.d(TAG, "Exception in updating mms: " + e); |
| } |
| } |
| |
| @Test |
| public void testDeleteUsingManagedProfile() { |
| when(mUserManager.isManagedProfile(anyInt())).thenReturn(true); |
| |
| try { |
| assertEquals(0, mContentResolver.delete(Telephony.Mms.CONTENT_URI, null, null)); |
| } catch (Exception e) { |
| Log.d(TAG, "Exception in deleting mms: " + e); |
| } |
| } |
| } |