package org.wamblee.system.spring;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;

import junit.framework.TestCase;

import org.wamblee.io.ClassPathResource;
import org.wamblee.system.AbstractProvidedInterfaceDescriptor;
import org.wamblee.system.AbstractProvidedInterfaceDescriptor;
import org.wamblee.system.DefaultRequiredInterfaceDescriptor;
import org.wamblee.system.DefaultServiceRegistry;
import org.wamblee.system.ProvidedInterfaceDescriptor;
import org.wamblee.system.RequiredInterfaceDescriptor;
import org.wamblee.system.Service;
import org.wamblee.system.InterfaceDescriptor;
import org.wamblee.system.ServiceRegistry;
import org.wamblee.system.SystemAssemblyException;

public class SpringComponentTest extends TestCase {

	private static final String HELLO_SERVICE_SPRING_XML = "test.org.wamblee.system.spring.xml";
	private static final String HELLO_SERVICE_SPRING_WITH_REQS_XML = "test.org.wamblee.system.springWithRequirements.xml";
	private static final String HELLO_SERVICE_SPRING_WITH_PROPERTIES_XML = "test.org.wamblee.system.springWithProperties.xml";
    private static final String PROPERTY_FILE = "test.org.wamblee.system.spring.properties";

	private ServiceRegistry _registry;

	@Override
	protected void setUp() throws Exception {
		super.setUp();
		_registry = new DefaultServiceRegistry();
	}

	public void testBlackboxSystem() {
		SpringComponent system = new SpringComponent("system", _registry,
				new String[] { HELLO_SERVICE_SPRING_XML },
				new HashMap<String, ProvidedInterfaceDescriptor>(),
				new HashMap<RequiredInterfaceDescriptor, String>());
		system.start("Hello", new Service[0]);
		Service[] services = system.getRunningServices();
		assertEquals(0, services.length);
		
		system.stop();
	}

	public void testOneProvidedService() {
		Map<String, ProvidedInterfaceDescriptor> provided = new HashMap<String, ProvidedInterfaceDescriptor>();
		provided.put("helloService", new AbstractProvidedInterfaceDescriptor(
				"hello", HelloService.class));

		SpringComponent system = new SpringComponent("system", _registry,
				new String[] { HELLO_SERVICE_SPRING_XML }, provided,
				new HashMap<RequiredInterfaceDescriptor, String>());
		system.start("Hello", new Service[0]);
		Service[] services = system.getRunningServices();
		assertEquals(1, services.length);
		assertTrue(services[0].reference(HelloService.class) instanceof HelloService);
		assertEquals("Hello world!", services[0].reference(HelloService.class)
				.say());
		system.stop();
	}
	
	public void testWithProperties() throws IOException {
		Map<String, ProvidedInterfaceDescriptor> provided = new HashMap<String, ProvidedInterfaceDescriptor>();
		provided.put("helloService", new AbstractProvidedInterfaceDescriptor(
				"hello", HelloService.class));
		SpringComponent system = new SpringComponent("system", _registry,
				new String[] { HELLO_SERVICE_SPRING_WITH_PROPERTIES_XML },
				provided,
				new HashMap<RequiredInterfaceDescriptor, String>());
		Properties props = new Properties();
		props.load(new ClassPathResource(PROPERTY_FILE).getInputStream());
		system.addProperties(props);
		
		system.start("Hello", new Service[0]);
		Service[] services = system.getRunningServices();
		assertEquals("Property Value", services[0].reference(HelloService.class).say());
	}

	public void testWithMissingRequirement() {
		try {
			SpringComponent system = new SpringComponent("system", _registry,
					new String[] { HELLO_SERVICE_SPRING_WITH_REQS_XML },
					new HashMap<String, ProvidedInterfaceDescriptor>(),
					new HashMap<RequiredInterfaceDescriptor, String>());
			system.start("Bla", new Service[0]);
		} catch (SystemAssemblyException e) {
			//e.printStackTrace();
			return;
		}
		fail();
	}

	public void testWithRequirement() {
		Map<RequiredInterfaceDescriptor, String> required = new HashMap<RequiredInterfaceDescriptor, String>();
		required.put(new DefaultRequiredInterfaceDescriptor("hello", HelloService.class),
				"helloService");
		SpringComponent system = new SpringComponent("system", _registry,
				new String[] { HELLO_SERVICE_SPRING_WITH_REQS_XML },
				new HashMap<String, ProvidedInterfaceDescriptor>(), required);
		
		HelloService helloObject = new HelloService("ladida"); 
		Service helloService = _registry.register(new AbstractProvidedInterfaceDescriptor("hello", HelloService.class), helloObject);
		system.start("Bla", new Service[] { helloService } );
		system.stop();
	}
	
	public void testWithRequirementAndProvidedService() {
		Map<RequiredInterfaceDescriptor, String> required = new HashMap<RequiredInterfaceDescriptor, String>();
		required.put(new DefaultRequiredInterfaceDescriptor("hello", HelloService.class),
				"helloService");
		Map<String,ProvidedInterfaceDescriptor> provided = new HashMap<String, ProvidedInterfaceDescriptor>();
		provided.put("blaService", new AbstractProvidedInterfaceDescriptor("bla",
				BlaService.class));

		SpringComponent system = new SpringComponent("system", _registry,
				new String[] { HELLO_SERVICE_SPRING_WITH_REQS_XML },
				provided, required);
		
		HelloService helloObject = new HelloService("ladida"); 
		Service helloService = _registry.register(new AbstractProvidedInterfaceDescriptor("hello", HelloService.class), helloObject);
		Service[] services = system.start("Bla", new Service[] { helloService } );
		assertEquals(1, services.length);
		
		assertTrue(services[0].reference(BlaService.class) instanceof BlaService);
		assertEquals("ladida", services[0].reference(BlaService.class)
				.execute());
		system.stop();
	}

}