Home | History | Annotate | Download | only in telephony 
      1 /*
      2  * Copyright (C) 2018 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License
     15  */
     16 
     17 package com.android.internal.telephony;
     18 
     19 import static org.junit.Assert.assertEquals;
     20 import static org.junit.Assert.assertNull;
     21 import static org.mockito.Matchers.any;
     22 import static org.mockito.Mockito.doNothing;
     23 import static org.mockito.Mockito.doReturn;
     24 import static org.mockito.Mockito.spy;
     25 import static org.mockito.Mockito.verify;
     26 
     27 import android.os.HandlerThread;
     28 import android.test.suitebuilder.annotation.SmallTest;
     29 
     30 import org.junit.After;
     31 import org.junit.Before;
     32 import org.junit.Test;
     33 import org.mockito.Mock;
     34 
     35 public class ImsSmsDispatcherTest extends TelephonyTest {
     36     @Mock private SmsDispatchersController mSmsDispatchersController;
     37     @Mock private SMSDispatcher.SmsTracker mSmsTracker;
     38     private ImsSmsDispatcher mImsSmsDispatcher;
     39     private ImsSmsDispatcherTestHandler mImsSmsDispatcherTestHandler;
     40 
     41 
     42     private class ImsSmsDispatcherTestHandler extends HandlerThread {
     43 
     44         private ImsSmsDispatcherTestHandler(String name) {
     45             super(name);
     46         }
     47 
     48         @Override
     49         public void onLooperPrepared() {
     50             mImsSmsDispatcher = spy(new ImsSmsDispatcher(mPhone, mSmsDispatchersController));
     51             setReady(true);
     52         }
     53     }
     54 
     55     @Before
     56     public void setUp() throws Exception {
     57         super.setUp(getClass().getSimpleName());
     58 
     59         mImsSmsDispatcherTestHandler = new ImsSmsDispatcherTestHandler(TAG);
     60         mImsSmsDispatcherTestHandler.start();
     61         waitUntilReady();
     62     }
     63 
     64     @Test @SmallTest
     65     public void testSendSms() {
     66         int token = mImsSmsDispatcher.mNextToken.get();
     67         int trackersSize = mImsSmsDispatcher.mTrackers.size();
     68 
     69         mImsSmsDispatcher.sendSms(mSmsTracker);
     70 
     71         assertEquals(token + 1, mImsSmsDispatcher.mNextToken.get());
     72         assertEquals(trackersSize + 1, mImsSmsDispatcher.mTrackers.size());
     73     }
     74 
     75     @Test @SmallTest
     76     public void testFallbackToPstn() {
     77         int token = 0;
     78         mImsSmsDispatcher.mTrackers.put(token, mSmsTracker);
     79         doNothing().when(mSmsDispatchersController).sendRetrySms(mSmsTracker);
     80 
     81         mImsSmsDispatcher.fallbackToPstn(token, mSmsTracker);
     82 
     83         verify(mSmsDispatchersController).sendRetrySms(mSmsTracker);
     84         assertNull(mImsSmsDispatcher.mTrackers.get(token));
     85     }
     86 
     87     @After
     88     public void tearDown() throws Exception {
     89         mImsSmsDispatcher = null;
     90         mImsSmsDispatcherTestHandler.quit();
     91         super.tearDown();
     92     }
     93 }
     94