AbstractStorageTest.java

/******************************************************************************
 * Copyright (c) 2006, 2010 VMware Inc.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * and Apache License v2.0 which accompanies this distribution. 
 * The Eclipse Public License is available at 
 * http://www.eclipse.org/legal/epl-v10.html and the Apache License v2.0
 * is available at http://www.opensource.org/licenses/apache2.0.php.
 * You may elect to redistribute this code under either of these licenses. 
 * 
 * Contributors:
 *   VMware Inc.
 *****************************************************************************/

package org.eclipse.gemini.blueprint.test.internal.util;

import java.io.InputStream;

import junit.framework.TestCase;

import org.eclipse.gemini.blueprint.test.internal.util.jar.storage.Storage;
import org.springframework.core.io.Resource;
import org.springframework.util.FileCopyUtils;

/**
 * Set of storage tests which can be applied on various Storage
 * implementations.This class should be subclassed to run the test against
 * concrete Storage implementations.
 * 
 * @author Costin Leau
 * 
 */
public abstract class AbstractStorageTest extends TestCase {

	protected Storage storage;

	/**
	 * Create the actual storage.
	 * 
	 * @return
	 */
	protected abstract Storage createStorage();

	protected void setUp() throws Exception {
		storage = createStorage();
	}

	protected void tearDown() throws Exception {
		storage.dispose();
	}

	public void testInitialInputStream() throws Exception {
		InputStream in = storage.getInputStream();
		try {
			assertEquals(-1, in.read());
		}
		finally {
			IOUtils.closeStream(in);
		}
	}

	public void testIdenticalInputStreams() throws Exception {
		assertTrue("streams not identical", compareStreams(storage.getInputStream(), storage.getInputStream()));
	}

	public void testIdenticalInputStreamsFromResource() throws Exception {
		assertTrue("streams not identical", compareStreams(storage.getResource().getInputStream(), storage
				.getResource().getInputStream()));
	}

	public void testReadWrite() throws Exception {
		int wrote = FileCopyUtils.copy(getSampleContentAsInputStream(), storage.getOutputStream());
		System.out.println("wrote " + wrote + " bytes");
		assertTrue("streams content is different", compareStreams(getSampleContentAsInputStream(), storage
				.getInputStream()));
	}

	public void testResource() throws Exception {
		Resource res = storage.getResource();
		assertNotNull(res);
		assertFalse("underlying storage is not reusable", res.isOpen());
	}

	public void testCompareInputStreamAndResourceInputStream() throws Exception {
		InputStream in1 = storage.getInputStream();
		InputStream in2 = storage.getResource().getInputStream();
		assertTrue("streams content is different", compareStreams(in1, in2));
	}

	private boolean compareStreams(InputStream in1, InputStream in2) throws Exception {
		int count = 0;
		try {
			int b;
			while ((b = in1.read()) != -1) {
				count++;
				int a = in2.read();
				boolean same = (a == b);
				if (!same) {
					System.out.println("expected " + b + " but was " + a + ";problem occured after reading " + count + " bytes");
					return false;
				}
			}
			// check we have reached the end on both streams
			return (in1.read() == in2.read());
		}
		finally {
			IOUtils.closeStream(in1);
			IOUtils.closeStream(in2);
		}
	}

	private InputStream getSampleContentAsInputStream() throws Exception {
		return getClass().getClassLoader().getResourceAsStream(getClass().getName().replace('.', '/') + ".class");
	}
}