tests/bundles/org.eclipse.ecf.tests.provider.dnssd/src/org/eclipse/ecf/tests/provider/dnssd/DnsSdAdvertiserServiceTest.java

/*******************************************************************************
 * Copyright (c) 2010 Markus Alexander Kuppe.
 * 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:
 *     Markus Alexander Kuppe (ecf-dev_eclipse.org <at> lemmster <dot> de) - initial API and implementation
 ******************************************************************************/
package org.eclipse.ecf.tests.provider.dnssd;

import java.io.IOException;
import java.net.UnknownHostException;
import java.util.Enumeration;
import java.util.Iterator;
import java.util.List;

import org.eclipse.ecf.core.ContainerConnectException;
import org.eclipse.ecf.core.util.ECFRuntimeException;
import org.eclipse.ecf.discovery.IServiceProperties;
import org.eclipse.ecf.provider.dnssd.DnsSdDiscoveryAdvertiser;
import org.xbill.DNS.DClass;
import org.xbill.DNS.Message;
import org.xbill.DNS.Name;
import org.xbill.DNS.Record;
import org.xbill.DNS.SRVRecord;
import org.xbill.DNS.SimpleResolver;
import org.xbill.DNS.TSIG;
import org.xbill.DNS.TXTRecord;
import org.xbill.DNS.TextParseException;
import org.xbill.DNS.Type;
import org.xbill.DNS.Update;
import org.xbill.DNS.ZoneTransferException;
import org.xbill.DNS.ZoneTransferIn;

public class DnsSdAdvertiserServiceTest extends DnsSdAbstractDiscoveryTest {

	public DnsSdAdvertiserServiceTest() {
		super();
	}

	/* (non-Javadoc)
	 * @see org.eclipse.ecf.tests.discovery.AbstractDiscoveryTest#setUp()
	 */
	protected void setUp() throws Exception {
		super.setUp();

		// purge all SRV records from the test domain
		ZoneTransferIn xfr = ZoneTransferIn.newAXFR(new Name(DnsSdTestHelper.REG_DOMAIN), DnsSdTestHelper.DNS_SERVER, null);
		List records  = xfr.run();
		for (Iterator itr = records.iterator(); itr.hasNext();) {
			Record record = (Record) itr.next();
			String name = record.getName().toString();
			if(name.startsWith("_" + DnsSdTestHelper.REG_SCHEME + "._" + DnsSdTestHelper.PROTO)) {
				Update update = new Update(Name.fromString(DnsSdTestHelper.REG_DOMAIN + "."));
				update.delete(record);
				SimpleResolver resolver = new SimpleResolver(DnsSdTestHelper.DNS_SERVER);
				resolver.setTCP(true);
				resolver.setTSIGKey(new TSIG(DnsSdTestHelper.TSIG_KEY_NAME, DnsSdTestHelper.TSIG_KEY));
				Message response = resolver.send(update);
				int rcode = response.getRcode();
				assertTrue("", rcode == 0);
			}
		}
	}
	
	private void createTXTRecords() throws TextParseException, IOException,
			UnknownHostException {
		final Name zone = Name.fromString(DnsSdTestHelper.REG_DOMAIN + ".");
		final Name name = Name.fromString("_" + DnsSdTestHelper.REG_SCHEME + "._" + DnsSdTestHelper.PROTO, zone);

		Update update = null;
		
		final IServiceProperties properties = serviceInfo.getServiceProperties();
		final Enumeration enumeration = properties.getPropertyNames();
		while (enumeration.hasMoreElements()) {
			final Object property = enumeration.nextElement();
			final String key = property.toString();
			final String value = (String) properties.getProperty(key).toString();
			final Record record = Record.fromString(name, Type.TXT, DClass.IN,
					serviceInfo.getTTL(), key + "=" + value, zone);
			update = new Update(zone);
			update.add(record);
		}
		final SimpleResolver resolver = new SimpleResolver(DnsSdTestHelper.DNS_SERVER);
		resolver.setTCP(true);
		resolver.setTSIGKey(new TSIG(DnsSdTestHelper.TSIG_KEY_NAME, DnsSdTestHelper.TSIG_KEY));
		final Message response = resolver.send(update);
		final int rcode = response.getRcode();
		assertTrue("", rcode == 0);
	}

