D-Net

package eu.dnetlib.enabling.is.registry; // NOPMD

import java.io.IOException;

import java.io.StringWriter;

import javax.xml.bind.JAXBException;

import javax.xml.transform.dom.DOMSource;

import javax.xml.xpath.XPathConstants;

import javax.xml.xpath.XPathFactory;

import eu.dnetlib.enabling.tools.OpaqueResource;

import eu.dnetlib.enabling.tools.StringOpaqueResource;

import eu.dnetlib.enabling.tools.blackboard.*;

import eu.dnetlib.miscutils.jaxb.JaxbFactory;

import eu.dnetlib.rmi.enabling.ISLookUpException;

import eu.dnetlib.rmi.enabling.ISLookUpService;

import eu.dnetlib.rmi.enabling.ISRegistryException;

import eu.dnetlib.rmi.enabling.ISRegistryService;

import org.apache.commons.io.IOUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.junit.Before;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.Mock;

import org.mockito.junit.MockitoJUnitRunner;

import org.w3c.dom.Element;

import org.w3c.dom.Node;

import static org.junit.Assert.*;

import static org.mockito.Matchers.*;

import static org.mockito.Mockito.verify;

import static org.mockito.Mockito.when;

/**

 * RegistryBlackboardManagerImpl test.

 * 

 * @author marko

 * 

 */

@RunWith(MockitoJUnitRunner.class)

public class RegistryBlackboardManagerImplTest {

	/**

	 * logger.

	 */

	private static final Log log = LogFactory.getLog(RegistryBlackboardManagerImplTest.class); // NOPMD by marko on 11/24/08 5:02 PM

	/**

	 * test message id.

	 */

	private static final String MSG_ID = "xxx";

	/**

	 * fake message date.

	 */

	private static final String MSG_DATE = "2009-03-19T17:55:45+01:00";

	/**

	 * service profile ID.

	 */

	private static final String PROF_ID = "123";

	/**

	 * lookup service mock.

	 */

	@Mock

	private transient ISLookUpService lookupService;

	/**

	 * registry service mock.

	 */

	@Mock

	private transient ISRegistryService registryService;

	/**

	 * instance under test.

	 */

	private transient RegistryBlackboardManagerImpl manager;

	/**

	 * common setup.

	 * 

	 * @throws JAXBException

	 *             could happen

	 * @throws IOException

	 *             shouldn't happen

	 * @throws ISLookUpException

	 *             mock

	 */

	@Before

	public void setUp() throws JAXBException, IOException, ISLookUpException {

		manager = new RegistryBlackboardManagerImpl();

		manager.setIsLookup(lookupService);

		manager.setRegistryService(registryService);

		manager.setMessageFactory(new JaxbFactory<BlackboardMessage>(BlackboardMessageImpl.class));

		manager.setMessageDater(new RegistryBlackboardManagerImpl.MessageDater() {

			@Override

			public String getCurrentDate() {

				return MSG_DATE;

			}

			@Override

			public String getNumericStamp() {

				return "1237483712.666000";

			}

		});

	}

	/**

	 * add message.

	 * 

	 * @throws JAXBException

	 *             could happen

	 * @throws ISRegistryException

	 *             could happen

	 * @throws IOException

	 *             could happen

	 * @throws ISLookUpException

	 *             mock

	 */

	@Test

	public void testAddMessage() throws JAXBException, ISRegistryException, IOException, ISLookUpException {

		final StringWriter serviceProfile = new StringWriter();

		IOUtils.copy(getClass().getResourceAsStream("serviceProfile.xml"), serviceProfile);

		when(lookupService.getResourceProfile(PROF_ID)).thenReturn(serviceProfile.toString());

		final BlackboardMessage message = manager.getMessageFactory().newInstance();

		message.setId(MSG_ID);

		message.setDate(MSG_DATE);

		message.setAction("CREATE");

		message.setActionStatus(ActionStatus.ASSIGNED);

		final BlackboardParameter param1 = new BlackboardParameterImpl();

		param1.setName("format");

		param1.setValue("DMF");

		message.getParameters().add(param1);

		final BlackboardParameter param2 = new BlackboardParameterImpl();

		param1.setName("id");

		param1.setValue("");

		message.getParameters().add(param2);

		manager.addMessage(PROF_ID, MSG_ID, manager.getMessageFactory().serialize(message));

		verify(registryService).updateProfile(eq(PROF_ID), argThat(argument -> {

			log.info("arg: " + argument);

			try {

				final OpaqueResource updatedResource = new StringOpaqueResource(argument);

				final Node messageNode = (Node) XPathFactory.newInstance().newXPath().evaluate("//BLACKBOARD/MESSAGE[last()]",

						updatedResource.asDom(), XPathConstants.NODE);

				final BlackboardMessage updatedMessage = manager.getMessageFactory().parse(new DOMSource(messageNode));

				assertEquals("messages should match", message, updatedMessage);

				final Element lastUpdated = (Element) XPathFactory.newInstance().newXPath().evaluate("//BLACKBOARD/LAST_REQUEST",

						updatedResource.asDom(), XPathConstants.NODE);

				assertEquals("stamp date", manager.getMessageDater().getNumericStamp(), lastUpdated.getAttribute("date"));

				assertEquals("stamp message", MSG_ID, lastUpdated.getTextContent());

			} catch (Exception e) {

				fail("got exception" + e);

			}

			return true;

		}), anyString());

	}

