web/org.eclipse.virgo.web.enterprise.javax.persistence.extension/src/test/java/org/eclipse/virgo/web/enterprise/javax/persistence/extension/CompositePersistenceProviderResolverTest.java

/*******************************************************************************
 * Copyright (c) 2013 SAP AG
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *   SAP AG - initial contribution
 *******************************************************************************/

package org.eclipse.virgo.web.enterprise.javax.persistence.extension;

import java.io.File;
import java.io.PrintWriter;
import java.util.List;

import javax.persistence.spi.PersistenceProvider;

import org.easymock.EasyMock;
import org.eclipse.persistence.javax.persistence.osgi.OSGiProviderResolver;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.osgi.framework.BundleContext;

import org.eclipse.virgo.web.enterprise.javax.persistence.extension.CompositePersistenceProviderResolver;

public class CompositePersistenceProviderResolverTest {

	private static final String TEST_CLASSES = "./build/classes/java/test";

	@Before
	public void prepare() throws Exception {
		clean();
		prepareServicesFile();
	}

	@Test
	public void testGetPersistenceProviders() {
		BundleContext context = EasyMock.createMock(BundleContext.class);
		CompositePersistenceProviderResolver compositePersistenceProviderResolver = new CompositePersistenceProviderResolver(context);
		List<PersistenceProvider> persistenceProviders = compositePersistenceProviderResolver.getPersistenceProviders();
		Assert.assertEquals("Found Persistence Providers are not the correct number", 2, persistenceProviders.size());
		Assert.assertEquals("First provider not OSGi Persistence Provider", persistenceProviders.get(0).getClass(),  OSGiProviderResolver.class);
		Assert.assertEquals("Second provider not Test Persistence Provider", persistenceProviders.get(1).getClass(),  TestPersistenceProvider.class);
	}

	@Test
	public void testGetPersistenceProvidersNoOSGiProvider() {
		BundleContext context = EasyMock.createMock(BundleContext.class);
		CompositePersistenceProviderResolver compositePersistenceProviderResolver = new CompositePersistenceProviderResolver(context);
		compositePersistenceProviderResolver.removeOsgiProviderResolver();
		List<PersistenceProvider> persistenceProviders = compositePersistenceProviderResolver.getPersistenceProviders();
		Assert.assertEquals("Found Persistence Providers are not the correct number", 1, persistenceProviders.size());
		Assert.assertEquals("Provider not Test Persistence Provider", persistenceProviders.get(0).getClass(),  TestPersistenceProvider.class);
	}
	
	private void prepareServicesFile() throws Exception {
		new File(TEST_CLASSES + "/META-INF").mkdir();
		File servicesDir = new File(TEST_CLASSES + "/META-INF/services");
		servicesDir.mkdir();
		
		File persistenceProviderFile = new File(TEST_CLASSES + "/META-INF/services/javax.persistence.spi.PersistenceProvider");

		try (PrintWriter writer = new PrintWriter(persistenceProviderFile)) {
			writer.println("org.eclipse.virgo.web.enterprise.javax.persistence.extension.TestPersistenceProvider");
			writer.flush();
		}
	}
	
	@After
	public void clean() {
		File persistenceProviderFile = new File(TEST_CLASSES + "/META-INF/services/javax.persistence.spi.PersistenceProvider");
		if (persistenceProviderFile.exists()) {
			persistenceProviderFile.delete();
		}
		
		File servicesDir = new File(TEST_CLASSES + "/META-INF/services");
		if (servicesDir.exists()) {
			servicesDir.delete();
		}
		
		File metaInfDir = new File(TEST_CLASSES + "/META-INF");
		if (metaInfDir.exists()) {
			metaInfDir.delete();
		}
	}
}