	private void createSRVRecord() throws TextParseException, IOException,
			UnknownHostException {
		// create a service manually
		final Name zone = Name.fromString(DnsSdTestHelper.REG_DOMAIN + ".");
		final Name type = Name.fromString("_" + DnsSdTestHelper.REG_SCHEME + "._" + DnsSdTestHelper.PROTO, zone);
		final String s = serviceInfo.getPriority() + " " + 
			serviceInfo.getWeight() + " " + 
			serviceInfo.getLocation().getPort() + " " + 
			serviceInfo.getLocation().getHost() + "."; 
		final Record record = Record.fromString(type, Type.SRV, DClass.IN, DnsSdTestHelper.TTL, s, zone);
		final Update update = new Update(zone);
		update.add(record);
		final SimpleResolver resolver = new SimpleResolver(DnsSdTestHelper.DNS_SERVER);
		resolver.setTCP(true);
		resolver.setTSIGKey(new TSIG(DnsSdTestHelper.TSIG_KEY_NAME, DnsSdTestHelper.TSIG_KEY));
		final Message response = resolver.send(update);
		final int rcode = response.getRcode();
		assertTrue("", rcode == 0);
	}
	
	/* (non-Javadoc)
	 * @see org.eclipse.ecf.tests.provider.dnssd.DnsSdDiscoveryServiceTest#testRegisterService()
	 */
	public void testRegisterService() throws Exception {
		
		// actually register the service
		discoveryAdvertiser.registerService(serviceInfo);
		
		// check postcondition service is registered
		final ZoneTransferIn xfr = ZoneTransferIn.newAXFR(new Name(DnsSdTestHelper.REG_DOMAIN), DnsSdTestHelper.DNS_SERVER, null);
		assertTrue("Mismatch between DNS response and IServiceInfo", comparator.compare(serviceInfo, xfr.run()) == 0);
	}
	
	/**
	 * Tests that a register is handled correctly when no key is present
	 * for that domain and the underlying ddns call fails
	 * @throws ContainerConnectException 
	 */
	public void testRegisterServiceWithoutHostKey() throws ContainerConnectException {
		final DnsSdDiscoveryAdvertiser advertiser = new DnsSdDiscoveryAdvertiser();
		advertiser.connect(null, null);
		
		try {
			advertiser.registerService(serviceInfo);
		} catch(ECFRuntimeException e) {
			advertiser.disconnect();
			return;
		}
		fail("Register should have failed without a host key given.");
	}

	/* (non-Javadoc)
	 * @see org.eclipse.ecf.tests.provider.dnssd.DnsSdDiscoveryServiceTest#testUnregisterService()
	 */
	public void testUnregisterService() throws ContainerConnectException, IOException, ZoneTransferException {
		createSRVRecord();
		
		// unregister via ECF discovery
		discoveryAdvertiser.unregisterService(serviceInfo);
		
		// check SRV record is gone
		final ZoneTransferIn xfr = ZoneTransferIn.newAXFR(new Name(DnsSdTestHelper.REG_DOMAIN), DnsSdTestHelper.DNS_SERVER, null);
		final List records  = xfr.run();
		for (final Iterator itr = records.iterator(); itr.hasNext();) {
			final Record record = (Record) itr.next();
			if(record instanceof SRVRecord) {
				if(comparator.compare(serviceInfo, record) >= 0) {
					fail("Service not removed/unregisterd");
				}
			}
		}
	}

	/**
	 * Test that all service properties are removed along with the service
	 */
	public void testUnregisterServiceWithProperties() throws ContainerConnectException, IOException, ZoneTransferException {
		createSRVRecord();
		createTXTRecords();
		
		// unregister via ECF discovery
		discoveryAdvertiser.unregisterService(serviceInfo);
		
		// check SRV record is gone
		final ZoneTransferIn xfr = ZoneTransferIn.newAXFR(new Name(DnsSdTestHelper.REG_DOMAIN), DnsSdTestHelper.DNS_SERVER, null);
		final List records  = xfr.run();
		for (final Iterator itr = records.iterator(); itr.hasNext();) {
			final Record record = (Record) itr.next();
			if(record instanceof SRVRecord) {
				if(comparator.compare(serviceInfo, record) >= 0) {
					fail("Service not removed/unregisterd");
				}
			} else if(record instanceof TXTRecord) {
				if(comparator.compare(serviceInfo, record) >= 0) {
					fail("TXT record not removed/unregisterd");
				}
			}
		}
	}
	
	/**
	 * Test that all service properties are removed along with the service
	 */
	public void testUnregisterServiceWithForeignProperties() throws ContainerConnectException, IOException, ZoneTransferException {
		fail("Not implemented yet");
	}
	
	/* (non-Javadoc)
	 * @see org.eclipse.ecf.tests.provider.dnssd.DnsSdDiscoveryServiceTest#testUnregisterAllServices()
	 */
	public void testUnregisterAllServices() throws ContainerConnectException {
		//TODO make sure not to accidentally remove pre existing SRV records
		fail("Not yet implemented");
	}
}