DownloadEmailUtilsTest.java

package org.gbif.occurrence.download.service;

import org.gbif.api.model.common.DOI;
import org.gbif.api.model.occurrence.Download;
import org.gbif.api.model.occurrence.DownloadFormat;
import org.gbif.api.model.occurrence.DownloadRequest;
import org.gbif.occurrence.query.TitleLookup;

import java.util.Date;

import org.junit.Test;
import org.mockito.Matchers;

import static org.junit.Assert.assertNotNull;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

public class DownloadEmailUtilsTest {

  @Test
  public void testSendSuccessNotificationMail() throws Exception {
    TitleLookup tl = mock(TitleLookup.class);
    when(tl.getDatasetTitle(Matchers.<String>any())).thenReturn("The little Mermaid");
    when(tl.getSpeciesName(Matchers.<String>any())).thenReturn("Abies alba Mill.");

    DownloadEmailUtils utils = new DownloadEmailUtils("1@mailinator.com, 2@mailinator.com", "http:///www.gbif.org", null, null, tl);
    Download d = new Download();
    d.setKey("0007082-141215154445624");
    d.setDoi(new DOI("10.5072/dl.j9spoa"));
    d.setDownloadLink("http://api.gbif.org/v1/occurrence/download/request/0007082-141215154445624.zip");
    d.setCreated(new Date());
    d.setStatus(Download.Status.SUCCEEDED);
    d.setModified(new Date());
    d.setNumberDatasets(3);
    d.setSize(1787823);
    d.setTotalRecords(8792);

    DownloadRequest req = new DownloadRequest();
    req.setFormat(DownloadFormat.SIMPLE_CSV);
    req.setCreator("markus");
    d.setRequest(req);
    String body = utils.buildBody(d, "success.ftl");

    //System.out.println(body);
    assertNotNull(body);


    d.setStatus(Download.Status.FAILED);
    body = utils.buildBody(d, "error.ftl");
    //System.out.println(body);
    assertNotNull(body);
  }
}