	/**

	 * delete message.

	 * 

	 * @throws ISLookUpException

	 *             mock

	 * @throws IOException

	 *             could happen

	 * @throws ISRegistryException

	 *             mock

	 */

	@Test

	public void testDeleteMessage() throws ISLookUpException, IOException, ISRegistryException {

		final StringWriter serviceProfile = new StringWriter();

		IOUtils.copy(getClass().getResourceAsStream("serviceProfileWithMessage.xml"), serviceProfile);

		when(lookupService.getResourceProfile(PROF_ID)).thenReturn(serviceProfile.toString());

		manager.deleteMessage(PROF_ID, MSG_ID);

		verify(registryService).updateProfile(eq(PROF_ID), argThat(argument -> {

			log.debug("arg: " + argument);

			try {

				final OpaqueResource updatedResource = new StringOpaqueResource(argument);

				final Node messageNode = (Node) XPathFactory.newInstance().newXPath().evaluate("//BLACKBOARD/MESSAGE[last()]",

						updatedResource.asDom(), XPathConstants.NODE);

				assertNull("message should be deleted", messageNode);

			} catch (Exception e) {

				fail("got exception" + e);

			}

			return true;

		}), anyString());

		// assertNotNull("dummy", manager);

	}

	/**

	 * replay message.

	 * 

	 * @throws ISLookUpException

	 *             mock

	 * @throws IOException

	 *             shouldn't happen

	 * @throws ISRegistryException

	 *             mock

	 * @throws JAXBException

	 *             could happen

	 */

	@Test

	public void testReplyMessage() throws ISLookUpException, IOException, ISRegistryException, JAXBException {

		final StringWriter serviceProfile = new StringWriter();

		IOUtils.copy(getClass().getResourceAsStream("serviceProfileWithMessage.xml"), serviceProfile);

		when(lookupService.getResourceProfile(PROF_ID)).thenReturn(serviceProfile.toString());

		final BlackboardMessage message = manager.getMessageFactory().newInstance();

		message.setId(MSG_ID);

		message.setDate(MSG_DATE);

		message.setAction("CREATE");

		message.setActionStatus(ActionStatus.ONGOING);

		final BlackboardParameter param1 = new BlackboardParameterImpl();

		param1.setName("format");

		param1.setValue("DMF");

		message.getParameters().add(param1);

		final BlackboardParameter param2 = new BlackboardParameterImpl();

		param1.setName("id");

		param1.setValue("");

		message.getParameters().add(param2);

		manager.replyMessage(PROF_ID, manager.getMessageFactory().serialize(message));

		verify(registryService).updateProfile(eq(PROF_ID), argThat(argument -> {

			log.info("arg: " + argument);

			try {

				final OpaqueResource updatedResource = new StringOpaqueResource(argument);

				final Node messageNode = (Node) XPathFactory.newInstance().newXPath().evaluate("//BLACKBOARD/MESSAGE[last()]",

						updatedResource.asDom(), XPathConstants.NODE);

				final BlackboardMessage updatedMessage = manager.getMessageFactory().parse(new DOMSource(messageNode));

				assertEquals("messages should match", message, updatedMessage);

				final Element lastUpdated = (Element) XPathFactory.newInstance().newXPath().evaluate("//BLACKBOARD/LAST_RESPONSE",

						updatedResource.asDom(), XPathConstants.NODE);

				assertEquals("stamp date", manager.getMessageDater().getNumericStamp(), lastUpdated.getAttribute("date"));

				assertEquals("stamp message", MSG_ID, lastUpdated.getTextContent());

			} catch (Exception e) {

				fail("got exception" + e);

			}

			return true;

		}), anyString());

	}

}