cmtest.java

JXTA&#8482 is a set of open, generalized peer-to-peer (P2P) protocols that allow any networked devi

/*
 * Copyright (c) 2001-2007 Sun Microsystems, Inc.  All rights reserved.
 *  
 *  The Sun Project JXTA(TM) Software License
 *  
 *  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 Sun Microsystems, Inc. for JXTA(TM) technology." 
 *     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.
 *  
 *  JXTA is a registered trademark of Sun Microsystems, Inc. in the United 
 *  States and other countries.
 *  
 *  Please see the license information page at :
 *  <http://www.jxta.org/project/www/license.html> for instructions on use of 
 *  the license in source files.
 *  
 *  ====================================================================
 *  
 *  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. 
 */
package net.jxta.impl.cm;

import net.jxta.peergroup.PeerGroupID;
import net.jxta.id.IDFactory;
import net.jxta.id.ID;

import java.util.Vector;
import java.util.Enumeration;
import java.util.Arrays;
import java.util.ArrayList;
import java.util.List;
import java.util.Random;
import java.util.Collections;
import java.io.File;
import java.io.IOException;
import java.io.ByteArrayInputStream;
import java.io.InputStream;

import net.jxta.pipe.PipeService;
import net.jxta.protocol.PeerAdvertisement;
import net.jxta.protocol.PipeAdvertisement;
import junit.framework.TestSuite;
import junit.framework.TestCase;
import junit.framework.Test;
import junit.textui.TestRunner;
import net.jxta.document.MimeMediaType;
import net.jxta.document.StructuredDocumentFactory;
import net.jxta.document.StructuredDocument;
import net.jxta.document.StructuredTextDocument;
import net.jxta.document.Element;
import net.jxta.document.AdvertisementFactory;

/**
 * A CmTest unit test
 */
public class CmTest extends TestCase {

    private static final int ITERATIONS = 1000;

    private static final String[] dirname = {"Peers", "Groups", "Adv", "Raw"};
    private static final PeerGroupID pgID = IDFactory.newPeerGroupID();

    private static Cm cm = null;
    private static boolean failed = false;

    private static Random random = new Random();

    private List<PeerAdvertisement> queue = Collections.synchronizedList(new ArrayList<PeerAdvertisement>());

    /**
     * Constructor for the CmTest object
     *
     * @param testName test name
     */
    public CmTest(String testName) {
        super(testName);
        synchronized (CmTest.class) {
            if (null == cm) {
                cm = new Cm(new File(new File(".cache"), "CmTest").toURI(), "CmTest");
            }
        }
    }

    /**
     * A unit test suite for JUnit
     *
     * @return The test suite
     */
    public static Test suite() {

        return new TestSuite(CmTest.class);
    }

    /**
     * {@inheritDoc}
     */
    public static void fail(String message) {
        failed = true;
        junit.framework.TestCase.fail(message);
    }

    /**
     * The main program to test Cm
     *
     * @param argv command line arguments
     * @throws IOException if an io error occurs
     */
    public static void main(String[] argv) throws IOException {
        try {
            TestRunner.run(suite());
        } finally {
            synchronized (CmTest.class) {
                if (null != cm) {
                    cm.stop();
                    cm = null;
                }
            }
        }
        System.err.flush();
        System.out.flush();
    }

    /**
     * Create expired adv, and GarbageCollect
     */
    public void testGarbageCollect() {
        deletePeer();
        createPeer(true);
        createPipe(true);
        cm.garbageCollect();
    }

    /**
     * Run all the Cm tests sequentially. There can only be one single Cm test because
     * otherwise tearDown (which is called after every test case) will stop the Cm.
     */
    public void testEverything() {
        deletePeer();
        createPeer(false);
        createPipe(false);
        searchPeer();
        multithreadPeer();
    }

    public void testRaw() {
        createRaw();
        checkRaw();
    }

    private void createPeer(boolean expired) {
        ID advID;
        String advName;

        long t0 = System.currentTimeMillis();

        for (int i = 0; i < ITERATIONS; i++) {
            PeerAdvertisement adv = generatePeerAdv(i);

            advID = adv.getID();
            advName = advID.getUniqueValue().toString();

            try {
                if (!expired) {
                    cm.save(dirname[0], advName, adv);
                } else {
                    cm.save(dirname[0], advName, adv, 1, 1);
                }
            } catch (Exception e) {
                e.printStackTrace();
                fail("Failed to create Peer Adv: " + e.getMessage());
            }
        }
        System.out.println("Completed Creation of " + ITERATIONS + " PeerAdvertisements in: " + (System.currentTimeMillis() - t0) / 1000
                + " seconds");
    }

