memorystoragetest.java

pastry的java实现的2.0b版

/*************************************************************************

"FreePastry" Peer-to-Peer Application Development Substrate 

Copyright 2002, Rice University. All rights reserved.

Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are
met:

- Redistributions of source code must retain the above copyright
notice, this list of conditions and the following disclaimer.

- 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.

- Neither  the name  of Rice  University (RICE) nor  the names  of its
contributors may be  used to endorse or promote  products derived from
this software without specific prior written permission.

This software is provided by RICE and the contributors on an "as is"
basis, without any representations or warranties of any kind, express
or implied including, but not limited to, representations or
warranties of non-infringement, merchantability or fitness for a
particular purpose. In no event shall RICE or 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.

********************************************************************************/

package rice.persistence.testing;

/*
 *  @(#) MemoryStorageTest.java
 *
 *  @author Ansley Post
 *  @author Alan Mislove
 *
 *  @version $Id: MemoryStorageTest.java 3274 2006-05-15 16:17:47Z jeffh $
 */
import java.io.*;
import java.util.*;

import rice.*;
import rice.environment.Environment;
import rice.p2p.commonapi.*;
import rice.p2p.commonapi.rawserialization.OutputBuffer;
import rice.pastry.commonapi.*;
import rice.persistence.*;

/**
 * This class is a class which tests the Storage class in the rice.persistence
 * package.
 *
 * @version $Id: pretty.settings 2305 2005-03-11 20:22:33Z jeffh $
 * @author jeffh
 */
public class MemoryStorageTest extends Test {

  private IdFactory FACTORY;
  // = new PastryIdFactory();

    /**
   * DESCRIBE THE FIELD
   */
  protected Storage storage;

  /**
   * DESCRIBE THE FIELD
   */
  protected boolean store;
  private Id[] data;
  private Integer[] metadata;

  /**
   * Builds a MemoryMemoryStorageTest
   *
   * @param store DESCRIBE THE PARAMETER
   * @param env DESCRIBE THE PARAMETER
   */
  public MemoryStorageTest(boolean store, Environment env) {
    super(env);
    FACTORY = new PastryIdFactory(env);
    storage = new MemoryStorage(FACTORY);
    data = new Id[500];
    metadata = new Integer[500];
    int[] x = new int[5];
    for (int i = 0; i < 500; i++) {
      x[3] = i;
      data[i] = FACTORY.buildId(x);
      metadata[i] = new Integer(i);
    }
    this.store = store;
  }

  /**
   * The JUnit setup method
   *
   * @param c The new Up value
   */
  public void setUp(final Continuation c) {
    final Continuation put4 =
      new Continuation() {
        public void receiveResult(Object o) {
          if (o.equals(new Boolean(true))) {
            stepDone(SUCCESS);
          } else {
            stepDone(FAILURE, "Fourth object was not inserted.");
            return;
          }

          sectionEnd();
          c.receiveResult(new Boolean(true));
        }

        public void receiveException(Exception e) {
          stepException(e);
        }
      };

    final Continuation put3 =
      new Continuation() {
        public void receiveResult(Object o) {
          if (o.equals(new Boolean(true))) {
            stepDone(SUCCESS);
          } else {
            stepDone(FAILURE, "Third object was not inserted.");
            return;
          }

          stepStart("Storing Fourth Object");
          storage.store(data[4], metadata[4], "Fourth Object", put4);
        }

        public void receiveException(Exception e) {
          stepException(e);
        }
      };

    final Continuation put2 =
      new Continuation() {
        public void receiveResult(Object o) {
          if (o.equals(new Boolean(true))) {
            stepDone(SUCCESS);
          } else {
            stepDone(FAILURE, "Second object was not inserted.");
            return;
          }

          stepStart("Storing Third Object");
          storage.store(data[3], metadata[3], "Third Object", put3);
        }

        public void receiveException(Exception e) {
          stepException(e);
        }
      };

    Continuation put1 =
      new Continuation() {
        public void receiveResult(Object o) {
          if (o.equals(new Boolean(true))) {
            stepDone(SUCCESS);
          } else {
            stepDone(FAILURE, "First object was not inserted.");
            return;
          }

          stepStart("Storing Second Object");
          storage.store(data[2], metadata[2], "Second Object", put2);
        }

        public void receiveException(Exception e) {
          stepException(e);
        }
      };

    sectionStart("Storing Objects");

    stepStart("Storing First Object");
    storage.store(data[1], metadata[1], "First Object", put1);
  }

