testsession.java

java开源的企业总线.xmlBlaster

/*------------------------------------------------------------------------------
Name:      TestSession.java
Project:   xmlBlaster.org
Copyright: xmlBlaster.org, see xmlBlaster-LICENSE file
------------------------------------------------------------------------------*/
package org.xmlBlaster.test.authentication;

import java.util.logging.Logger;
import org.xmlBlaster.util.Global;
import org.xmlBlaster.util.XmlBlasterException;
import org.xmlBlaster.client.qos.ConnectQos;
import org.xmlBlaster.client.qos.DisconnectQos;
import org.xmlBlaster.client.I_XmlBlasterAccess;
import org.xmlBlaster.client.I_Callback;
import org.xmlBlaster.client.key.UpdateKey;
import org.xmlBlaster.client.qos.UpdateQos;
import org.xmlBlaster.util.MsgUnit;

import org.xmlBlaster.test.Util;

import junit.framework.*;


/**
 * This client does test login sessions.<br />
 * login/logout combinations are checked with subscribe()/publish() calls
 * <p />
 * This client may be invoked multiple time on the same xmlBlaster server,
 * as it cleans up everything after his tests are done.
 * <p>
 * Invoke examples:<br />
 * <pre>
 *    java junit.textui.TestRunner org.xmlBlaster.test.authentication.TestSession
 *    java junit.swingui.TestRunner org.xmlBlaster.test.authentication.TestSession
 * </pre>
 */
public class TestSession extends TestCase implements I_Callback
{
   private final Global glob;
   private static Logger log = Logger.getLogger(TestSession.class.getName());
   private String name;
   private String passwd = "secret";
   private int numReceived = 0;         // error checking

   /**
    * Constructs the TestSession object.
    * <p />
    * @param testName   The name used in the test suite
    * @param name       The name to login to the xmlBlaster
    */
   public TestSession(Global glob, String testName, String name)
   {
       super(testName);
       this.glob = glob;

       this.name = name;
   }

   /**
    * Sets up the fixture.
    * <p />
    * Connect to xmlBlaster and login
    */
   protected void setUp()
   {
   }

   /**
    * Tears down the fixture.
    * <p />
    * cleaning up .... erase() the previous message OID and logout
    */
   protected void tearDown()
   {
   }

   /**
    */
   public void testZeroSessions()
   {
      log.info("testZeroSessions() ...");
      try {
         log.info("Connecting ...");
         Global glob = this.glob.getClone(null);
         I_XmlBlasterAccess con = glob.getXmlBlasterAccess();
         ConnectQos qos = new ConnectQos(glob, name, passwd);
         qos.setMaxSessions(-16);
         con.connect(qos, this); // Login to xmlBlaster
         assertTrue("Connecting with zero sessions should not be possible", false);
      }
      catch (Exception e) {
         log.info("Success, can't connect with zero sessions");
      }
      log.info("Success in testZeroSessions()");
   }


   /**
    */
   public void testSessionOverflow()
   {
      log.info("testSessionOverflow() ...");
      int numLogin = 5;
      int maxSessions = numLogin - 2;
      I_XmlBlasterAccess[] con = new I_XmlBlasterAccess[5];
      try {
         for (int ii=0; ii<numLogin; ii++) {
            try {
               log.info("Connecting number " + ii + " of " + numLogin + " max=" + maxSessions);
               con[ii] = glob.getClone(null).getXmlBlasterAccess();
               ConnectQos qos = new ConnectQos(con[ii].getGlobal(), name, passwd);
               qos.setMaxSessions(maxSessions);
               con[ii].connect(qos, this); // Login to xmlBlaster
               if (ii >= maxSessions)
                  assertTrue("Connecting number " + ii + " of max=" + maxSessions + " is not allowed", false);
            }
            catch (Exception e) {
               if (ii >= maxSessions) {
                  log.info("Success, connecting number " + ii + " of max=" + maxSessions + " was denied: " + e.toString());
               }
               else {
                  log.severe(e.toString());
                  assertTrue("Connecting number " + ii + " of max=" + maxSessions + " should be possible", false);
               }
            }
         }
      }
      finally { // clean up
         try {
            for (int ii=0; ii<maxSessions; ii++) {
               DisconnectQos disQos = null;
               if (con[ii] != null) {
                  con[ii].disconnect(disQos);
               }
            }
         }
         catch (Throwable e) {
            assertTrue(e.toString(), false);
         }
      }
      log.info("Success in testSessionOverflow()");
   }