    private void createRaw() {

        long t0 = System.currentTimeMillis();

        for (int i = 0; i < ITERATIONS; i++) {
            byte[] testdata = new byte[1 << (i % 16)];

            Arrays.fill(testdata, (byte) (i % 16));

            try {
                cm.save(dirname[3], Integer.toString(i), testdata, Long.MAX_VALUE, Long.MAX_VALUE);
            } catch (Exception e) {
                e.printStackTrace();
                fail("Failed to raw data: " + e.getMessage());
            }
        }
        System.out.println(
                "Completed Creation of " + ITERATIONS + " Raw data in: " + (System.currentTimeMillis() - t0) / 1000 + " seconds");
    }

    private void checkRaw() {

        long t0 = System.currentTimeMillis();

        for (int i = 0; i < ITERATIONS; i++) {
            byte[] testdata = new byte[1 << (i % 16)];

            Arrays.fill(testdata, (byte) (i % 16));

            try {
                byte[] check = cm.restoreBytes(dirname[3], Integer.toString(i));

                assertTrue("values should have been equal at" + i, Arrays.equals(testdata, check));
            } catch (Exception e) {
                e.printStackTrace();
                fail("Failed to raw data: " + e.getMessage());
            }
        }
        System.out.println(
                "Completed checking of " + ITERATIONS + " Raw data in: " + (System.currentTimeMillis() - t0) / 1000 + " seconds");
    }

    private void createPipe(boolean expired) {
        ID advID;
        String advName;
        StructuredTextDocument doc = null;

        long t0 = System.currentTimeMillis();

        for (int i = 0; i < ITERATIONS; i++) {
            PipeAdvertisement adv = generatePipeAdv(i);

            advID = adv.getID();
            if (advID == null || advID.equals(ID.nullID)) {
                advName = Cm.createTmpName(doc);
            } else {
                advName = advID.getUniqueValue().toString();
            }
            try {
                if (!expired) {
                    cm.save(dirname[2], advName, adv);
                } else {
                    cm.save(dirname[2], advName, adv, 1, 1);
                }
            } catch (Exception e) {
                e.printStackTrace();
                fail("Failed to create Pipe Adv: " + e.getMessage());
            }
        }

        System.out.println(
                "Completed Creation of " + ITERATIONS + " PipeAdvertisements in: " + (System.currentTimeMillis() - t0) / 1000
                        + " seconds");
    }

    private void searchPeer() {
        long t0 = System.currentTimeMillis();
        List<net.jxta.protocol.SrdiMessage.Entry> entries = cm.getEntries(dirname[0], false);

        assertTrue("empty keys", entries.size() != 0);
        System.out.println(
                "getEntries retrieved " + entries.size() + " peers in: " + (System.currentTimeMillis() - t0) / 1000 + " seconds");

        for (int i = 0; i < (ITERATIONS / 10); i++) {
            findPeerAdv(random.nextInt(ITERATIONS));
            findPeerAdvEndswith(random.nextInt(ITERATIONS));
            findPeerAdvStartswith(random.nextInt(ITERATIONS));
            findPeerAdvContains(random.nextInt(ITERATIONS));
        }

        t0 = System.currentTimeMillis();
        List<InputStream> searchResults = cm.search(dirname[0], null, null, 10000, null);

        System.out.println("non-existent test should find 0, found: " + searchResults.size());
        System.out.println("retrieved " + searchResults.size() + " records in: " + (System.currentTimeMillis() - t0) + " ms");

        int threshold = 10;
        List<Long> expirations = new Vector<Long>();
        List<InputStream> results = cm.getRecords(dirname[0], threshold, expirations);

        assertTrue("cm.getRecords failed", threshold == results.size());
        System.out.println("Testing Query for non-existent records");
        results = cm.getRecords(dirname[1], threshold, expirations);
        assertTrue("cm.getRecords(dirname[1]) should not return results", results.size() == 0);
        System.out.println("End Testing Query for non-existent records");
    }

    private void findPeerAdv(int i) {
        long t0 = System.currentTimeMillis();

        try {
            List<InputStream> searchResults = cm.search(dirname[0], "Name", "CmTestPeer" + i, 1, null);

            assertNotNull("Null search result", searchResults);
            Enumeration result = Collections.enumeration(searchResults);

            assertNotNull("Null search enumerator", result);
            assertTrue("empty Search Result for query attr=Name value=CmTestPeer" + i, result.hasMoreElements());
            while (result.hasMoreElements()) {
                ByteArrayInputStream dataStream = (ByteArrayInputStream) result.nextElement();
                StructuredDocument doc = StructuredDocumentFactory.newStructuredDocument(MimeMediaType.XMLUTF8, dataStream);
                Enumeration en = doc.getChildren("Name");

                while (en.hasMoreElements()) {
                    String val = (String) ((Element) en.nextElement()).getValue();

                    assertTrue("Name mismatch ", val.equals("CmTestPeer" + i));
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
            fail("findPeerAdv failed: " + e.getMessage());
        }
        System.out.println("findPeerAdv retrieved CmTestPeer" + i + " in: " + (System.currentTimeMillis() - t0) + " ms");
    }

    private void findPeerAdvEndswith(int i) {