/*
 * A CCNx library test.
 *
 * Copyright (C) 2008-2013 Palo Alto Research Center, Inc.
 *
 * This work is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License version 2 as published by the
 * Free Software Foundation. 
 * This work 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 this program; if not, write to the
 * Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 */

package org.ccnx.ccn.profiles.ccnd;

import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.fail;

import org.ccnx.ccn.LibraryTestBase;
import org.ccnx.ccn.impl.support.Log;
import org.ccnx.ccn.profiles.ccnd.PrefixRegistrationManager.ForwardingEntry;
import org.ccnx.ccn.protocol.ContentName;
import org.ccnx.ccn.protocol.MalformedContentNameStringException;
import org.junit.After;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;

/**
 * Test basic version manipulation.
 */
public class PrefixRegistrationManagerTest extends LibraryTestBase {
	
	PrefixRegistrationManager prm;
	ContentName contentNameToUse;
	NotReallyAContentName notReallyAContentNameToUse;
	public final static String prefixToUse = "ccnx:/prefix/to/test/with/";
		
	 class NotReallyAContentName extends ContentName  {
		private static final long serialVersionUID = 7618128398055066315L;
		
		public NotReallyAContentName() {
			super();
		}
		
		public NotReallyAContentName(ContentName name) {
			super(name);
		}
		
		@Override
		public long getElementLabel() { 
			return -42;
		}
	}



	/**
	 * @throws java.lang.Exception
	 */
	@BeforeClass
	public static void setUpBeforeClass() throws Exception {
		LibraryTestBase.setUpBeforeClass();
	}

	/**
	 * @throws java.lang.Exception
	 */
	@AfterClass
	public static void tearDownAfterClass() throws Exception {
		LibraryTestBase.tearDownAfterClass();
	}

	/**
	 * @throws java.lang.Exception
	 */
	@Before
	public void setUp() throws Exception {
		prm = new PrefixRegistrationManager();
		contentNameToUse = ContentName.fromURI(prefixToUse);
		notReallyAContentNameToUse = new NotReallyAContentName(contentNameToUse);
	}

	/**
	 * @throws java.lang.Exception
	 */
	@After
	public void tearDown() throws Exception {
	}
	
	@Test
	public void testCreation() {
		Log.info(Log.FAC_TEST, "Starting testCreation");

		System.out.println();
		System.out.println("PrefixRegistrationManagerTest.testCreation:");
		Integer faceID = null;
		ContentName testCN = null;
		PrefixRegistrationManager manager = null;
		
		try {
			testCN = ContentName.fromURI(prefixToUse);
		} catch (MalformedContentNameStringException e1) {
			e1.printStackTrace();
			fail("ContentName.fromURI(prefixToUse) Failed.  Reason: " + e1.getMessage());
		}
		
		try {
			manager = new PrefixRegistrationManager(putHandle);
			ForwardingEntry entry = manager.selfRegisterPrefix(testCN);
			faceID = entry.getFaceID();
			System.out.println("Created prefix: " + testCN + " on face " + faceID + " with lifetime " + entry.getLifetime());
		} catch (CCNDaemonException e) {
			System.out.println("Exception " + e.getClass().getName() + ", message: " + e.getMessage());
			System.out.println("Failed to self register prefix.");
			e.printStackTrace();
			fail("Failed to self register prefix.  Reason: " + e.getMessage());
		}
		assertNotNull(prm);
		try {
			manager.unRegisterPrefix(testCN, null, faceID);
		}catch (CCNDaemonException e) {
			System.out.println("Exception " + e.getClass().getName() + ", message: " + e.getMessage());
			System.out.println("Failed to delete prefix.");
			e.printStackTrace();
			fail("Failed to delete prefix.  Reason: " + e.getMessage());
		}
		System.out.println();
		
		try {
			manager.registerPrefix(testCN, faceID, null);
			System.out.println("Created prefix: " + testCN + " on face " + faceID);
		} catch (CCNDaemonException e) {
			System.out.println("Exception " + e.getClass().getName() + ", message: " + e.getMessage());
			System.out.println("Failed to self register prefix.");
			e.printStackTrace();
			fail("Failed to self register prefix.  Reason: " + e.getMessage());
		}
		assertNotNull(prm);
		try {
			manager.unRegisterPrefix(testCN, null, faceID);
		}catch (CCNDaemonException e) {
			System.out.println("Exception " + e.getClass().getName() + ", message: " + e.getMessage());
			System.out.println("Failed to delete prefix.");
			e.printStackTrace();
			fail("Failed to delete prefix.  Reason: " + e.getMessage());
		}
		
		Log.info(Log.FAC_TEST, "Completed testCreation");
	}
	
	@Test
	public void testException() {
		Log.info(Log.FAC_TEST, "Starting testException");

		System.out.println();
		System.out.println("PrefixRegistrationManagerTest.testException:");
		ContentName testCN = null;
		PrefixRegistrationManager manager = null;
		
		try {
			testCN = ContentName.fromURI(prefixToUse);
		} catch (MalformedContentNameStringException e1) {
			e1.printStackTrace();
			fail("ContentName.fromURI(prefixToUse) Failed.  Reason: " + e1.getMessage());
		}
		
		try {
			manager = new PrefixRegistrationManager(putHandle);
		} catch (CCNDaemonException e) {
			System.out.println("Exception " + e.getClass().getName() + ", message: " + e.getMessage());
			System.out.println("Failed to create PrefixRegistrationManager.");
			e.printStackTrace();
			fail("Failed to create PrefixRegistrationManager  Reason: " + e.getMessage());
		}
		try {
			manager = new PrefixRegistrationManager(putHandle);
			ForwardingEntry entry = manager.selfRegisterPrefix(testCN, new Integer(10024));
			fail("Failed to receive exception CCNDaemonException on selfRegisterPrefix to non-existent faceID: " + entry.getFaceID());
		} catch (CCNDaemonException e) {
			System.out.println("Received expected exception " + e.getClass().getName() + ", message: " + e.getMessage());
		}
		
		Log.info(Log.FAC_TEST, "Completed testException");
	}
}