testrouteadv.java

jxta_src_2.41b jxta 2.41b 最新版源码 from www.jxta.org

/*
 *  Copyright (c) 2001 Sun Microsystems, Inc.  All rights
 *  reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions
 *  are met:
 *
 *  1. Redistributions of source code must retain the above copyright
 *  notice, this list of conditions and the following disclaimer.
 *
 *  2. Redistributions in binary form must reproduce the above copyright
 *  notice, this list of conditions and the following disclaimer in
 *  the documentation and/or other materials provided with the
 *  distribution.
 *
 *  3. The end-user documentation included with the redistribution,
 *  if any, must include the following acknowledgment:
 *  "This product includes software developed by the
 *  Sun Microsystems, Inc. for Project JXTA."
 *  Alternately, this acknowledgment may appear in the software itself,
 *  if and wherever such third-party acknowledgments normally appear.
 *
 *  4. The names "Sun", "Sun Microsystems, Inc.", "JXTA" and "Project JXTA" must
 *  not be used to endorse or promote products derived from this
 *  software without prior written permission. For written
 *  permission, please contact Project JXTA at http://www.jxta.org.
 *
 *  5. Products derived from this software may not be called "JXTA",
 *  nor may "JXTA" appear in their name, without prior written
 *  permission of Sun.
 *
 *  THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
 *  WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 *  OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 *  DISCLAIMED.  IN NO EVENT SHALL SUN MICROSYSTEMS OR
 *  ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
 *  USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 *  ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 *  OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
 *  OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 *  SUCH DAMAGE.
 *  ====================================================================
 *
 *  This software consists of voluntary contributions made by many
 *  individuals on behalf of Project JXTA.  For more
 *  information on Project JXTA, please see
 *  <http://www.jxta.org/>.
 *
 *  This license is based on the BSD license adopted by the Apache Foundation.
 *
 *  $Id: TestRouteAdv.java,v 1.6 2003/10/27 22:49:25 bondolo Exp $
 */

package net.jxta.protocol;

import junit.framework.*;

import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.ByteArrayInputStream;
import java.io.StringReader;
import java.io.StringWriter;
import java.io.Reader;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;

import java.util.Enumeration;
import java.util.Vector;


import net.jxta.document.Advertisement;
import net.jxta.document.AdvertisementFactory;
import net.jxta.document.StructuredDocument;
import net.jxta.document.StructuredDocumentFactory;
import net.jxta.document.StructuredTextDocument;
import net.jxta.document.MimeMediaType;
import net.jxta.protocol.AccessPointAdvertisement;
import net.jxta.protocol.RouteAdvertisement;
import net.jxta.id.IDFactory;
import net.jxta.peer.PeerID;

import net.jxta.impl.protocol.RouteQuery;
import net.jxta.impl.protocol.RouteResponse;

/**
 *  This is a simple test for Route and AccessPoint advertisement
 */

public class TestRouteAdv extends TestCase {
    
    
    public TestRouteAdv(java.lang.String testName) {
        super(testName);
    }
    
    public static void main(java.lang.String[] args) {
        junit.textui.TestRunner.run(suite());
    }
    
    public static Test suite() {
        TestSuite suite = new TestSuite(TestRouteAdv.class);
        return suite;
    }
    
    
    protected void setUp() {
    }
    