   /**
    * We login with session timeout 1 sec and sleep for 2 sec.
    * A get() invocation should fail since the session is expired.
    */
   public void testSessionTimeout()
   {
      log.info("testSessionTimeout() ...");
      long timeout = 1000L;
      I_XmlBlasterAccess con = null;
      Global glob = this.glob.getClone(null);
      try {
         try {
            con = glob.getXmlBlasterAccess();
            ConnectQos qos = new ConnectQos(glob, name, passwd);
            qos.setSessionTimeout(timeout);
            con.connect(qos, this);
         }
         catch (Exception e) {
            log.severe(e.toString());
            assertTrue("Login failed" + e.toString(), false);
         }

         try { Thread.sleep(timeout*2); } catch (Exception e) { } // wait until session expires

         try {
            log.info("Check that session has dissapeared ...");
            MsgUnit[] msgs = Util.adminGet(glob, "__cmd:?clientList");
            assertEquals("Can't access __cmd:?clientList", 1, msgs.length);
            log.info("Got userList=" + msgs[0].getContentStr());
            assertEquals("Session of " + name + " was not destroyed by failing callback",
                      -1, msgs[0].getContentStr().indexOf(name));
         }
         catch (XmlBlasterException e) {
            fail("Session was not destroyed: " + e.toString());
         }
      }
      finally { // clean up
         try {
            con.disconnect(null);
         }
         catch (Throwable e) {
            assertTrue(e.toString(), false);
         }
      }
      log.info("Success in testSessionTimeout()");
   }


   /**
    * We login with session timeout 2 sec and let it automatically refresh. 
    * A get() invocation should fail since the session is expired.
    */
   public void testSessionRefresh()
   {
      log.info("testSessionRefresh() ...");
      long timeout = 2000L;
      I_XmlBlasterAccess con = null;
      Global glob = this.glob.getClone(null);
      try {
         try {
            con = glob.getXmlBlasterAccess();
            ConnectQos qos = new ConnectQos(glob, name, passwd);
            qos.setRefreshSession(true);
            qos.setSessionTimeout(timeout);
            con.connect(qos, this);
         }
         catch (Exception e) {
            log.severe(e.toString());
            assertTrue("Login failed" + e.toString(), false);
         }

         log.info("Wait " + timeout*2 + " sec if session expires (because of inactivity)");
         try { Thread.sleep(timeout*2); } catch (Exception e) { }

         try {
            for (int ii=0; ii<1; ii++) {
               try { Thread.sleep(timeout/2); } catch (Exception e) { }
               log.info("Check access #" + ii + " ...");
               con.get("<key oid='__cmd:?freeMem'/>", null);
               log.info("Check access #" + ii + " OK");
            }
         }
         catch (Exception e) {
            log.severe("No access: " + e.toString());
            assertTrue("Session is expired", false);
         }
      }
      finally { // clean up
         try {
            con.disconnect(null);
         }
         catch (Throwable e) {
            assertTrue(e.toString(), false);
         }
      }
      log.info("Success in testSessionRefresh()");
   }


   /**
    * We login with session timeout 2 sec, call every 1000 millis get()
    * which should respan the session timeout. 
    * If this goes well for 8 sec, the refresh seems to work
    */
   public void testSessionTimeoutRespan()
   {
      log.info("testSessionTimeoutRespan() ...");
      long timeout = 2000L;
      I_XmlBlasterAccess con = null;
      Global glob = this.glob.getClone(null);
      try {
         try {
            con = glob.getXmlBlasterAccess();
            ConnectQos qos = new ConnectQos(glob, name, passwd);
            qos.setSessionTimeout(timeout);
            con.connect(qos, this);
         }
         catch (Exception e) {
            log.severe(e.toString());
            assertTrue("Login failed" + e.toString(), false);
         }

         try {
            for (int ii=0; ii<4; ii++) {
               try { Thread.sleep(timeout/2); } catch (Exception e) { }
               log.info("Check access #" + ii + " ...");
               con.get("<key oid='__cmd:?freeMem'/>", null);
               log.info("Check access #" + ii + " OK");
            }
         }
         catch (Exception e) {
            log.severe("No access: " + e.toString());
            assertTrue("Session is expired", false);
         }
      }
      finally { // clean up
         try {
            con.disconnect(null);
         }
         catch (Throwable e) {
            assertTrue(e.toString(), false);
         }
      }
      log.info("Success in testSessionTimeoutRespan()");
   }