  /**
   * A unit test for JUnit
   *
   * @param c DESCRIBE THE PARAMETER
   */
  public void testRetreival(final Continuation c) {
    final Continuation get5 =
      new Continuation() {
        public void receiveResult(Object o) {
          if (o == null) {
            stepDone(SUCCESS);
          } else {
            stepDone(FAILURE, "Fifth object was returned (should not be present).");
            return;
          }

          sectionEnd();
          c.receiveResult(new Boolean(true));
        }

        public void receiveException(Exception e) {
          stepException(e);
        }
      };

    final Continuation get4 =
      new Continuation() {
        public void receiveResult(Object o) {
          if (o == null) {
            stepDone(FAILURE, "Returned object was null.");
            return;
          }

          if (o.equals("Fourth Object")) {
            stepDone(SUCCESS);
          } else {
            stepDone(FAILURE, "Returned object was not correct: " + o);
            return;
          }

          stepStart("Attempting Fifth Object");
          storage.getObject(data[5], get5);
        }

        public void receiveException(Exception e) {
          stepException(e);
        }
      };

    final Continuation get3 =
      new Continuation() {
        public void receiveResult(Object o) {
          if (o == null) {
            stepDone(FAILURE, "Returned object was null.");
            return;
          }

          if (o.equals("Third Object")) {
            stepDone(SUCCESS);
          } else {
            stepDone(FAILURE, "Returned object was not correct: " + o);
            return;
          }

          stepStart("Retrieving Fourth Object");
          storage.getObject(data[4], get4);
        }

        public void receiveException(Exception e) {
          stepException(e);
        }
      };

    final Continuation get2 =
      new Continuation() {
        public void receiveResult(Object o) {
          if (o == null) {
            stepDone(FAILURE, "Returned object was null.");
            return;
          }

          if (o.equals("Second Object")) {
            stepDone(SUCCESS);
          } else {
            stepDone(FAILURE, "Returned object was not correct: " + o);
            return;
          }

          stepStart("Retrieving Third Object");
          storage.getObject(data[3], get3);
        }

        public void receiveException(Exception e) {
          stepException(e);
        }
      };

    final Continuation get1 =
      new Continuation() {
        public void receiveResult(Object o) {
          if (o == null) {
            stepDone(FAILURE, "Returned object was null.");
            return;
          }

          if (o.equals("First Object")) {
            stepDone(SUCCESS);
          } else {
            stepDone(FAILURE, "Returned object was not correct: " + o);
            return;
          }

          stepStart("Retrieving Second Object");
          storage.getObject(data[2], get2);
        }

        public void receiveException(Exception e) {
          stepException(e);
        }
      };

    Continuation get0 =
      new Continuation() {
        public void receiveResult(Object o) {
          if (o.equals(new Boolean(true))) {
            sectionStart("Retrieving Objects");

            stepStart("Retrieving First Object");
            storage.getObject(data[1], get1);
          } else {
            stepException(new RuntimeException("SetUp did not complete correctly."));
          }
        }

        public void receiveException(Exception e) {
          stepException(e);
        }
      };

    if (store) {
      setUp(get0);
    } else {
      get1.receiveResult("First Object");
    }
  }

