From 6a1bbedae86c52bf198ce97b2e209c87add01ee2 Mon Sep 17 00:00:00 2001 From: Rhodey Orbits Date: Mon, 25 May 2015 15:23:04 -0700 Subject: [PATCH] move magic SMS mocking code into own util class // FREEBIE --- .../securesms/service/SmsListenerTest.java | 71 +------------- .../thoughtcrime/securesms/util/SmsUtil.java | 94 +++++++++++++++++++ 2 files changed, 96 insertions(+), 69 deletions(-) create mode 100644 test/androidTest/java/org/thoughtcrime/securesms/util/SmsUtil.java diff --git a/test/androidTest/java/org/thoughtcrime/securesms/service/SmsListenerTest.java b/test/androidTest/java/org/thoughtcrime/securesms/service/SmsListenerTest.java index b08de85212..85fddd4a46 100644 --- a/test/androidTest/java/org/thoughtcrime/securesms/service/SmsListenerTest.java +++ b/test/androidTest/java/org/thoughtcrime/securesms/service/SmsListenerTest.java @@ -17,22 +17,14 @@ package org.thoughtcrime.securesms.service; -import android.annotation.SuppressLint; import android.content.Context; import android.content.Intent; import android.content.SharedPreferences; -import android.os.Bundle; -import android.provider.Telephony; -import android.telephony.PhoneNumberUtils; import android.util.Log; import org.mockito.ArgumentCaptor; import org.thoughtcrime.securesms.TextSecureTestCase; - -import java.io.ByteArrayOutputStream; -import java.lang.reflect.Method; -import java.util.Calendar; -import java.util.GregorianCalendar; +import org.thoughtcrime.securesms.util.SmsUtil; import static org.mockito.Matchers.anyBoolean; import static org.mockito.Matchers.anyInt; @@ -65,65 +57,6 @@ public class SmsListenerTest extends TextSecureTestCase { CHALLENGE_4_4, CHALLENGE_4_4, CHALLENGE_4_4, }; - /* - credit :D - http://stackoverflow.com/a/12338541 - */ - @SuppressWarnings("unchecked") - private static byte[] buildSmsPdu(String sender, String body) throws Exception{ - byte[] scBytes = PhoneNumberUtils.networkPortionToCalledPartyBCD("0000000000"); - byte[] senderBytes = PhoneNumberUtils.networkPortionToCalledPartyBCD(sender); - int lsmcs = scBytes.length; - byte[] dateBytes = new byte[7]; - Calendar calendar = new GregorianCalendar(); - - dateBytes[0] = reverseByte((byte) (calendar.get(Calendar.YEAR))); - dateBytes[1] = reverseByte((byte) (calendar.get(Calendar.MONTH) + 1)); - dateBytes[2] = reverseByte((byte) (calendar.get(Calendar.DAY_OF_MONTH))); - dateBytes[3] = reverseByte((byte) (calendar.get(Calendar.HOUR_OF_DAY))); - dateBytes[4] = reverseByte((byte) (calendar.get(Calendar.MINUTE))); - dateBytes[5] = reverseByte((byte) (calendar.get(Calendar.SECOND))); - dateBytes[6] = reverseByte((byte) ((calendar.get(Calendar.ZONE_OFFSET) + calendar.get(Calendar.DST_OFFSET)) / (60 * 1000 * 15))); - - ByteArrayOutputStream bo = new ByteArrayOutputStream(); - bo.write(lsmcs); - bo.write(scBytes); - bo.write(0x04); - bo.write((byte) sender.length()); - bo.write(senderBytes); - bo.write(0x00); - bo.write(0x00); - bo.write(dateBytes); - - String sReflectedClassName = "com.android.internal.telephony.GsmAlphabet"; - Class cReflectedNFCExtras = Class.forName(sReflectedClassName); - Method stringToGsm7BitPacked = cReflectedNFCExtras.getMethod("stringToGsm7BitPacked", new Class[] { String.class }); - - stringToGsm7BitPacked.setAccessible(true); - byte[] bodybytes = (byte[]) stringToGsm7BitPacked.invoke(null, body); - bo.write(bodybytes); - - return bo.toByteArray(); - } - - private static byte reverseByte(byte b) { - return (byte) ((b & 0xF0) >> 4 | (b & 0x0F) << 4); - } - - @SuppressLint("NewApi") - private Intent buildSmsReceivedIntent(String smsBody) throws Exception { - final Intent smsIntent = mock(Intent.class); - final Bundle smsExtras = new Bundle(); - final byte[] smsPdu = buildSmsPdu("15555555555", smsBody); - - smsExtras.putSerializable("pdus", new Object[]{smsPdu}); - - when(smsIntent.getAction()).thenReturn(Telephony.Sms.Intents.SMS_RECEIVED_ACTION); - when(smsIntent.getExtras()).thenReturn(smsExtras); - - return smsIntent; - } - public void testReceiveChallenges() throws Exception { final SmsListener smsListener = new SmsListener(); @@ -140,7 +73,7 @@ public class SmsListenerTest extends TextSecureTestCase { when(mockPreferences.getBoolean(contains("pref_verifying"), anyBoolean())).thenReturn(true); try { - smsListener.onReceive(mockContext, buildSmsReceivedIntent(CHALLENGE_SMS)); + smsListener.onReceive(mockContext, SmsUtil.buildSmsReceivedIntent("15555555555", (CHALLENGE_SMS))); } catch (IllegalStateException e) { Log.d(getClass().getName(), "some api levels are picky with abortBroadcast()"); } diff --git a/test/androidTest/java/org/thoughtcrime/securesms/util/SmsUtil.java b/test/androidTest/java/org/thoughtcrime/securesms/util/SmsUtil.java new file mode 100644 index 0000000000..6ac5ca5c6f --- /dev/null +++ b/test/androidTest/java/org/thoughtcrime/securesms/util/SmsUtil.java @@ -0,0 +1,94 @@ +/* + * Copyright (C) 2015 Open Whisper Systems + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ +package org.thoughtcrime.securesms.util; + +import android.annotation.SuppressLint; +import android.content.Intent; +import android.os.Bundle; +import android.provider.Telephony; +import android.telephony.PhoneNumberUtils; + +import java.io.ByteArrayOutputStream; +import java.lang.reflect.Method; +import java.util.Calendar; +import java.util.GregorianCalendar; + +import static org.mockito.Mockito.mock; +import static org.mockito.Mockito.when; + +public class SmsUtil { + + private static byte reverseByte(byte b) { + return (byte) ((b & 0xF0) >> 4 | (b & 0x0F) << 4); + } + + /* + credit :D + http://stackoverflow.com/a/12338541 + */ + @SuppressWarnings("unchecked") + public static byte[] buildSmsPdu(String senderPstnNumber, String body) throws Exception{ + byte[] scBytes = PhoneNumberUtils.networkPortionToCalledPartyBCD("0000000000"); + byte[] senderBytes = PhoneNumberUtils.networkPortionToCalledPartyBCD(senderPstnNumber); + int lsmcs = scBytes.length; + byte[] dateBytes = new byte[7]; + Calendar calendar = new GregorianCalendar(); + + dateBytes[0] = reverseByte((byte) (calendar.get(Calendar.YEAR))); + dateBytes[1] = reverseByte((byte) (calendar.get(Calendar.MONTH) + 1)); + dateBytes[2] = reverseByte((byte) (calendar.get(Calendar.DAY_OF_MONTH))); + dateBytes[3] = reverseByte((byte) (calendar.get(Calendar.HOUR_OF_DAY))); + dateBytes[4] = reverseByte((byte) (calendar.get(Calendar.MINUTE))); + dateBytes[5] = reverseByte((byte) (calendar.get(Calendar.SECOND))); + dateBytes[6] = reverseByte((byte) ((calendar.get(Calendar.ZONE_OFFSET) + calendar.get(Calendar.DST_OFFSET)) / (60 * 1000 * 15))); + + ByteArrayOutputStream bo = new ByteArrayOutputStream(); + bo.write(lsmcs); + bo.write(scBytes); + bo.write(0x04); + bo.write((byte) senderPstnNumber.length()); + bo.write(senderBytes); + bo.write(0x00); + bo.write(0x00); + bo.write(dateBytes); + + String sReflectedClassName = "com.android.internal.telephony.GsmAlphabet"; + Class cReflectedNFCExtras = Class.forName(sReflectedClassName); + Method stringToGsm7BitPacked = cReflectedNFCExtras.getMethod("stringToGsm7BitPacked", new Class[] { String.class }); + + stringToGsm7BitPacked.setAccessible(true); + byte[] bodybytes = (byte[]) stringToGsm7BitPacked.invoke(null, body); + bo.write(bodybytes); + + return bo.toByteArray(); + } + + @SuppressLint("NewApi") + public static Intent buildSmsReceivedIntent(String senderPstnNumber, String smsBody) throws Exception { + final Intent smsIntent = mock(Intent.class); + final Bundle smsExtras = new Bundle(); + final byte[] smsPdu = SmsUtil.buildSmsPdu(senderPstnNumber, smsBody); + + smsExtras.putSerializable("pdus", new Object[]{smsPdu}); + + when(smsIntent.getAction()).thenReturn(Telephony.Sms.Intents.SMS_RECEIVED_ACTION); + when(smsIntent.getExtras()).thenReturn(smsExtras); + + return smsIntent; + } + +}