package eu.dnetlib.enabling.is.lookup;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.mockito.Matchers.anyString;
import static org.mockito.Mockito.when;

import java.io.IOException;
import java.io.StringReader;
import java.io.StringWriter;
import java.util.Arrays;
import java.util.List;

import org.apache.commons.io.IOUtils;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.io.SAXReader;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnit44Runner;

import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpDocumentNotFoundException;
import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpException;
import eu.dnetlib.enabling.is.store.rmi.ISStoreException;
import eu.dnetlib.enabling.is.store.rmi.ISStoreService;
import eu.dnetlib.enabling.tools.StaticServiceLocator;
import eu.dnetlib.enabling.tools.XQueryUtilsImpl;

/**
 * ISLookUpService test.
 *
 * @author marko
 *
 */
@RunWith(MockitoJUnit44Runner.class)
public class ISLookUpServiceImplTest {

	/**
	 * test profile id.
	 */
	private static final String PROF_ID = "first_c2Vjb25k";

	/**
	 * List of ids.
	 */
	private static final List<String> ID_LIST = Arrays.asList(new String[] { "fff", "ggg", "hhh" });

	/**
	 * class under test.
	 */
	private transient ISLookUpServiceImpl lookup;

	/**
	 * is store service mock.
	 */
	@Mock
	private transient ISStoreService storeService;

	/**
	 * common test case init.
	 */
	@Before
	public void setUp() {
		lookup = new ISLookUpServiceImpl();
		lookup.setXqueryUtils(new XQueryUtilsImpl());
		lookup.setStoreLocator(new StaticServiceLocator<ISStoreService>(storeService));
	}

	/**
	 * test get file name from id.
	 */
	@Test
	public void testGetFileNameForId() {
		assertEquals("check", "first", lookup.getFileNameForId(PROF_ID));
	}

	/**
	 * test get file coll from id.
	 */
	@Test
	public void testGetFileCollForId() {
		assertEquals("check", "second", lookup.getFileCollForId(PROF_ID));
	}

	/**
	 * test get resource profile.
	 *
	 * @throws ISLookUpException
	 *             shouldn't happen
	 * @throws ISStoreException
	 *             shouldn't happen
	 */
	@Test
	public void testGetResourceProfile() throws ISLookUpException, ISStoreException {
		when(storeService.getXML("first", "/db/DRIVER/second")).thenReturn("<someResult/>");

		assertNotNull("get profile", lookup.getResourceProfile(PROF_ID));
	}

	/**
	 * test get resource profile when no such document exists.
	 *
	 * @throws ISLookUpException
	 *             expected
	 */
	@Test(expected = ISLookUpDocumentNotFoundException.class)
	public void testGetResourceProfileNonExisting() throws ISLookUpException {
		lookup.getResourceProfile(PROF_ID);
	}

	/**
	 * test getCollection.
	 * @throws IOException could happen
	 * @throws ISStoreException could happen
	 * @throws ISLookUpException could happen
	 * @throws DocumentException could happen
	 */
	@Test
	public void testGetCollection() throws IOException, ISStoreException, ISLookUpException, DocumentException {
		final StringWriter writer = new StringWriter();
		IOUtils.copy(getClass().getResourceAsStream("collProfile.xml"), writer);
		final String simpleProf = writer.getBuffer().toString();

		when(storeService.getXML(anyString(), anyString())).thenReturn(simpleProf);
		when(
				storeService.quickSearchXML("for $x in collection('/db/DRIVER/CollectionDSResources') where $x//FATHER/@id = '" + PROF_ID
						+ "' return $x//RESOURCE_IDENTIFIER/@value/string()")).thenReturn(ID_LIST);

		final String profile = lookup.retrieveCollection(PROF_ID);
		final SAXReader reader = new SAXReader();
		final Document doc = reader.read(new StringReader(profile));

		assertEquals("children", ID_LIST.size(), doc.selectNodes("//CHILDREN/CHILD").size());
		for (final String id : ID_LIST)
			assertNotNull("child", doc.selectSingleNode("//CHILDREN/CHILD[@id='" + id + "']"));
	}

}