  /**
   * A unit test for JUnit
   *
   * @param c DESCRIBE THE PARAMETER
   */
  public void testExists(final Continuation c) {
    testRetreival(
      new Continuation() {
        public void receiveResult(Object o) {
          if (o.equals(new Boolean(true))) {
            sectionStart("Checking for Objects");
            stepStart("Checking for First Object");
            if (storage.exists(data[1])) {
              stepDone(SUCCESS);
            } else {
              stepDone(FAILURE);
            }
            stepStart("Checking for Second Object");
            if (storage.exists(data[2])) {
              stepDone(SUCCESS);
            } else {
              stepDone(FAILURE);
            }
            stepStart("Checking for Third Object");
            if (storage.exists(data[3])) {
              stepDone(SUCCESS);
            } else {
              stepDone(FAILURE);
            }
            stepStart("Checking for Fourth Object");
            if (storage.exists(data[4])) {
              stepDone(SUCCESS);
            } else {
              stepDone(FAILURE);
            }
            stepStart("Checking for Fifth Object");
            if (!storage.exists(data[5])) {
              stepDone(SUCCESS);
            } else {
              stepDone(FAILURE);
            }
            sectionEnd();

            sectionStart("Checking for Metadata");
            stepStart("Checking for First Object Metadata");
            if (metadata[1].equals(storage.getMetadata(data[1]))) {
              stepDone(SUCCESS);
            } else {
              stepDone(FAILURE);
            }
            stepStart("Checking for Second Object Metadata");
            if (metadata[2].equals(storage.getMetadata(data[2]))) {
              stepDone(SUCCESS);
            } else {
              stepDone(FAILURE);
            }
            stepStart("Checking for Third Object Metadata");
            if (metadata[3].equals(storage.getMetadata(data[3]))) {
              stepDone(SUCCESS);
            } else {
              stepDone(FAILURE);
            }
            stepStart("Checking for Fourth Object Metadata");
            if (metadata[4].equals(storage.getMetadata(data[4]))) {
              stepDone(SUCCESS);
            } else {
              stepDone(FAILURE);
            }
            stepStart("Checking for Fifth Object Metadata");
            if (!metadata[5].equals(storage.getMetadata(data[5]))) {
              stepDone(SUCCESS);
            } else {
              stepDone(FAILURE);
            }
            sectionEnd();

            sectionStart("Modifying Metadata");
            stepStart("Changing Metadata");
            storage.setMetadata(data[4], new Integer(5001),
              new Continuation() {
                public void receiveResult(Object o) {
                  stepDone(SUCCESS);

                  stepStart("Checking for New Metadata");
                  if ((new Integer(5001)).equals(storage.getMetadata(data[4]))) {
                    stepDone(SUCCESS);
                  } else {
                    stepDone(FAILURE);
                  }
                  sectionEnd();

                  c.receiveResult(new Boolean(true));
                }

                public void receiveException(Exception e) {
                  stepException(e);
                }
              });
          } else {
            throw new RuntimeException("SetUp did not complete correctly!");
          }
        }

        public void receiveException(Exception e) {
          stepException(e);
        }
      });
  }

  /**
   * A unit test for JUnit
   *
   * @param c DESCRIBE THE PARAMETER
   */
  private void testRemoval(final Continuation c) {
    final Continuation done1 =
      new Continuation() {
        public void receiveResult(Object o) {
          if (o == null) {
            stepDone(SUCCESS);
          } else {
            stepDone(FAILURE);
          }

          sectionEnd();

          c.receiveResult(new Boolean(true));
        }

        public void receiveException(Exception e) {
          stepException(e);
        }
      };

    final Continuation retrieve1 =
      new Continuation() {
        public void receiveResult(Object o) {

        }

        public void receiveException(Exception e) {
          stepException(e);
        }
      };

    final Continuation check1 =
      new Continuation() {
        public void receiveResult(Object o) {
          if ((!store) || o.equals(new Boolean(true))) {
            stepDone(SUCCESS);
          } else {
            stepDone(FAILURE);
          }

          stepStart("Checking for First Object");
          boolean result = storage.exists(data[1]);

          if ((!store) || (!result)) {
            stepDone(SUCCESS);
          } else {
            stepDone(FAILURE);
          }

          stepStart("Attempting to Retrieve First Object");
          storage.getObject(data[1], done1);
        }

        public void receiveException(Exception e) {
          stepException(e);
        }
      };

    Continuation remove1 =
      new Continuation() {
        public void receiveResult(Object o) {
          if (o.equals(new Boolean(true))) {
            sectionStart("Testing Removal");

            stepStart("Removing First Object");
            storage.unstore(data[1], check1);
          } else {
            stepException(new RuntimeException("Exists did not complete correctly."));
          }
        }

        public void receiveException(Exception e) {
          stepException(e);
        }
      };

    testExists(remove1);
  }

  /**
   * A unit test for JUnit
   *
   * @param c DESCRIBE THE PARAMETER
   */
  private void testScan(final Continuation c) {
    final Continuation handleBadScan =
      new Continuation() {
        public void receiveResult(Object o) {
          stepDone(FAILURE, "Query returned; should have thrown exception");
        }

        public void receiveException(Exception e) {
          stepDone(SUCCESS);

          sectionEnd();

          c.receiveResult(new Boolean(true));
        }
      };

    final Continuation query =
      new Continuation() {
        public void receiveResult(Object o) {
          if (o.equals(new Boolean(true))) {
            stepDone(SUCCESS);
          } else {
            stepDone(FAILURE);
          }

          stepStart("Requesting Scan from 3 to 6");
          IdSet result = storage.scan(FACTORY.buildIdRange(data[3], data[6]));