D-Net

package gr.uoa.di.driver.xml;

import java.io.ByteArrayInputStream;

import java.io.File;

import java.io.IOException;

import java.util.Date;

import javax.xml.XMLConstants;

import javax.xml.bind.JAXBException;

import javax.xml.transform.sax.SAXSource;

import javax.xml.validation.Schema;

import javax.xml.validation.SchemaFactory;

import javax.xml.validation.Validator;

import org.junit.Assert;

import org.junit.Test;

import org.xml.sax.InputSource;

import org.xml.sax.SAXException;

import eu.dnetlib.domain.functionality.SavedQuery;

import eu.dnetlib.domain.functionality.UserProfile;

import eu.dnetlib.domain.functionality.UserStore;

public class TestUserProfileXmlConverter {

	UserProfileXmlConverter conv;

	Schema schema;

	@Test

	public void testSimpleUserProfile() throws Exception {

		conv = new UserProfileXmlConverter();

		UserProfile profile = new UserProfile();

		profile.setEmail("Helenaki");

		profile.setActivationId("123");

		profile.setActive(true);

		profile.setDateOfCreation(new Date());

		testConverter(profile);

		conv = null;

	}

	@Test

	public void testUserProfile() throws Exception {

		conv = new UserProfileXmlConverter();

		UserProfile profile = new UserProfile();

		profile.setEmail("Helenaki");

		profile.setActivationId("123");

		profile.setActive(true);

		profile.setDateOfCreation(new Date());

		profile.setFilter(new SavedQuery());

		profile.getFilter().setCqlText("cql");

		profile.getFilter().setRefineText("refine");

		profile.getFilter().getCollectionIds().add("Collection 1");

		profile.getDocumentIds().add("Document 1");

		profile.getDocumentIds().add("Document 2");

		profile.setUserPhotoID("photoID");

		profile.setStore(new UserStore());

		profile.getStore().setStoreID("store ID");

		profile.getStore().setStoreServiceURL("service URL");

		SavedQuery squery = new SavedQuery();

		squery.setCqlText("cqlTesxt");

		squery.getCommunityIds().add("My community1");

		profile.getSavedQueries().add(squery);

		testConverter(profile);

		conv = null;

	}

	private void testConverter(UserProfile userProfile) throws JAXBException,

			SAXException, IOException {

		String xml1 = conv.ObjectToXml(userProfile);

		UserProfile user = conv.XmlToObject(xml1);

		String xml2 = conv.ObjectToXml(user);

		Assert.assertEquals(xml1, xml2);

		System.out.println(xml1 + "\n\n\n");

		schema = loadSchema("src/xsd/UserDSResourceType.xsd");

		validateXml(schema, xml1);

	}

	public static void validateXml(Schema schema, String xml)

			throws SAXException, IOException {

		Validator validator = schema.newValidator();

		SAXSource source = new SAXSource(new InputSource(

				new ByteArrayInputStream(xml.getBytes())));

		validator.validate(source);

	}

	private Schema loadSchema(String name) throws SAXException {

		String language = XMLConstants.W3C_XML_SCHEMA_NS_URI;

		SchemaFactory factory = SchemaFactory.newInstance(language);

		Schema schema = factory.newSchema(new File(name));

		return schema;

	}

}