   /**
    */
   public void testClearSession()
   {
      log.info("***testClearSession() ...");
      int numLogin = 5;
      int maxSessions = numLogin - 2;
      I_XmlBlasterAccess[] con = new I_XmlBlasterAccess[5];
      for (int ii=0; ii<numLogin; ii++) {
         try {
            log.info("Connecting number " + ii + " of " + numLogin + " max=" + maxSessions);
            con[ii] = glob.getClone(null).getXmlBlasterAccess();
            ConnectQos qos = new ConnectQos(con[ii].getGlobal(), name, passwd);
            qos.setMaxSessions(maxSessions);
            con[ii].connect(qos, this); // Login to xmlBlaster
         }
         catch (XmlBlasterException e) {
            if (ii >= maxSessions) {
               log.info("Success, connecting number " + ii + " of max=" + maxSessions + " was denied: " + e.toString());
               log.info("We try to clear the old sessions now");
               try {
                  ConnectQos qos = new ConnectQos(null, name, passwd);
                  qos.setMaxSessions(maxSessions);
                  qos.clearSessions(true);
                  con[ii].connect(qos, this);
                  log.info("Success, login is possible again");
                  con[ii].get("<key oid='__cmd:?freeMem'/>", null);
                  log.info("Success, get works");
               }
               catch (Exception e2) {
                  log.severe("Clear session failed: " + e2.toString());
                  e2.printStackTrace();
                  fail("Login failed" + e2.toString());
               }
            }
            else {
               log.severe(e.toString());
               assertTrue("Connecting number " + ii + " of max=" + maxSessions + " should be possible", false);
            }
         }
      }
         
      // clean up
      for (int ii=maxSessions; ii<numLogin; ii++) {
         DisconnectQos disQos = null;
         if (con[ii] != null) {
            con[ii].disconnect(disQos);
         }
      }
      log.info("***Success in testClearSession()");
   }

   /**
    * This is the callback method invoked from xmlBlaster
    * delivering us a new asynchronous message. 
    * @see org.xmlBlaster.client.I_Callback#update(String, UpdateKey, byte[], UpdateQos)
    */
   public String update(String cbSessionId, UpdateKey updateKey, byte[] content, UpdateQos updateQos)
   {
      log.info("Receiving update of a message " + updateKey.getOid());
      numReceived++;
      return "";
   }


   /**
    * Little helper, waits until the wanted number of messages are arrived
    * or returns when the given timeout occurs.
    * <p />
    * @param timeout in milliseconds
    * @param numWait how many messages to wait
    */
   private void waitOnUpdate(final long timeout, final int numWait)
   {
      long pollingInterval = 50L;  // check every 0.05 seconds
      if (timeout < 50)  pollingInterval = timeout / 10L;
      long sum = 0L;
      // check if too few are arriving
      while (numReceived < numWait) {
         try { Thread.sleep(pollingInterval); } catch( InterruptedException i) {}
         sum += pollingInterval;
         assertTrue("Timeout of " + timeout + " occurred without update", sum <= timeout);
      }

      // check if too many are arriving
      try { Thread.sleep(timeout); } catch( InterruptedException i) {}
      assertEquals("Wrong number of messages arrived", numWait, numReceived);

      numReceived = 0;
   }

   /**
    * Method is used by TestRunner to load these tests
    */
   public static Test suite()
   {
       TestSuite suite= new TestSuite();
       String loginName = "Tim";
       Global glob = new Global();
       suite.addTest(new TestSession(glob, "testZeroSessions", loginName));
       suite.addTest(new TestSession(glob, "testSessionOverflow", loginName));
       suite.addTest(new TestSession(glob, "testSessionTimeout", loginName));
       suite.addTest(new TestSession(glob, "testSessionRefresh", loginName));
       suite.addTest(new TestSession(glob, "testSessionTimeoutRespan", loginName));
       suite.addTest(new TestSession(glob, "testClearSession", loginName));
       return suite;
   }

   /**
    * Invoke: 
    * <pre>
    *   java org.xmlBlaster.test.authentication.TestSession
    *   java -Djava.compiler= junit.textui.TestRunner org.xmlBlaster.test.authentication.TestSession
    * <pre>
    */
   public static void main(String args[])
   {
      TestSession testSub = new TestSession(new Global(args), "TestSession", "Tim");
      testSub.setUp();
      testSub.testZeroSessions();
      testSub.testSessionOverflow();
      testSub.testSessionTimeout();
      testSub.testSessionRefresh();
      testSub.testSessionTimeoutRespan();
      testSub.testClearSession();
      testSub.tearDown();
   }
}