79 lines
3.3 KiB
Java
79 lines
3.3 KiB
Java
package org.thoughtcrime.securesms.database;
|
|
|
|
import android.net.Uri;
|
|
|
|
import org.thoughtcrime.securesms.TextSecureTestCase;
|
|
import org.thoughtcrime.securesms.attachments.AttachmentId;
|
|
import org.thoughtcrime.securesms.attachments.DatabaseAttachment;
|
|
import org.thoughtcrime.securesms.crypto.MasterSecret;
|
|
import org.thoughtcrime.securesms.util.BitmapDecodingException;
|
|
|
|
import java.io.FileNotFoundException;
|
|
import java.io.InputStream;
|
|
|
|
import static org.mockito.Matchers.any;
|
|
import static org.mockito.Matchers.anyFloat;
|
|
import static org.mockito.Matchers.eq;
|
|
import static org.mockito.Mockito.doNothing;
|
|
import static org.mockito.Mockito.doReturn;
|
|
import static org.mockito.Mockito.mock;
|
|
import static org.mockito.Mockito.never;
|
|
import static org.mockito.Mockito.spy;
|
|
import static org.mockito.Mockito.verify;
|
|
import static org.mockito.Mockito.when;
|
|
|
|
public class AttachmentDatabaseTest extends TextSecureTestCase {
|
|
private static final long ROW_ID = 1L;
|
|
private static final long UNIQUE_ID = 2L;
|
|
|
|
private AttachmentDatabase database;
|
|
|
|
@Override
|
|
public void setUp() {
|
|
super.setUp();
|
|
database = spy(DatabaseFactory.getAttachmentDatabase(getInstrumentation().getTargetContext()));
|
|
}
|
|
|
|
public void testThumbnailGenerationTaskNotRunWhenThumbnailExists() throws Exception {
|
|
final AttachmentId attachmentId = new AttachmentId(ROW_ID, UNIQUE_ID);
|
|
|
|
DatabaseAttachment mockAttachment = getMockAttachment("x/x");
|
|
when(database.getAttachment(attachmentId)).thenReturn(mockAttachment);
|
|
|
|
InputStream mockInputStream = mock(InputStream.class);
|
|
doReturn(mockInputStream).when(database).getDataStream(any(MasterSecret.class), any(AttachmentId.class), eq("thumbnail"));
|
|
database.getThumbnailStream(mock(MasterSecret.class), attachmentId);
|
|
|
|
// Works as the Future#get() call in AttachmentDatabase#getThumbnailStream() makes updating synchronous
|
|
verify(database, never()).updateAttachmentThumbnail(any(MasterSecret.class), any(AttachmentId.class), any(InputStream.class), anyFloat());
|
|
}
|
|
|
|
public void testThumbnailGenerationTaskRunWhenThumbnailMissing() throws Exception {
|
|
final AttachmentId attachmentId = new AttachmentId(ROW_ID, UNIQUE_ID);
|
|
|
|
DatabaseAttachment mockAttachment = getMockAttachment("image/png");
|
|
when(database.getAttachment(attachmentId)).thenReturn(mockAttachment);
|
|
|
|
doReturn(null).when(database).getDataStream(any(MasterSecret.class), any(AttachmentId.class), eq("thumbnail"));
|
|
doNothing().when(database).updateAttachmentThumbnail(any(MasterSecret.class), any(AttachmentId.class), any(InputStream.class), anyFloat());
|
|
|
|
try {
|
|
database.new ThumbnailFetchCallable(mock(MasterSecret.class), attachmentId).call();
|
|
throw new AssertionError("Didn't try to generate thumbnail");
|
|
} catch (BitmapDecodingException bde) {
|
|
if (!(bde.getCause() instanceof FileNotFoundException)) {
|
|
throw new AssertionError("Thumbnail generation failed for another reason than a FileNotFoundException: " + bde.getMessage());
|
|
} // else success
|
|
}
|
|
}
|
|
|
|
private DatabaseAttachment getMockAttachment(String contentType) {
|
|
DatabaseAttachment attachment = mock(DatabaseAttachment.class);
|
|
when(attachment.getContentType()).thenReturn(contentType);
|
|
when(attachment.getDataUri()).thenReturn(Uri.EMPTY);
|
|
when(attachment.hasData()).thenReturn(true);
|
|
|
|
return attachment;
|
|
}
|
|
}
|