// FREEBIE
This commit is contained in:
Moxie Marlinspike 2015-10-23 12:24:43 -07:00
parent 91785e5590
commit 23a2a5e5f3

View File

@ -5,12 +5,11 @@ import android.content.Context;
import org.junit.Before;
import org.junit.Test;
import org.thoughtcrime.securesms.BaseUnitTest;
import org.thoughtcrime.securesms.database.AttachmentDatabase.AttachmentId;
import org.thoughtcrime.securesms.attachments.Attachment;
import org.thoughtcrime.securesms.attachments.AttachmentId;
import org.thoughtcrime.securesms.jobs.AttachmentDownloadJob.InvalidPartException;
import org.thoughtcrime.securesms.util.Util;
import ws.com.google.android.mms.pdu.PduPart;
import static org.mockito.Mockito.when;
import static org.powermock.api.mockito.PowerMockito.mock;
public class AttachmentDownloadJobTest extends BaseUnitTest {
@ -25,15 +24,19 @@ public class AttachmentDownloadJobTest extends BaseUnitTest {
@Test(expected = InvalidPartException.class)
public void testCreateAttachmentPointerInvalidId() throws Exception {
PduPart part = new PduPart();
part.setContentDisposition(Util.toIsoBytes("a long and acceptable valid key like we all want"));
job.createAttachmentPointer(masterSecret, part);
Attachment attachment = mock(Attachment.class);
when(attachment.getLocation()).thenReturn(null);
when(attachment.getKey()).thenReturn("a long and acceptable valid key like we all want");
job.createAttachmentPointer(masterSecret, attachment);
}
@Test(expected = InvalidPartException.class)
public void testCreateAttachmentPointerInvalidKey() throws Exception {
PduPart part = new PduPart();
part.setContentDisposition(Util.toIsoBytes("1"));
job.createAttachmentPointer(masterSecret, part);
Attachment attachment = mock(Attachment.class);
when(attachment.getLocation()).thenReturn("something");
when(attachment.getKey()).thenReturn(null);
job.createAttachmentPointer(masterSecret, attachment);
}
}