    public void testRouteAdv() {
        // create access point advertisment for destination
        System.out.println("Create an access point advertisement");
        AccessPointAdvertisement ap = (AccessPointAdvertisement)
        AdvertisementFactory.newAdvertisement(
        AccessPointAdvertisement.getAdvertisementType());
        ap.setPeerID(IDFactory.newPeerID(IDFactory.newPeerGroupID()));
        Vector addresses = new Vector();
        addresses.addElement("TCP:123.123.123.123");
        addresses.addElement("TCP:134.134.134.134");
        ap.setEndpointAddresses(addresses);
        
        try {
            // let's print the advertisement as a plain text document
            StructuredTextDocument doc = (StructuredTextDocument)
            ap.getDocument(MimeMediaType.XMLUTF8);
            
            System.out.println("AccessPointAdvertisement original : ");
            System.out.println( doc.toString() );
            
            StringWriter out = new StringWriter( );
            out.write( doc.toString() );
            out.close();
            
            StringReader in = new StringReader( out.toString() );
            AccessPointAdvertisement apAdv = (AccessPointAdvertisement)
            AdvertisementFactory.newAdvertisement(
            MimeMediaType.XMLUTF8, in);
            in.close();
            
            doc = (StructuredTextDocument)
            apAdv.getDocument(MimeMediaType.XMLUTF8);
            System.out.println("AccessPointAdvertisement reconstructed : ");
            System.out.println( doc.toString() );
            
            //verify advertisement
            assertEquals(ap.getPeerID(), apAdv.getPeerID());
            Enumeration e1 =  apAdv.getEndpointAddresses();
            for (Enumeration e = addresses.elements(); e.hasMoreElements();) {
                assertEquals(e.nextElement(), e1.nextElement());
            }
        } catch (Exception ex) {
            ex.printStackTrace();
            fail("Error constructing advertisement");
        }
        
        
        // create Route advertisment with a single destination
        System.out.println("Create Route with single destination ");
        RouteAdvertisement route = (RouteAdvertisement)
        AdvertisementFactory.newAdvertisement(
        RouteAdvertisement.getAdvertisementType());
        route.setDest(ap);
        try {
            // let's print the advertisement as a plain text document
            StructuredTextDocument doc = (StructuredTextDocument)
            route.getDocument(MimeMediaType.XMLUTF8);
            
            System.out.println( doc.toString());
            
            StringWriter out = new StringWriter( );
            out.write( doc.toString() );
            out.close();
            
            StringReader in = new StringReader( out.toString() );
            RouteAdvertisement routeAdv = (RouteAdvertisement)
            AdvertisementFactory.newAdvertisement(
            MimeMediaType.XMLUTF8, in);
            in.close();
            
            doc = (StructuredTextDocument)
            routeAdv.getDocument(MimeMediaType.XMLUTF8);
            System.out.println("RouteAdvertisement reconstructed : ");
            System.out.println( doc.toString() );
                        
            //verify advertisement
            ap = route.getDest();
            AccessPointAdvertisement ap1 = routeAdv.getDest();
            assertEquals(ap.getPeerID(), ap1.getPeerID());
            Enumeration e1 =  ap1.getEndpointAddresses();
            for (Enumeration e =ap.getEndpointAddresses(); e.hasMoreElements();) {
                assertEquals(e.nextElement(), e1.nextElement());
            }
            
            Enumeration r1 = routeAdv.getHops();
            for (Enumeration e =routeAdv.getHops(); e.hasMoreElements();) {
                ap = (AccessPointAdvertisement) e.nextElement();
                ap1 = (AccessPointAdvertisement) r1.nextElement();
                assertEquals(ap.getPeerID(), ap1.getPeerID());
                e1 =  ap1.getEndpointAddresses();
                for (Enumeration e2 =ap.getEndpointAddresses(); e2.hasMoreElements();) {
                    assertEquals(e2.nextElement(), e1.nextElement());
                }
            }
            
        } catch (Exception ex) {
            ex.printStackTrace();
            fail("Error constructing advertisement");
        }
        
        
        // create access point advertisment for hops
        AccessPointAdvertisement ap1 = (AccessPointAdvertisement)
        AdvertisementFactory.newAdvertisement(
        AccessPointAdvertisement.getAdvertisementType());
        ap1.setPeerID(IDFactory.newPeerID(IDFactory.newPeerGroupID()));
        addresses = new Vector();
        addresses.addElement("TCP:222.222.222.222");
        addresses.addElement("TCP:244.244.244.244");
        ap1.setEndpointAddresses(addresses);
        
        
        AccessPointAdvertisement ap2 = (AccessPointAdvertisement)
        AdvertisementFactory.newAdvertisement(
        AccessPointAdvertisement.getAdvertisementType());
        ap2.setPeerID(IDFactory.newPeerID(IDFactory.newPeerGroupID()));
        addresses = new Vector();
        addresses.addElement("TCP:666.666.666.666");
        addresses.addElement("TCP:777.777.777.777");
        ap2.setEndpointAddresses(addresses);
        
        
        // create Route advertisment with a single destination
        System.out.println("Create Route with destination and one hope");
        RouteAdvertisement route1 = (RouteAdvertisement)
        AdvertisementFactory.newAdvertisement(
        RouteAdvertisement.getAdvertisementType());
        route1.setDest(ap);
        Vector hops = new Vector();
        hops.addElement(ap1);
        hops.addElement(ap2);
        route1.setHops(hops);
        try {
            // let's print the advertisement as a plain text document
            StructuredTextDocument doc = (StructuredTextDocument)
            route1.getDocument(MimeMediaType.XMLUTF8);
            
            StringWriter out = new StringWriter();
            doc.sendToWriter(out);
            System.out.println(out.toString());
            FileOutputStream fp = new FileOutputStream("route1.adv");
            fp.write(out.toString().getBytes());
            fp.close();
            out.close();
            
            FileInputStream is = new FileInputStream("route1.adv");
            RouteAdvertisement routeAdv = (RouteAdvertisement)
            AdvertisementFactory.newAdvertisement(
            MimeMediaType.XMLUTF8, is);
            is.close();
            
            doc = (StructuredTextDocument)
            routeAdv.getDocument(MimeMediaType.XMLUTF8);
            System.out.println("RouteAdvertisement reconstructed from file");
            out = new StringWriter();
            doc.sendToWriter(out);
            System.out.println(out.toString());
            out.close();
            
            //verify advertisement
            ap = route1.getDest();
            ap1 = routeAdv.getDest();
            assertEquals(ap.getPeerID(), ap1.getPeerID());
            Enumeration e1 =  ap1.getEndpointAddresses();
            for (Enumeration e =ap.getEndpointAddresses(); e.hasMoreElements();) {
                assertEquals(e.nextElement(), e1.nextElement());
            }
            
            Enumeration r1 = routeAdv.getHops();
            for (Enumeration e =routeAdv.getHops(); e.hasMoreElements();) {
                ap = (AccessPointAdvertisement) e.nextElement();
                ap1 = (AccessPointAdvertisement) r1.nextElement();
                assertEquals(ap.getPeerID(), ap1.getPeerID());
                e1 =  ap1.getEndpointAddresses();
                for (Enumeration e2 =ap.getEndpointAddresses(); e2.hasMoreElements();) {
                    assertEquals(e2.nextElement(), e1.nextElement());
                }
            }
            
        } catch (Exception ex) {
            ex.printStackTrace();
            fail("Error constructing advertisement");
        }
        
    }
    
