session-android/test/androidTest/java/org/thoughtcrime/securesms/database/AttachmentDatabaseTest.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;
  }
}