package net.sf.jabref.logic.importer.fetcher;

import java.io.IOException;
import java.net.URL;
import java.util.Collections;
import java.util.Optional;

import net.sf.jabref.logic.importer.FetcherException;
import net.sf.jabref.logic.importer.ImportFormatPreferences;
import net.sf.jabref.model.entry.BibEntry;
import net.sf.jabref.model.entry.BibLatexEntryTypes;
import net.sf.jabref.testutils.category.FetcherTests;

import org.junit.Assert;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.rules.ExpectedException;

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

@Category(FetcherTests.class)
public class ArXivTest {

    @Rule public ExpectedException expectedException = ExpectedException.none();
    private ArXiv finder;
    private BibEntry entry;
    private BibEntry sliceTheoremPaper;

    @Before
    public void setUp() {
        ImportFormatPreferences importFormatPreferences = mock(ImportFormatPreferences.class);
        when(importFormatPreferences.getKeywordSeparator()).thenReturn(',');
        finder = new ArXiv(importFormatPreferences);
        entry = new BibEntry();

        sliceTheoremPaper = new BibEntry();
        sliceTheoremPaper.setType(BibLatexEntryTypes.ARTICLE);
        sliceTheoremPaper.setField("author", "Tobias Diez");
        sliceTheoremPaper.setField("title", "Slice theorem for Fréchet group actions and covariant symplectic field theory");
        sliceTheoremPaper.setField("date", "2014-05-09");
        sliceTheoremPaper.setField("abstract", "A general slice theorem for the action of a Fr\\'echet Lie group on a Fr\\'echet manifolds is established. The Nash-Moser theorem provides the fundamental tool to generalize the result of Palais to this infinite-dimensional setting. The presented slice theorem is illustrated by its application to gauge theories: the action of the gauge transformation group admits smooth slices at every point and thus the gauge orbit space is stratified by Fr\\'echet manifolds. Furthermore, a covariant and symplectic formulation of classical field theory is proposed and extensively discussed. At the root of this novel framework is the incorporation of field degrees of freedom F and spacetime M into the product manifold F * M. The induced bigrading of differential forms is used in order to carry over the usual symplectic theory to this new setting. The examples of the Klein-Gordon field and general Yang-Mills theory illustrate that the presented approach conveniently handles the occurring symmetries.");
        sliceTheoremPaper.setField("eprint", "1405.2249v1");
        sliceTheoremPaper.setField("file", "online:http\\://arxiv.org/pdf/1405.2249v1:PDF");
        sliceTheoremPaper.setField("eprinttype", "arXiv");
        sliceTheoremPaper.setField("eprintclass", "math-ph");
        sliceTheoremPaper.setField("keywords", "math-ph, math.DG, math.MP, math.SG, 58B99, 58Z05, 58B25, 22E65, 58D19, 53D20, 53D42");
    }

    @Test
    public void doiNotPresent() throws IOException {
        assertEquals(Optional.empty(), finder.findFullText(entry));
    }

    @Test(expected = NullPointerException.class)
    public void rejectNullParameter() throws IOException {
        finder.findFullText(null);
        Assert.fail();
    }

    @Test
    public void findByDOI() throws IOException {
        entry.setField("doi", "10.1529/biophysj.104.047340");

        assertEquals(Optional.of(new URL("http://arxiv.org/pdf/cond-mat/0406246v1")), finder.findFullText(entry));
    }

    @Test
    public void findByEprint() throws IOException {
        entry.setField("eprint", "1603.06570");

        assertEquals(Optional.of(new URL("http://arxiv.org/pdf/1603.06570v1")), finder.findFullText(entry));
    }

    @Test
    public void findByEprintWithUnknownDOI() throws IOException {
        entry.setField("doi", "10.1529/unknown");
        entry.setField("eprint", "1603.06570");

        assertEquals(Optional.of(new URL("http://arxiv.org/pdf/1603.06570v1")), finder.findFullText(entry));
    }

    @Test
    public void notFoundByUnknownDOI() throws IOException {
        entry.setField("doi", "10.1529/unknown");

        assertEquals(Optional.empty(), finder.findFullText(entry));
    }

    @Test
    public void notFoundByUnknownId() throws IOException {
        entry.setField("eprint", "1234.12345");

        assertEquals(Optional.empty(), finder.findFullText(entry));
    }

    @Test
    public void searchEntryByPartOfTitle() throws Exception {
        assertEquals(Collections.singletonList(sliceTheoremPaper),
                finder.performSearch("ti:\"slice theorem for Frechet\""));
    }

    @Test
    public void searchEntryByPartOfTitleWithAcuteAccent() throws Exception {
        assertEquals(Collections.singletonList(sliceTheoremPaper),
                finder.performSearch("ti:\"slice theorem for Fréchet\""));
    }

    @Test
    public void searchEntryByOldId() throws Exception {
        BibEntry expected = new BibEntry();
        expected.setType(BibLatexEntryTypes.ARTICLE);
        expected.setField("author", "H1 Collaboration");
        expected.setField("title", "Multi-Electron Production at High Transverse Momenta in ep Collisions at HERA");
        expected.setField("date", "2003-07-07");
        expected.setField("abstract", "Multi-electron production is studied at high electron transverse momentum in positron- and electron-proton collisions using the H1 detector at HERA. The data correspond to an integrated luminosity of 115 pb-1. Di-electron and tri-electron event yields are measured. Cross sections are derived in a restricted phase space region dominated by photon-photon collisions. In general good agreement is found with the Standard Model predictions. However, for electron pair invariant masses above 100 GeV, three di-electron events and three tri-electron events are observed, compared to Standard Model expectations of 0.30 \\pm 0.04 and 0.23 \\pm 0.04, respectively.");
        expected.setField("eprint", "hep-ex/0307015v1");
        expected.setField("file", "online:http\\://arxiv.org/pdf/hep-ex/0307015v1:PDF");
        expected.setField("eprinttype", "arXiv");
        expected.setField("eprintclass", "hep-ex");
        expected.setField("keywords", "hep-ex");
        expected.setField("doi", "10.1140/epjc/s2003-01326-x");
        expected.setField("journaltitle", "Eur.Phys.J.C31:17-29,2003");

        assertEquals(Optional.of(expected), finder.performSearchById("hep-ex/0307015"));
    }

    @Test
    public void searchEntryByIdWith4DigitsAndVersion() throws Exception {
        assertEquals(Optional.of(sliceTheoremPaper), finder.performSearchById("1405.2249v1"));
    }

    @Test
    public void searchEntryByIdWith4Digits() throws Exception {
        assertEquals(Optional.of(sliceTheoremPaper), finder.performSearchById("1405.2249"));
    }

    @Test
    public void searchEntryByIdWith5Digits() throws Exception {
        assertEquals(Optional.of(
                "An Optimal Convergence Theorem for Mean Curvature Flow of Arbitrary Codimension in Hyperbolic Spaces"),
                finder.performSearchById("1503.06747").flatMap(entry -> entry.getField("title")));
    }

    @Test
    public void searchWithMalformedIdThrowsException() throws Exception {
        expectedException.expect(FetcherException.class);
        expectedException.expectMessage("incorrect id format");
        finder.performSearchById("123412345");
    }
}
