CatalogPersisterTest.java

/*******************************************************************************
 * Copyright 2012 Geoscience Australia
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *   http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 ******************************************************************************/
package au.gov.ga.earthsci.catalog;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.net.URL;

import org.junit.Test;
import org.xml.sax.SAXParseException;

import au.gov.ga.earthsci.common.persistence.PersistenceException;
import au.gov.ga.earthsci.intent.IntentManager;

/**
 * Unit tests for the {@link CatalogPersister} class
 * 
 * @author James Navin (james.navin@ga.gov.au)
 */
public class CatalogPersisterTest
{
	@Test
	public void testSaveModelNullModel() throws Exception
	{
		ICatalogModel model = null;
		ByteArrayOutputStream os = new ByteArrayOutputStream();

		CatalogPersister.saveCatalogModel(model, os);

		assertEquals(0, os.size());
	}

	@Test
	public void testSaveModelEmptyModel() throws Exception
	{
		ICatalogModel model = new CatalogModel();
		ByteArrayOutputStream os = new ByteArrayOutputStream();

		CatalogPersister.saveCatalogModel(model, os);

		String expected =
				"<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?><catalogModel><model><catalogs/></model></catalogModel>";
		String actual = stripNewLineIndents(os.toString());

		assertEquals(expected, actual);
	}

	@Test
	public void testSaveModelNonEmptyModel() throws Exception
	{
		ICatalogModel model = new CatalogModel();
		ByteArrayOutputStream os = new ByteArrayOutputStream();

		URL source = getClass().getResource("dummyresource.xml");
		model.addTopLevelCatalog(new DummyCatalogNode(source.toURI()));

		CatalogPersister.saveCatalogModel(model, os);

		String expected =
				"<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?><catalogModel><model><catalogs><element><catalog uri=\""
						+ source.toExternalForm() + "\"/></element></catalogs></model></catalogModel>";
		String actual = stripNewLineIndents(os.toString());

		assertEquals(expected, actual);
	}

	@Test(expected = IllegalArgumentException.class)
	public void testLoadModelNullStream() throws Exception
	{
		InputStream is = null;

		CatalogPersister.loadCatalogModel(is, null, null, null, false);
	}

	@Test(expected = SAXParseException.class)
	public void testLoadModelInvalidXMLDocument() throws Exception
	{
		String document = "This is not a document!";

		InputStream is = new ByteArrayInputStream(document.getBytes());

		CatalogPersister.loadCatalogModel(is, null, null, null, false);
	}

	@Test(expected = PersistenceException.class)
	public void testLoadModelValidXMLDocumentInvalidModelDocument() throws Exception
	{
		String document =
				"<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?><catalogModels><model></model></catalogModels>";

		InputStream is = new ByteArrayInputStream(document.getBytes());

		CatalogPersister.loadCatalogModel(is, null, null, null, false);
	}

	@Test
	public void testLoadModelValidEmptyModelDocument() throws Exception
	{
		String document =
				"<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?><catalogModel><model><catalogs></catalogs></model></catalogModel>";

		InputStream is = new ByteArrayInputStream(document.getBytes());

		ICatalogModel catalogModel = CatalogPersister.loadCatalogModel(is, null, null, null, false);

		assertNotNull(catalogModel);
		assertNotNull(catalogModel.getRoot());
		assertNotNull(catalogModel.getTopLevelCatalogs());
		assertEquals(0, catalogModel.getTopLevelCatalogs().size());
	}

	@Test
	public void testLoadModelValidNonEmptyModelDocument() throws Exception
	{
		URL catalog = getClass().getResource("dummyresource.xml");
		String document =
				"<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?><catalogModel><model><catalogs><element><catalog uri=\""
						+ catalog.toExternalForm() + "\"/></element></catalogs></model></catalogModel>";

		InputStream is = new ByteArrayInputStream(document.getBytes());

		IntentManager.setInstance(new DummyIntentManager());
		ICatalogModel catalogModel = CatalogPersister.loadCatalogModel(is, null, null, null, false);

		assertNotNull(catalogModel);
		assertNotNull(catalogModel.getRoot());
		assertNotNull(catalogModel.getTopLevelCatalogs());
		assertEquals(1, catalogModel.getTopLevelCatalogs().size());

		assertEquals(catalog.toExternalForm(), catalogModel.getTopLevelCatalogs().get(0).getURI().toASCIIString());
	}

	private static String stripNewLineIndents(String s)
	{
		return s.replaceAll("[\\r|\\n]+ +", "").replaceAll("\\r?\\n?", "");
	}

}