TestV6NaAddrBindingManager.java

/*
 * Copyright 2009-2014 Jagornet Technologies, LLC.  All Rights Reserved.
 *
 * This software is the proprietary information of Jagornet Technologies, LLC. 
 * Use is subject to license terms.
 *
 */

/*
 *   This file TestV6NaAddrBindingManager.java is part of Jagornet DHCP.
 *
 *   Jagornet DHCP is free software: you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License as published by
 *   the Free Software Foundation, either version 3 of the License, or
 *   (at your option) any later version.
 *
 *   Jagornet DHCP is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with Jagornet DHCP.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
package com.jagornet.dhcp.server.request.binding;

import java.net.InetAddress;
import java.util.Iterator;

import com.jagornet.dhcp.db.BaseTestCase;
import com.jagornet.dhcp.db.IdentityAssoc;
import com.jagornet.dhcp.option.v6.DhcpV6ClientIdOption;
import com.jagornet.dhcp.option.v6.DhcpV6IaNaOption;
import com.jagornet.dhcp.server.config.DhcpLink;
import com.jagornet.dhcp.server.request.binding.Binding;
import com.jagornet.dhcp.server.request.binding.BindingObject;
import com.jagornet.dhcp.server.request.binding.V6BindingAddress;
import com.jagornet.dhcp.server.request.binding.V6NaAddrBindingManager;
import com.jagornet.dhcp.xml.OpaqueData;
import com.jagornet.dhcp.xml.V6ClientIdOption;

// TODO: Auto-generated Javadoc
/**
 * The Class TestBindingManager.
 */
public class TestV6NaAddrBindingManager extends BaseTestCase
{
	/** The manager. */
	private V6NaAddrBindingManager manager;
	
	/** The client id option. */
	private DhcpV6ClientIdOption clientIdOption;
	
	/* (non-Javadoc)
	 * @see com.jagornet.dhcpv6.db.BaseDbTestCase#setUp()
	 */
	@Override
	protected void setUp() throws Exception {
		super.setUp();
		manager = (V6NaAddrBindingManager) ctx.getBean("v6NaAddrBindingManager");
		manager.init();
		OpaqueData opaque = OpaqueData.Factory.newInstance();
		opaque.setHexValue(new byte[] { (byte)0xde, (byte)0xbb, (byte)0x1e,
				(byte)0xde, (byte)0xbb, (byte)0x1e });
		V6ClientIdOption clientId = V6ClientIdOption.Factory.newInstance();
		clientId.setOpaqueData(opaque);
		clientIdOption = new DhcpV6ClientIdOption(clientId);
	}
	
	/* (non-Javadoc)
	 * @see org.dbunit.DatabaseTestCase#tearDown()
	 */
	@Override
	protected void tearDown() throws Exception {
		super.tearDown();
		manager = null;
		clientIdOption = null;
	}
	
	/**
	 * Test find no current binding.
	 * 
	 * @throws Exception the exception
	 */
	public void testFindNoCurrentBinding() throws Exception {
		DhcpV6IaNaOption dhcpIaNa = new DhcpV6IaNaOption();
		dhcpIaNa.setIaId(1);
		dhcpIaNa.setT1(0);	// client SHOULD set to zero RFC3315 - 18.1.2
		dhcpIaNa.setT2(0);		
		DhcpLink clientLink = config.findLinkForAddress(InetAddress.getByName("2001:DB8:1::a"));
		assertNotNull(clientLink);
		Binding binding = manager.findCurrentBinding(clientLink, clientIdOption, 
				dhcpIaNa, null);
		assertNull(binding);
	}
	
	/**
	 * Test create solicit binding.
	 * 
	 * @throws Exception the exception
	 */
	public void testCreateSolicitBinding() throws Exception {
		DhcpV6IaNaOption dhcpIaNa = new DhcpV6IaNaOption();
		dhcpIaNa.setIaId(1);
		dhcpIaNa.setT1(0);	// client SHOULD set to zero RFC3315 - 18.1.2
		dhcpIaNa.setT2(0);
		DhcpLink clientLink = config.findLinkForAddress(InetAddress.getByName("2001:DB8:1::a"));
		assertNotNull(clientLink);
		Binding binding = manager.createSolicitBinding(clientLink, clientIdOption, 
				dhcpIaNa, null, IdentityAssoc.ADVERTISED);
		assertNotNull(binding);
		Binding binding2 = manager.findCurrentBinding(clientLink, clientIdOption, 
				dhcpIaNa, null);
		assertNotNull(binding2);
		assertEquals(binding.getDuid().length, binding2.getDuid().length);
		for (int i=0; i<binding.getDuid().length; i++) {
			assertEquals(binding.getDuid()[i], binding.getDuid()[i]);
		}
		assertEquals(binding.getIaid(), binding2.getIaid());
		assertEquals(binding.getIatype(), binding2.getIatype());
		assertEquals(binding.getState(), binding2.getState());
		assertEquals(binding.getBindingObjects().size(), binding2.getBindingObjects().size());
		Iterator<BindingObject> binding1Addrs = binding.getBindingObjects().iterator();
		Iterator<BindingObject> binding2Addrs = binding2.getBindingObjects().iterator();
		while (binding1Addrs.hasNext()) {
			V6BindingAddress ba1 = (V6BindingAddress) binding1Addrs.next();
			V6BindingAddress ba2 = (V6BindingAddress) binding2Addrs.next();
			assertEquals(ba1.getId(), ba2.getId());
			assertEquals(ba1.getIpAddress(), ba2.getIpAddress());
			assertEquals(ba1.getStartTime().getTime(), ba2.getStartTime().getTime());
			assertEquals(ba1.getPreferredEndTime().getTime(), ba2.getPreferredEndTime().getTime());
			assertEquals(ba1.getValidEndTime().getTime(), ba2.getValidEndTime().getTime());
		}
	}
}