D-Net

package eu.dnetlib.wds.parser;

import com.google.gson.Gson;

import com.google.gson.GsonBuilder;

import eu.dnetlib.enabling.locators.UniqueServiceLocator;

import eu.dnetlib.rmi.enabling.ISLookUpService;

import eu.dnetlib.springutils.stringtemplate.StringTemplateFactory;

import eu.dnetlib.wds.resolver.OpenAireParser;

import eu.dnetlib.wds.resolver.WDSResolvedObject;

import eu.dnetlib.wds.resolver.serializers.WDSResolverSerializer;

import eu.dnetlib.wds.utils.WDSUtils;

import org.antlr.stringtemplate.StringTemplate;

import org.apache.commons.io.IOUtils;

import static org.junit.Assert.*;

import static org.mockito.Mockito.mock;

import static org.mockito.Mockito.when;

import org.junit.Assert;

import org.junit.Before;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.Mock;

import org.mockito.runners.MockitoJUnitRunner;

import java.io.IOException;

import java.io.InputStream;

import java.util.Arrays;

import java.util.List;

import java.util.regex.Matcher;

import java.util.regex.Pattern;

@RunWith(MockitoJUnitRunner.class)

public class ParserTest {

    @Mock

    UniqueServiceLocator serviceLocator;

    @Mock

    ISLookUpService isLookUpService;

    @Mock

    WDSUtils wdsUtils;

    @Before

    public void setup() throws Exception {

        serviceLocator = mock(UniqueServiceLocator.class);

        isLookUpService = mock(ISLookUpService.class);

        wdsUtils = mock(WDSUtils.class);

        when(serviceLocator.getService(ISLookUpService.class)).thenReturn(isLookUpService);

        WDSUtils.setInstance(wdsUtils);

        when(wdsUtils.getServiceLocator()).thenReturn(serviceLocator);

        when(isLookUpService.quickSearchProfile(WDSUtils.COLLECTEDFROM_DATASOURCE_MAP_QUERY)).thenReturn(

                Arrays.asList(

                        "wds_________::r3d100010134<-->r3d100010134<-->Pangaea",

                        "wds_________::r3d100010255<-->r3d100010255<-->Inter-university Consortium for Political and Social Research",

                        "wds_________::r3d100010214<-->r3d100010214<-->World Data Center for Climate",

                        "wds_________::earthdata<-->cmr_________<-->EarthData",

                        "wds_________::r3d100000037<-->r3d100000037<-->Oak Ridge National Laboratory Distributed Active Archive Center for Biogeochemical Dynamics",

                        "wds_________::r3d100010299<-->r3d100010299<-->World Data Center for Climate",

                        "openaire____::bootstrap<-->openaire____<-->OpenAIRE Bootstrap"));

    }

    @Test

    public void testRegEx() {

        final Pattern pattern = Pattern.compile("10\\.\\d{4,9}/[-._;()/:A-Z0-9]+$", Pattern.CASE_INSENSITIVE);

        final List<String> input = Arrays.asList("https://doi.org/10.1594/PANGAEA.821878", "http://hs.pangaea.de/Images/Airphoto/PS", "https://doi.org/10.1051/epjcONnf/201009006");

        assertEquals(

                input.stream()

                        .filter(it ->

                        {

                            Matcher matcher = pattern.matcher(it);

                            return matcher.find();

                        }).count(),

                2);

    }

    @Test

    public void DMFParserTest() throws Exception {

        final InputStream inputStream = this.getClass().getResourceAsStream("/eu/dnetlib/wds/parser/DMFinput.xml");

        final String dmf = IOUtils.toString(inputStream);

        DMFResolverParser parser = new DMFResolverParser();

        WDSResolvedObject resolvedObject = (WDSResolvedObject) parser.parseObject(dmf);

        Assert.assertNotNull(resolvedObject.getType());

        Gson g = new GsonBuilder().setPrettyPrinting().create();

        assertNotNull(resolvedObject);

        System.out.println("resolvedObject = " + g.toJson(resolvedObject));

        WDSResolverSerializer serializer = new WDSResolverSerializer();

        final InputStream aStream = this.getClass().getResourceAsStream("/eu/dnetlib/wds/templates/DMFXML.st");

        serializer.setDmfTemplate(new StringTemplate(IOUtils.toString(aStream)));

        String xml = serializer.serializeToXML(resolvedObject);

        System.out.println("xml = " + xml);

    }

    @Test

    public void testOpenAireParser() throws IOException {

        OpenAireParser parser = new OpenAireParser();

        final InputStream inputStream = this.getClass().getResourceAsStream("/eu/dnetlib/wds/parser/openaire_input.xml");

        System.out.println(parser.parseObject(IOUtils.toString(inputStream)));

    }

}