D-Net

package eu.dnetlib.resolver.store;

import java.util.Arrays;

import java.util.HashSet;

import java.util.Set;

import eu.dnetlib.dli.resolver.model.CompletionStatus;

import eu.dnetlib.dli.resolver.model.DLIObjectProvenance;

import eu.dnetlib.dli.resolver.model.DLIResolvedObject;

import eu.dnetlib.dli.resolver.model.ObjectProvisionMode;

import eu.dnetlib.pid.resolver.store.ResolverStore;

import org.junit.After;

import org.junit.Assert;

import org.junit.Ignore;

import org.junit.Test;

import org.springframework.beans.factory.annotation.Autowired;

/**

 * Created by sandro on 9/8/16.

 */

//@RunWith(SpringJUnit4ClassRunner.class)

//@ContextConfiguration(classes = ConfigurationResolverStoreTestConfig.class)

@Ignore

public class ResolverStoreTest {

	private static Set<String> expectedIndex = new HashSet<>(Arrays.asList("_id", "pid", "pid_type", "resolver", "dnet_identifier"));

	@Autowired

	private ResolverStore store;

	@After

	public void dropDown() {

		store.getMongoDatabase().drop();

	}

	@Test

	public void testIndexes() {

		store.checkIntegrityCollection();

		Set<String> inputSet = new HashSet<>(store.getIndexName());

		Assert.assertTrue(inputSet.containsAll(expectedIndex));

	}

	@Test

	public void testInsertSingleItem() {

        final DLIResolvedObject expected = createdMockObject("1234", "DOI");

        store.insertRecord("Resolver Test", expected);

        final DLIResolvedObject doi = (DLIResolvedObject) store.getRecord(expected.getPid(), expected.getPidType());

        Assert.assertNotNull(doi);

		doi.getTitles().forEach(it -> Assert.assertTrue(expected.getTitles().contains(it)));

		Assert.assertEquals(doi.getPid(), expected.getPid());

		Assert.assertEquals(doi.getPidType(), expected.getPidType());

	}

    private DLIResolvedObject createdMockObject(final String pid, final String pidType) {

        DLIResolvedObject tmp = new DLIResolvedObject();

        tmp.setPid(pid);

		tmp.setPidType(pidType);

		tmp.setTitles(Arrays.asList("Titolo 1", "Titolo 2"));

        DLIObjectProvenance p1 = new DLIObjectProvenance("datasource1", ObjectProvisionMode.collected.toString(),

                CompletionStatus.complete.toString(), "{pid, pidtype", "01/01/01", true);

        DLIObjectProvenance p2 = new DLIObjectProvenance("datasource2", ObjectProvisionMode.collected.toString(),

                CompletionStatus.incomplete.toString(), "{pid, pidtype", "02/02/02", true);

		tmp.setDatasourceProvenance(Arrays.asList(p1, p2));

		tmp.setAuthors(Arrays.asList("Author 1", "Author 2"));

		tmp.setDate("20/09/2016");

		return tmp;

	}

}