    public void testAddDestRoute() {
        // create access point advertisment for destination
        System.out.println("Test add and remove of endpoint addresses");
        AccessPointAdvertisement ap = (AccessPointAdvertisement)
        AdvertisementFactory.newAdvertisement(
        AccessPointAdvertisement.getAdvertisementType());
        ap.setPeerID(IDFactory.newPeerID(IDFactory.newPeerGroupID()));
        Vector addresses = new Vector();
        addresses.addElement("TCP:123.123.123.123");
        addresses.addElement("TCP:134.134.134.134");
        ap.setEndpointAddresses(addresses);
        
        // create the route
        RouteAdvertisement route = (RouteAdvertisement)
        AdvertisementFactory.newAdvertisement(
        RouteAdvertisement.getAdvertisementType());
        route.setDest(ap);
        
        // add new addresses to the destination
        Vector newaddresses = new Vector();
        newaddresses.addElement("TCP:222.123.123.123");
        newaddresses.addElement("TCP:222.134.134.134");
        route.addDestEndpointAddresses(newaddresses);
        addresses.addElement("TCP:222.123.123.123");
        addresses.addElement("TCP:222.134.134.134");
        
        
        //verify advertisement
        Enumeration e1 =  addresses.elements();
        for (Enumeration e =route.getDest().getEndpointAddresses();
        e.hasMoreElements();) {
            assertEquals(e.nextElement(), e1.nextElement());
        }
        
        // test to remove access point
        route.removeDestEndpointAddresses(newaddresses);
        addresses.remove("TCP:222.123.123.123");
        addresses.remove("TCP:222.134.134.134");
        
        //verify advertisement
        e1 =  addresses.elements();
        for (Enumeration e =route.getDest().getEndpointAddresses();
        e.hasMoreElements();) {
            assertEquals(e.nextElement(), e1.nextElement());
        }